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!