Update config.yml (#1736)

Co-authored-by: gs-olive <113141689+gs-olive@users.noreply.github.com>

Author: frank-wei

.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"

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

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