Skip to content

Commit e929df0

Browse files
matthewdouglasakx
andauthored
Update bitsandbytes/functional.py
Co-authored-by: Aarni Koskela <[email protected]>
1 parent 6fa7905 commit e929df0

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

bitsandbytes/functional.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2347,7 +2347,12 @@ def igemmlt(A, B, out=None, Sout=None, dtype=torch.int32):
23472347
ptrB = get_ptr(B)
23482348
ptrC = get_ptr(out)
23492349
ptrRowScale = get_ptr(None)
2350-
m, n, k, lda, ldb, ldc = map(ct.c_int32, (m, n, k, lda, ldb, ldc))
2350+
m = ct.c_int32(m)
2351+
n = ct.c_int32(n)
2352+
k = ct.c_int32(k)
2353+
lda = ct.c_int32(lda)
2354+
ldb = ct.c_int32(ldb)
2355+
ldc = ct.c_int32(ldc)
23512356
stream = get_tensor_stream(A)
23522357

23532358
if dtype == torch.int32:

0 commit comments

Comments
 (0)