-
Notifications
You must be signed in to change notification settings - Fork 123
Open
Labels
enhancementNew feature or requestNew feature or request
Description
The ToCupy operator does not allow one to choose on which device the array will be moved to. Add a device parameter allowing the user to specify it. This can follow what we already do here
pylops/pylops/utils/backend.py
Line 547 in 81b6ff9
| with cp.cuda.Device(x.device): |
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
enhancementNew feature or requestNew feature or request