diff options
| author | Willem Jan Palenstijn <Willem.Jan.Palenstijn@cwi.nl> | 2017-11-09 20:57:31 +0100 |
|---|---|---|
| committer | Willem Jan Palenstijn <Willem.Jan.Palenstijn@cwi.nl> | 2018-01-09 14:23:54 +0100 |
| commit | 0ad6cb01b9176e99c3e4bb5643ac6a5dd477aad6 (patch) | |
| tree | 69b27fa6406cf318d785c827e1fd05b786da2876 /python | |
| parent | de27e439a0c59fade175fba4e0b4a1e0c84b933d (diff) | |
| download | astra-0ad6cb01b9176e99c3e4bb5643ac6a5dd477aad6.tar.gz astra-0ad6cb01b9176e99c3e4bb5643ac6a5dd477aad6.tar.bz2 astra-0ad6cb01b9176e99c3e4bb5643ac6a5dd477aad6.tar.xz astra-0ad6cb01b9176e99c3e4bb5643ac6a5dd477aad6.zip | |
Check for CUDA support at run-time in use_cuda()
Diffstat (limited to 'python')
| -rw-r--r-- | python/astra/astra_c.pyx | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/python/astra/astra_c.pyx b/python/astra/astra_c.pyx index f25fc2a..8858244 100644 --- a/python/astra/astra_c.pyx +++ b/python/astra/astra_c.pyx @@ -38,6 +38,7 @@ from .PyIndexManager cimport CAstraObjectManagerBase cdef extern from "astra/Globals.h" namespace "astra": bool cudaEnabled() + bool cudaAvailable() IF HAVE_CUDA==True: cdef extern from "../cuda/2d/astra.h" namespace "astraCUDA": @@ -74,7 +75,7 @@ def credits(): def use_cuda(): - return cudaEnabled() + return cudaAvailable() IF HAVE_CUDA==True: def set_gpu_index(idx, memory=0): |