Remove debug graph output

Signed-off-by: luka <[email protected]>
vllm-project · Dec 4, 2024 · 7901c70 · 7901c70
1 parent 5d0cbaf
commit 7901c70
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/tests/compile/test_fusion.py b/tests/compile/test_fusion.py
@@ -64,10 +64,8 @@ def test_fusion_rmsnorm_quant(dtype, hidden_size, num_tokens, eps, static):
     torch.manual_seed(1)
 
     # Reshape pass is needed for the fusion pass to work
-    config = CompilationConfig.PassConfig(
-        enable_fusion=True,
-        enable_reshape=True,
-        dump_graph_stages=["before_fusion", "after_fusion"])
+    config = CompilationConfig.PassConfig(enable_fusion=True,
+                                          enable_reshape=True)
     reshape_pass = RedundantReshapesPass(config)
     fusion_pass = FusionPass.instance(config)