summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-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