Skip to content

Commit

Permalink
Update trex to 0.1.5
Browse files Browse the repository at this point in the history
Signed-off-by: Rajeev Rao <[email protected]>
  • Loading branch information
rajeevsrao committed Dec 7, 2022
1 parent cf43abd commit 82a1a11
Show file tree
Hide file tree
Showing 5 changed files with 9 additions and 5 deletions.
3 changes: 3 additions & 0 deletions tools/experimental/trt-engine-explorer/CHANGELOG.md
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,9 @@

Dates are in YYYY-MM-DD format.

## v0.1.5 (2022-12-06)
- Updated requirements.txt for Ubuntu 20.04 and 22.04

## v0.1.4 (2022-09-01)
- Validated using Python 3.8.
- Added more tests.
Expand Down
2 changes: 1 addition & 1 deletion tools/experimental/trt-engine-explorer/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -29,7 +29,7 @@ To measure per-layer execution times, when `trtexec` enqueues kernel layers for
## Supported TensorRT Versions
Starting with TensorRT 8.2, engine-plan graph and profiling data can be exported to JSON files. `trex` supports TensortRT 8.2 and 8.4.

`trex` has only been tested on Ubuntu 18.04 LTS, with Python 3.8.<br>
`trex` has only been tested on Ubuntu 18.04, 20.04, 22.04, with Python 3.8.<br>
`trex` does not require a GPU, but generating the input JSON file(s) does require a GPU.

<details><summary><h1>Installation</h1></summary>
Expand Down
5 changes: 3 additions & 2 deletions tools/experimental/trt-engine-explorer/requirements.txt
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
setuptools # for qgrid
wheel # for qgrid
protobuf==3.16.0
onnx==1.11.0
onnx==1.12.0
numpy
pandas==1.1.5
plotly
Expand All @@ -10,10 +10,11 @@ graphviz
jupyter
netron
openpyxl # for excel reporting
ipywidgets
ipywidgets==7.7.2
ipyfilechooser
jupyterlab
jupyter-dash
pytest
dtale==2.2.0
xlsxwriter
Flask==2.0.0
2 changes: 1 addition & 1 deletion tools/experimental/trt-engine-explorer/setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -33,7 +33,7 @@ def main():

setup(
name="trex",
version="0.1.4",
version="0.1.5",
description="TREX: TensorRT Engine Exploration Toolkit",
long_description=open("README.md", "r", encoding="utf-8").read(),
author="NVIDIA",
Expand Down
2 changes: 1 addition & 1 deletion tools/experimental/trt-engine-explorer/trex/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -30,4 +30,4 @@
from trex.compare_engines import *
from trex.excel_summary import *

__version__ = "0.1.4"
__version__ = "0.1.5"

0 comments on commit 82a1a11

Please sign in to comment.