summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorWillem Jan Palenstijn <Willem.Jan.Palenstijn@cwi.nl>2015-12-02 17:48:02 +0100
committerWillem Jan Palenstijn <Willem.Jan.Palenstijn@cwi.nl>2015-12-03 14:06:28 +0100
commitc335c53178cf63374599682dfbd7e08d318a20f2 (patch)
tree51a7dbcea4f383fae7dec328e39a14a07b8ae2ea
parentfd65fb03397ded59cfb872eb361e7d2e154c3335 (diff)
downloadastra-c335c53178cf63374599682dfbd7e08d318a20f2.tar.gz
astra-c335c53178cf63374599682dfbd7e08d318a20f2.tar.bz2
astra-c335c53178cf63374599682dfbd7e08d318a20f2.tar.xz
astra-c335c53178cf63374599682dfbd7e08d318a20f2.zip
Detect ASTRA_CUDA in Windows CL flags for Python as well
-rw-r--r--python/builder.py6
1 files changed, 6 insertions, 0 deletions
diff --git a/python/builder.py b/python/builder.py
index cfdb7d1..44d9c3b 100644
--- a/python/builder.py
+++ b/python/builder.py
@@ -41,6 +41,12 @@ try:
usecuda=True
except KeyError:
pass
+try:
+ if os.environ['CL'].find('/DASTRA_CUDA')!=-1:
+ usecuda=True
+except KeyError:
+ pass
+
cfgToWrite = 'DEF HAVE_CUDA=' + str(usecuda) + "\n"
cfgHasToBeUpdated = True