[Bugfix] SymIntArrayRef expected to contain concrete integers (#10170)

Signed-off-by: Bill Nell <[email protected]>
vllm-project · Nov 8, 2024 · 10b67d8 · 10b67d8
1 parent 4f93dfe
commit 10b67d8
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/vllm/compilation/backends.py b/vllm/compilation/backends.py
@@ -329,7 +329,8 @@ def run(self, *args):
             self.fake_mode.from_tensor(t) if isinstance(t, torch.Tensor) else t
             for t in args
         ]
-        return super().run(*fake_args)
+        with self.fake_mode:
+            return super().run(*fake_args)
 
     def call_module(self, target: torch.fx.node.Target,
                     args: Tuple[torch.fx.node.Argument,