-
Notifications
You must be signed in to change notification settings - Fork 18
Closed
Labels
enhancementNew feature or requestNew feature or request
Description
The _PrecompositionOperator currently does not support CuPy arrays for b because of this check if not isinstance(b, (float, np.ndarray)):. We should consider checking if not isinstance(b, (float, np.ndarray, cp.ndarray)):, however since we cannot directly import cupy this will likely require creating a cupy_type method in a new backend module (similar to the one of pylops) that returns cp.ndarray if cupy is available and np.ndarray if not... (note that this could be added to the PyLops backend module, however it would require bumping up the required minimal version of PyLops).
Metadata
Metadata
Assignees
Labels
enhancementNew feature or requestNew feature or request