remove print

pytorch · Feb 27, 2025 · b5c8acb · b5c8acb
1 parent 277728c
commit b5c8acb
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 4 deletions.
diff --git a/test/dtypes/test_affine_quantized.py b/test/dtypes/test_affine_quantized.py
@@ -180,7 +180,6 @@ def test_print_quantized_module(self, apply_quant):
         "apply_quant", get_quantization_functions(False, True, "cuda", False)
     )
     def test_copy_(self, apply_quant):
-        print("apply_quant:", apply_quant)
         linear = torch.nn.Linear(128, 256, dtype=torch.bfloat16, device="cuda")
         ql = apply_quant(linear)
         linear2 = torch.nn.Linear(128, 256, dtype=torch.bfloat16, device="cuda")

diff --git a/torchao/dtypes/uintx/tensor_core_tiled_layout.py b/torchao/dtypes/uintx/tensor_core_tiled_layout.py
@@ -36,9 +36,6 @@ def _aqt_is_tensor_core_tile_uint4(aqt):
 def _same_metadata(
     self: "TensorCoreTiledAQTTensorImpl", src: "TensorCoreTiledAQTTensorImpl"
 ) -> bool:
-    print(
-        f"{isinstance(self, TensorCoreTiledAQTTensorImpl)}, {isinstance(src, TensorCoreTiledAQTTensorImpl)}, {self.packed_weight.shape == src.packed_weight.shape}, {self.scale_and_zero.shape == src.scale_and_zero.shape}, {self.transposed, src.transposed}, {type(self._layout)}, {type(src._layout)}"
-    )
     return (
         isinstance(self, TensorCoreTiledAQTTensorImpl)
         and isinstance(src, TensorCoreTiledAQTTensorImpl)