Commit 320eacb4 authored by Tim Dettmers's avatar Tim Dettmers
Browse files

Removed print statement.

parent 451fd950
......@@ -928,8 +928,6 @@ def igemmlt(A, B, SA, SB, out=None, Sout=None, dtype=torch.int32):
rows = n = shapeB[0]
assert math.prod(list(shapeA)) > 0, f'Input tensor dimensions need to be > 0: {shapeA}'
print(shapeA, math.prod(shapeA), math.prod(list(shapeA)))
print('aaa')
# if the tensor is empty, return a transformed empty tensor with the right dimensions
if shapeA[0] == 0 and dimsA == 2:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment