summaryrefslogtreecommitdiff
path: root/bitsandbytes/optim/adam.py
diff options
context:
space:
mode:
authorTim Dettmers <tim.dettmers@gmail.com>2022-08-03 09:05:37 -0700
committerTim Dettmers <tim.dettmers@gmail.com>2022-08-03 09:05:37 -0700
commit2f01865a2ff4ad3345c156f7a2f76fe79ec4ed9a (patch)
tree8c0e2df4d9ad79c3f67dfe2b3e27dd52c1a9aeda /bitsandbytes/optim/adam.py
parentdc8c9efdb33130f960adc864916b67d0cb744dbb (diff)
Added CUDA block assert and is_on_gpu check.
Diffstat (limited to 'bitsandbytes/optim/adam.py')
0 files changed, 0 insertions, 0 deletions