diff --git a/experimental/torch_xla2/test/test_exports.py b/experimental/torch_xla2/test/test_exports.py index 96bce09bcc5..aec955b360b 100644 --- a/experimental/torch_xla2/test/test_exports.py +++ b/experimental/torch_xla2/test/test_exports.py @@ -135,7 +135,6 @@ def test_export_dtypes(self): ## jnp.tensor(dtype=None) maps to f64 continue arg = (torch.randn(10).to(torch_dtype),) - print(arg) with torch.no_grad(): exported = torch.export.export(model, arg) stablehlo = torch_xla2.export.exported_program_to_stablehlo(exported) diff --git a/experimental/torch_xla2/test/test_symbolic_shapes.py b/experimental/torch_xla2/test/test_symbolic_shapes.py index 0c4c8d7f6d3..42e9778893a 100644 --- a/experimental/torch_xla2/test/test_symbolic_shapes.py +++ b/experimental/torch_xla2/test/test_symbolic_shapes.py @@ -69,7 +69,6 @@ def test_constraints_multiply(self): self.assertRegex(module_str, r"shape_assertion.*s[0-9]+ <= 10") self.assertRegex(module_str, r"stablehlo.constant.*2") self.assertRegex(module_str, r"shape_assertion.*2\*s[0-9]+") - print(stablehlo.mlir_module()) def test_constraint_indirection(self): """Test a model where none of the shapes are directly symbolic variables @@ -90,5 +89,4 @@ def test_constraint_indirection(self): self.assertRegex(module_str, r"shape_assertion.*s[0-9]+ <= 10") self.assertRegex(module_str, r"shape_assertion.*2\*s[0-9]+") - print(stablehlo.mlir_module())