From: =?utf-8?q?Picca_Fr=C3=A9d=C3=A9ric-Emmanuel?= Date: Fri, 29 Jan 2021 15:45:25 +0100 Subject: support cuda 11 --- setup.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/setup.py b/setup.py index 84b436e..8047d51 100644 --- a/setup.py +++ b/setup.py @@ -406,13 +406,14 @@ def do_pyhst(): CUDA={'include':[]} CUDA .update(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["arch"] = [ "-gencode", "arch=compute_50,code=compute_50" , + "-gencode", "arch=compute_60,code=compute_60" , + "-gencode", "arch=compute_70,code=compute_70" , + "-gencode", "arch=compute_80,code=compute_80" , ] # print( CUDA) - + CUDA["Xptxas"] = [" ", " -Xptxas=-v"] def build_Cspace():