[Bugfix] SymIntArrayRef expected to contain concrete integers (vllm-p…

…roject#10170) Signed-off-by: Bill Nell <[email protected]> Signed-off-by: Maxime Fournioux <[email protected]>
mfournioux · Nov 20, 2024 · 8013cb2 · 8013cb2
1 parent 385ff71
commit 8013cb2
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,