From 7901c705d34bce6cb2509ca71072704d406f781a Mon Sep 17 00:00:00 2001 From: luka Date: Wed, 4 Dec 2024 22:39:18 +0000 Subject: [PATCH] Remove debug graph output Signed-off-by: luka --- tests/compile/test_fusion.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/tests/compile/test_fusion.py b/tests/compile/test_fusion.py index 6dc989f0d634c..6c13be791f2ea 100644 --- 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)