From 4e0d950b616347bd00809d7224e885e2a4fc9a03 Mon Sep 17 00:00:00 2001 From: Wei Date: Mon, 13 Mar 2023 22:43:17 -0700 Subject: [PATCH 1/4] Update config.yml Update to 20230312 to include the fixes for flaky tests introduced by dynamo --- .circleci/config.yml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/.circleci/config.yml b/.circleci/config.yml index 3a1e376405..93e6ba0955 100644 --- a/.circleci/config.yml +++ b/.circleci/config.yml @@ -263,7 +263,7 @@ commands: parameters: torch-build: type: string - default: "2.0.0.dev20230219+cu117" + default: "2.0.0.dev20230312+cu117" torch-build-index: type: string default: "https://download.pytorch.org/whl/nightly/cu117" @@ -1171,7 +1171,7 @@ parameters: # Nightly platform config torch-build: type: string - default: "2.0.0.dev20230219+cu117" + default: "2.0.0.dev20230312+cu117" torch-build-index: type: string default: "https://download.pytorch.org/whl/nightly/cu117" From 5c8254a9c7c8d45686b87aaf7f6e5a0be9ffac82 Mon Sep 17 00:00:00 2001 From: Wei Date: Wed, 15 Mar 2023 10:49:18 -0700 Subject: [PATCH 2/4] Update config.yml --- .circleci/config.yml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/.circleci/config.yml b/.circleci/config.yml index 93e6ba0955..91e6a71f7e 100644 --- a/.circleci/config.yml +++ b/.circleci/config.yml @@ -263,7 +263,7 @@ commands: parameters: torch-build: type: string - default: "2.0.0.dev20230312+cu117" + default: "2.1.0.dev20230314+cu117" torch-build-index: type: string default: "https://download.pytorch.org/whl/nightly/cu117" @@ -1171,7 +1171,7 @@ parameters: # Nightly platform config torch-build: type: string - default: "2.0.0.dev20230312+cu117" + default: "2.1.0.dev20230314+cu117" torch-build-index: type: string default: "https://download.pytorch.org/whl/nightly/cu117" From 579445d2336b85a10bc211d1ad0791dccdb8c662 Mon Sep 17 00:00:00 2001 From: Wei Date: Wed, 15 Mar 2023 10:50:20 -0700 Subject: [PATCH 3/4] Update requirements.txt --- py/requirements.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/py/requirements.txt b/py/requirements.txt index d28469ba15..ca1e13a11c 100644 --- a/py/requirements.txt +++ b/py/requirements.txt @@ -1,7 +1,7 @@ numpy pybind11==2.6.2 --extra-index-url https://download.pytorch.org/whl/nightly/cu117 -torch==2.0.0.dev20230219+cu117 +torch==2.1.0.dev20230314+cu117 torchvision==0.15.0.dev20230219+cu117 --extra-index-url https://pypi.ngc.nvidia.com tensorrt==8.5.1.7 From 000336c49b172af4800ba29bfdf1d0352e9419ac Mon Sep 17 00:00:00 2001 From: gs-olive <113141689+gs-olive@users.noreply.github.com> Date: Wed, 22 Mar 2023 14:10:18 -0700 Subject: [PATCH 4/4] fix: Remove references to `specialize_int_float` - This argument was removed from the Dynamo config in new versions and causes the CI to fail. --- py/torch_tensorrt/fx/tracer/dispatch_tracer/aten_tracer.py | 4 ---- 1 file changed, 4 deletions(-) diff --git a/py/torch_tensorrt/fx/tracer/dispatch_tracer/aten_tracer.py b/py/torch_tensorrt/fx/tracer/dispatch_tracer/aten_tracer.py index 2a252bd965..67cb9cb739 100644 --- a/py/torch_tensorrt/fx/tracer/dispatch_tracer/aten_tracer.py +++ b/py/torch_tensorrt/fx/tracer/dispatch_tracer/aten_tracer.py @@ -42,28 +42,24 @@ def __init__( capture_scalar_outputs: bool = True, guard_nn_modules: bool = True, dynamic_shapes: bool = True, - specialize_int_float: bool = True, verbose: bool = True, ) -> None: self.capture_scalar_outputs = capture_scalar_outputs self.guard_nn_modules = guard_nn_modules self.dynamic_shapes = dynamic_shapes - self.specialize_int_float = specialize_int_float self.verbose = verbose def activate(self) -> None: torchdynamo.config.capture_scalar_outputs = self.capture_scalar_outputs torchdynamo.config.guard_nn_modules = self.guard_nn_modules torchdynamo.config.dynamic_shapes = self.dynamic_shapes - torchdynamo.config.specialize_int_float = self.specialize_int_float torchdynamo.config.verbose = self.verbose def deactivate(self) -> None: torchdynamo.config.capture_scalar_outputs = True torchdynamo.config.guard_nn_modules = True torchdynamo.config.dynamic_shapes = True - torchdynamo.config.specialize_int_float = True torchdynamo.config.verbose = True