Skip to content

Commit

Permalink
Add note
Browse files Browse the repository at this point in the history
  • Loading branch information
matthewdouglas committed Jan 28, 2025
1 parent 6268912 commit 04e1bc6
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion bitsandbytes/_ops.py
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,8 @@
# consider a separate op without aliased return:
# int8_linear_matmul_out(
# Tensor A, Tensor B, Tensor out, ScalarType dtype=int32
# ) -> None
# ) -> ()
# return () instead of `None` for compatibility, see here: https://github.com/pytorch/pytorch/issues/125044
torch.library.define(
"bitsandbytes::int8_linear_matmul",
"(Tensor A, Tensor B, Tensor(a!)? out=None, ScalarType dtype=int32) -> Tensor(a!)",
Expand Down

0 comments on commit 04e1bc6

Please sign in to comment.