Update bitsandbytes/functional.py

Co-authored-by: Aarni Koskela <[email protected]>
bitsandbytes-foundation · Oct 31, 2024 · e929df0 · e929df0
1 parent 6fa7905
commit e929df0
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/bitsandbytes/functional.py b/bitsandbytes/functional.py
@@ -2347,7 +2347,12 @@ def igemmlt(A, B, out=None, Sout=None, dtype=torch.int32):
         ptrB = get_ptr(B)
         ptrC = get_ptr(out)
         ptrRowScale = get_ptr(None)
-        m, n, k, lda, ldb, ldc = map(ct.c_int32, (m, n, k, lda, ldb, ldc))
+        m = ct.c_int32(m)
+        n = ct.c_int32(n)
+        k = ct.c_int32(k)
+        lda = ct.c_int32(lda)
+        ldb = ct.c_int32(ldb)
+        ldc = ct.c_int32(ldc)
         stream = get_tensor_stream(A)
 
         if dtype == torch.int32: