summaryrefslogtreecommitdiffstats
path: root/python/astra/astra.py
diff options
context:
space:
mode:
authorWillem Jan Palenstijn <Willem.Jan.Palenstijn@cwi.nl>2016-01-06 16:40:26 +0100
committerWillem Jan Palenstijn <Willem.Jan.Palenstijn@cwi.nl>2016-01-19 16:50:17 +0100
commit76d182ace088b7ab6d64019079c049a0f9a29e69 (patch)
tree6056c1f266408eb13f0e43e1ac59c60b0887ad12 /python/astra/astra.py
parent687c5e244e46e51786afad77f5015cae9abad129 (diff)
downloadastra-76d182ace088b7ab6d64019079c049a0f9a29e69.tar.gz
astra-76d182ace088b7ab6d64019079c049a0f9a29e69.tar.bz2
astra-76d182ace088b7ab6d64019079c049a0f9a29e69.tar.xz
astra-76d182ace088b7ab6d64019079c049a0f9a29e69.zip
Change python set_gpu_index to match
Diffstat (limited to 'python/astra/astra.py')
-rw-r--r--python/astra/astra.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/python/astra/astra.py b/python/astra/astra.py
index 26b1ff0..9328b6b 100644
--- a/python/astra/astra.py
+++ b/python/astra/astra.py
@@ -49,10 +49,10 @@ def version(printToScreen=False):
"""
return a.version(printToScreen)
-def set_gpu_index(idx):
+def set_gpu_index(idx, memory=0):
"""Set default GPU index to use.
:param idx: GPU index
:type idx: :class:`int`
"""
- a.set_gpu_index(idx)
+ a.set_gpu_index(idx, memory)