diff --git a/.circleci/config.yml b/.circleci/config.yml index 3a1e376405..91e6a71f7e 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.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.dev20230219+cu117" + default: "2.1.0.dev20230314+cu117" torch-build-index: type: string default: "https://download.pytorch.org/whl/nightly/cu117" 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 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