Fix cpu offloading
#5
by
tolgacangoz - opened
Thanks a lot! These look correct to me and will save the additional moves from cpu -> cuda, and then back to cpu due to offloading. Will wait for @sayakpaul to review once as well
tolgacangoz changed pull request title from
Fix cpu offload
to Fix cpu offloading
a-r-r-o-w changed pull request status to
merged
Thanks for merging!