summaryrefslogtreecommitdiffstats
path: root/python/astra/astra.py
diff options
context:
space:
mode:
authorWillem Jan Palenstijn <wjp@usecode.org>2016-02-02 14:25:01 +0100
committerWillem Jan Palenstijn <wjp@usecode.org>2016-02-02 14:25:01 +0100
commit278618e2f70f6b1308bf9c0aa79f76137cc582c1 (patch)
treec7f402b60e4baa529149a44bbe3fa2570ed610b2 /python/astra/astra.py
parentcd12edcd3ada212f27088a4e26e3eee5f89bcbad (diff)
parentb529ff854f0e1191108e31d6be294d31b50c666e (diff)
downloadastra-278618e2f70f6b1308bf9c0aa79f76137cc582c1.tar.gz
astra-278618e2f70f6b1308bf9c0aa79f76137cc582c1.tar.bz2
astra-278618e2f70f6b1308bf9c0aa79f76137cc582c1.tar.xz
astra-278618e2f70f6b1308bf9c0aa79f76137cc582c1.zip
Merge pull request #107 from wjp/multi
Add multi-GPU support to CompositeGeometryManager
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)