Skip to content

Commit 04e1bc6

Browse files
Add note
1 parent 6268912 commit 04e1bc6

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

bitsandbytes/_ops.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,8 @@
2323
# consider a separate op without aliased return:
2424
# int8_linear_matmul_out(
2525
# Tensor A, Tensor B, Tensor out, ScalarType dtype=int32
26-
# ) -> None
26+
# ) -> ()
27+
# return () instead of `None` for compatibility, see here: https://github.com/pytorch/pytorch/issues/125044
2728
torch.library.define(
2829
"bitsandbytes::int8_linear_matmul",
2930
"(Tensor A, Tensor B, Tensor(a!)? out=None, ScalarType dtype=int32) -> Tensor(a!)",

0 commit comments

Comments
 (0)