fix: Set dynamic=False in torch.compile call (#2240)

pytorch · Aug 22, 2023 · 1133432 · 1133432
1 parent 148b3ba
commit 1133432
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/py/torch_tensorrt/_compile.py b/py/torch_tensorrt/_compile.py
@@ -239,7 +239,10 @@ def torch_compile(module: torch.nn.Module, **kwargs: Any) -> Any:
     """
     from torch_tensorrt.dynamo.backend import torch_tensorrt_backend
 
-    boxed_fn = torch.compile(module, backend=torch_tensorrt_backend, options={**kwargs})
+    # TODO: Remove dynamic=False when SymInt Dynamic shape support is ready
+    boxed_fn = torch.compile(
+        module, backend=torch_tensorrt_backend, dynamic=False, options={**kwargs}
+    )
 
     return boxed_fn