diff options
author | Willem Jan Palenstijn <Willem.Jan.Palenstijn@cwi.nl> | 2018-11-07 18:13:33 +0100 |
---|---|---|
committer | Willem Jan Palenstijn <Willem.Jan.Palenstijn@cwi.nl> | 2018-11-07 18:13:33 +0100 |
commit | 7b268a1f78ef6c0ec4445093dd38f1cb3474cc9e (patch) | |
tree | 08fd935706d5c462db09500e513cee2e89019bf7 /python | |
parent | 2ef3cc3d0a2d750fa2992321535e53c77d67a3b0 (diff) | |
download | astra-7b268a1f78ef6c0ec4445093dd38f1cb3474cc9e.tar.gz astra-7b268a1f78ef6c0ec4445093dd38f1cb3474cc9e.tar.bz2 astra-7b268a1f78ef6c0ec4445093dd38f1cb3474cc9e.tar.xz astra-7b268a1f78ef6c0ec4445093dd38f1cb3474cc9e.zip |
Fix python set_gpu_index default memory argument
Diffstat (limited to 'python')
-rw-r--r-- | python/astra/astra_c.pyx | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/python/astra/astra_c.pyx b/python/astra/astra_c.pyx index 69909b5..6289f3e 100644 --- a/python/astra/astra_c.pyx +++ b/python/astra/astra_c.pyx @@ -87,7 +87,7 @@ IF HAVE_CUDA==True: if use_cuda()==True: if not isinstance(idx, collections.Iterable) or isinstance(idx, six.string_types + (six.text_type,six.binary_type)): idx = (idx,) - if memory < 1024*1024: + if memory != 0 and memory < 1024*1024: raise ValueError("Setting GPU memory lower than 1MB is not supported.") params.memory = memory params.GPUIndices = idx |