Skip to content

Commit f6978ae

Browse files
committed
Fix critical bug in histogram_scatter_add_2d: Undefined variable
1 parent 7a3c9af commit f6978ae

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

bitsandbytes/functional.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -959,7 +959,7 @@ def histogram_scatter_add_2d(
959959

960960
maxdim1 = ct.c_int32(histogram.shape[0])
961961
n = ct.c_int32(index1.numel())
962-
is_on_gpu([histogram, index1, index2d, source])
962+
is_on_gpu([histogram, index1, index2, source])
963963
lib.chistogram_scatter_add_2d(get_ptr(histogram), get_ptr(index1), get_ptr(index2), get_ptr(source), maxdim1, n)
964964

965965
def check_matmul(A, B, out, transposed_A, transposed_B, expected_type=torch.int8):

0 commit comments

Comments
 (0)