Merge pull request #922 from jaybdub/dev_trt_10_refactor

[WIP] Dev trt 10 refactor
NVIDIA-AI-IOT · May 3, 2024 · 4e820ae · 4e820ae
2 parents 19a3176 + 85ee502
commit 4e820ae
Show file tree

Hide file tree

Showing 112 changed files with 3,441 additions and 5,925 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -2,6 +2,10 @@
 
 ## [master](https://github.com/NVIDIA-AI-IOT/torch2trt/tree/master)
 
+- Added inference and conversion support for TensorRT 10
+- Removed redundant converters, and merged converters for ND convolutions, pooling, etc.
+- Migrated test cases to use PyTest
+- Added unique axis names when using ONNX to support mis-matched dynamic axes (needed for whisper)
 
 ## [v0.5.0](https://github.com/NVIDIA-AI-IOT/torch2trt/tree/v0.5.0) - 05/3/2024
 

diff --git a/requirements/requirements_10.txt b/requirements/requirements_10.txt
@@ -0,0 +1,5 @@
+tensorrt==10.0.1
+torch
+torchvision
+timm
+onnx_graphsurgeon
diff --git a/requirements/requirements_8.txt b/requirements/requirements_8.txt
@@ -0,0 +1,5 @@
+tensorrt==8.6.1
+torch
+torchvision
+timm
+onnx_graphsurgeon
diff --git a/torch2trt/tests/__init__.py → tests/__init__.py b/torch2trt/tests/__init__.py → tests/__init__.py
diff --git a/torch2trt/tests/timm/__init__.py → tests/converter_tests/__init__.py b/torch2trt/tests/timm/__init__.py → tests/converter_tests/__init__.py