mrq/bitsandbytes-rocm
1
0
Fork 1
Code Issues 1 Pull Requests Packages Projects Releases Wiki Activity

Removed print statement.

Browse Source
This commit is contained in:
Tim Dettmers 2022-08-03 14:17:54 -07:00
parent 451fd9506e
commit 320eacb4c2
1 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
bitsandbytes/functional.py
View File

@ -928,8 +928,6 @@ def igemmlt(A, B, SA, SB, out=None, Sout=None, dtype=torch.int32):
rows = n = shapeB[0] rows = n = shapeB[0]
assert math.prod(list(shapeA)) > 0, f'Input tensor dimensions need to be > 0: {shapeA}' 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 the tensor is empty, return a transformed empty tensor with the right dimensions
if shapeA[0] == 0 and dimsA == 2: if shapeA[0] == 0 and dimsA == 2: