Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/torchani.git" did not exist on "f73d8ae1ca88b1553938abbcc982f314bbb31221"
Commit a9fe0ff9 authored by justheuristic's avatar justheuristic
Browse files

recast to fp16

parent fc4a135e
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
bitsandbytes/autograd/_functions.py
View file @ a9fe0ff9
...@@ -275,7 +275,7 @@ class MatMul8bitLt(torch.autograd.Function): ...@@ -275,7 +275,7 @@ class MatMul8bitLt(torch.autograd.Function):
state.SCB, state.SCB,
state.SCBt, state.SCBt,
coo_tensorB, coo_tensorB,
) = F.double_quant(B) ) = F.double_quant(B.to(torch.float16))
state.CxB, state.SB = F.transform(CB, to_order=formatB) state.CxB, state.SB = F.transform(CB, to_order=formatB)
else: else:
has_grad = False has_grad = False
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment