remove debug prints

pytorch · May 22, 2024 · 9e22442 · 9e22442
1 parent cd470bc
commit 9e22442
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 3 deletions.
diff --git 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
@@ -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())