summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorWillem Jan Palenstijn <Willem.Jan.Palenstijn@cwi.nl>2018-09-25 14:00:07 +0200
committerWillem Jan Palenstijn <Willem.Jan.Palenstijn@cwi.nl>2018-09-25 18:34:07 +0200
commit6730face204f40c57c3e8fe58ea9749494d669a6 (patch)
tree6ffeb1e86d92c43c91e296da597a269fad28b38f
parentc3137acbebd3159f7dee7d6817e47188991510d4 (diff)
downloadastra-6730face204f40c57c3e8fe58ea9749494d669a6.tar.gz
astra-6730face204f40c57c3e8fe58ea9749494d669a6.tar.bz2
astra-6730face204f40c57c3e8fe58ea9749494d669a6.tar.xz
astra-6730face204f40c57c3e8fe58ea9749494d669a6.zip
Raise exception when setting GPU memory unrealistically small
-rw-r--r--python/astra/astra_c.pyx2
1 files changed, 2 insertions, 0 deletions
diff --git a/python/astra/astra_c.pyx b/python/astra/astra_c.pyx
index 1d3fad7..65dbf28 100644
--- a/python/astra/astra_c.pyx
+++ b/python/astra/astra_c.pyx
@@ -84,6 +84,8 @@ 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:
+ raise ValueError("Setting GPU memory lower than 1MB is not supported.")
params.memory = memory
params.GPUIndices = idx
setGlobalGPUParams(params)