diff --git a/backends/cadence/aot/tests/test_fusion_ops_passes.py b/backends/cadence/aot/tests/test_fusion_ops_passes.py index d0cf63fbaa8..2c9d56819c0 100644 --- a/backends/cadence/aot/tests/test_fusion_ops_passes.py +++ b/backends/cadence/aot/tests/test_fusion_ops_passes.py @@ -328,7 +328,6 @@ def forward(self, x): model = M() graph_module = export_to_edge(model, (inputs,)).exported_program().graph_module graph_module = FuseQuantDequantToRequantizePass()(graph_module).graph_module - graph_module.print_readable() self.check_op_counts( graph_module, diff --git a/backends/cadence/aot/tests/test_memory_passes.py b/backends/cadence/aot/tests/test_memory_passes.py index 245f3d64003..c32809c2bff 100644 --- a/backends/cadence/aot/tests/test_memory_passes.py +++ b/backends/cadence/aot/tests/test_memory_passes.py @@ -711,7 +711,6 @@ def forward(self, x) -> torch.Tensor: .exported_program() .graph_module ) - graph_module.print_readable() self.assertEqual(count_node(graph_module, torch.ops.aten._cat_nop.out), 1) self.assertEqual( count_node(graph_module, torch.ops.aten._slice_copy_nop.Tensor_out), 0 @@ -741,7 +740,6 @@ def forward(self, x) -> torch.Tensor: .exported_program() .graph_module ) - graph_module.print_readable() self.assertEqual(count_node(graph_module, torch.ops.aten._cat_nop.out), 2) self.assertEqual(count_node(graph_module, torch.ops.aten.cat.out), 0) self.verify_nop_memory_alloc(graph_module) diff --git a/backends/cadence/aot/tests/test_remove_ops_passes.py b/backends/cadence/aot/tests/test_remove_ops_passes.py index dba4f711864..25805000848 100644 --- a/backends/cadence/aot/tests/test_remove_ops_passes.py +++ b/backends/cadence/aot/tests/test_remove_ops_passes.py @@ -100,7 +100,6 @@ def forward(self, t: torch.Tensor): p = RemoveNopAddOpPass() graph_after_passes = cast(PassResult, p(graph_module)).graph_module - graph_module.print_readable() self.assertEqual( count_node(graph_after_passes, exir_ops.edge.aten.add.Tensor), 0, @@ -140,7 +139,6 @@ def forward(self, t: torch.Tensor): p = RemoveNopMulOpPass() graph_after_passes = cast(PassResult, p(graph_module)).graph_module - graph_module.print_readable() self.assertEqual( count_node(graph_after_passes, exir_ops.edge.aten.mul.Tensor), 0,