From 7b268a1f78ef6c0ec4445093dd38f1cb3474cc9e Mon Sep 17 00:00:00 2001
From: Willem Jan Palenstijn <Willem.Jan.Palenstijn@cwi.nl>
Date: Wed, 7 Nov 2018 18:13:33 +0100
Subject: Fix python set_gpu_index default memory argument

---
 python/astra/astra_c.pyx | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'python/astra/astra_c.pyx')

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
-- 
cgit v1.2.3