Remove debug print for tensor shapes

This commit is contained in:
comfyanonymous 2025-11-03 19:00:23 -05:00 committed by GitHub
parent 6c84d84a48
commit 8230919053
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -461,7 +461,6 @@ def fp8_addmm(func, args, kwargs):
plain_input, scale_a = TensorCoreFP8Layout.get_plain_tensors(input_tensor)
plain_weight, scale_b = TensorCoreFP8Layout.get_plain_tensors(weight)
print(plain_input.shape, plain_weight.shape, bias.shape)
output = torch._scaled_mm(
plain_input.contiguous(),
plain_weight,