From e929df032f380709b131d5ce8b2c00f5780ca5b2 Mon Sep 17 00:00:00 2001
From: Matthew Douglas <38992547+matthewdouglas@users.noreply.github.com>
Date: Thu, 31 Oct 2024 11:11:16 -0400
Subject: [PATCH] Update bitsandbytes/functional.py
Co-authored-by: Aarni Koskela <akx@iki.fi>
---
bitsandbytes/functional.py | 7 ++++++-
1 file changed, 6 insertions(+), 1 deletion(-)
diff --git a/bitsandbytes/functional.py b/bitsandbytes/functional.py
index d07162219..9aadfcada 100644
--- 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: