remove print

Author: jerryzh168

tutorials/developer_api_guide/my_dtype_tensor_subclass.py

@@ -175,8 +175,6 @@ def from_float(
         dtype = torch.int16
         scale, zero_point = choose_qparams_affine(input_float, mapping_type, block_size, dtype)
         int_data = quantize_affine(input_float, block_size, scale, zero_point, dtype)
-        # int_data = (input_float / scale).to(torch.int8)
-        print("initial:", scale.shape, " int data:", int_data.shape)
         layout_tensor_ctr = get_layout_tensor_constructor(type(layout_type))
         layout_tensor = layout_tensor_ctr(int_data, scale, layout_type)
         return cls(layout_tensor, input_float.shape)
@@ -309,6 +307,7 @@ def __torch_dispatch__(cls, func, types, args, kwargs):
                 func, args, kwargs, args[0]._apply_fn_to_data(torch.detach)
             )
 
+        # Tensor parallel support START
         elif func in [aten._to_copy.default, aten.clone.default]:
             return return_and_correct_aliasing(
                 func, args, kwargs, args[0]._apply_fn_to_data(torch.clone)
@@ -334,6 +333,8 @@ def __torch_dispatch__(cls, func, types, args, kwargs):
         elif func is aten.t.default:
             return return_and_correct_aliasing(func, args, kwargs, PlainMyDTypeLayout(args[0].int_data, args[0].scale, not args[0].transposed, args[0].layout_type))
 
+        # Tensor parallel support END
+
         raise NotImplementedError(
             f"PlainMyDTypeLayout dispatch: attempting to run {func}, this is not supported"
         )