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!