mrq/bitsandbytes-rocm
1
0
Fork 1
Code Issues 1 Pull Requests Packages Projects Releases Wiki Activity

typo

Browse Source
This commit is contained in:
Mitchell Wortsman 2023-04-08 19:36:17 +00:00
parent da524d97c9
commit d677a71607
1 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
bitsandbytes/nn/triton_based_modules.py
View File

@ -112,8 +112,6 @@ class _switchback_global_mem_efficient(torch.autograd.Function):
del X del X
W_int8, state_W = quantize_global(W) W_int8, state_W = quantize_global(W)
print('in mem eff backward.')
# save for backward. # save for backward.
ctx.save_for_backward = X_int8, state_X, W_int8, state_W ctx.save_for_backward = X_int8, state_X, W_int8, state_W