ENH: add cupy tensor space

odlgroup · Nov 13, 2017 · 4fc8c3b · 4fc8c3b
1 parent 6b73551
commit 4fc8c3b
Show file tree

Hide file tree

Showing 3 changed files with 2,254 additions and 0 deletions.
diff --git a/odl/space/__init__.py b/odl/space/__init__.py
@@ -19,6 +19,9 @@
 from .npy_tensors import *
 __all__ += npy_tensors.__all__
 
+from .cupy_tensors import *
+__all__ += cupy_tensors.__all__
+
 from .pspace import *
 __all__ += pspace.__all__