Commit 19ce86d5 authored by Alessandro Mirone's avatar Alessandro Mirone

small change to setup.py

parent 871452e9
......@@ -75,13 +75,13 @@ ONLYCUDA=0
DO_LINK=0
### CHANGE HERE
DOCUDA = 1
DOCUDA = 0
ONLYCUDA = 0
DO_LINK = 0
global version
global aumento_versione
aumento_versione=""
aumento_versione="b"
global version
......@@ -383,12 +383,16 @@ def do_pyhst():
cmdclass['build_ext'] = custom_build_ext
CUDA={'include':[]}
if DOCUDA:
if True or DOCUDA:
CUDA = locate_cuda()
CUDA["arch"] = ["-gencode", "arch=compute_20,code=compute_20",
"-gencode", "arch=compute_30,code=compute_30",
"-gencode", "arch=compute_50,code=compute_50"
]
CUDA['include']=[]
print CUDA
else:
CUDA={'include':[]}
......@@ -673,7 +677,7 @@ def do_pyhst():
platforms='any',
ext_modules=ext_modules,
cmdclass=cmdclass ,
scripts=script_files,
scripts=script_files
)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment