You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
It seems that this feature has already been implemented at the C++ level (here). So, I am wondering if this can be implemented on the Python interface level to allow changing devices (GPU to CPU or vice versa) easily.
Thank you!
The text was updated successfully, but these errors were encountered:
Normally, If you have StorageView object in the python, you can use to function to move data from CPU to GPU or vice versa. Read cpp/storage_view.cc for more.
@minhthuc2502 Thank you for your answer! However, looking at this line in cpp/storage_view.cc, it seems that the function to only supports dtype as the input argument.
It seems that this feature has already been implemented at the C++ level (here). So, I am wondering if this can be implemented on the Python interface level to allow changing devices (GPU to CPU or vice versa) easily.
Thank you!
The text was updated successfully, but these errors were encountered: