We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 7c9281c commit 83cea6bCopy full SHA for 83cea6b
bitsandbytes/autograd/_functions.py
@@ -337,7 +337,7 @@ def backward(ctx, grad_output):
337
# compute grad_bias first before changing grad_output dtype
338
grad_bias = grad_output.sum(0, dtype=ctx.dtype_bias)
339
340
- # Cast grad_output to fp16
+ # Cast grad_output to fp16
341
if len(grad_output.shape) == 3:
342
grad_output = grad_output.reshape(-1, grad_output.shape[-1]).contiguous()
343
0 commit comments