support A100 GPU

2022-06-27 21:20:56 +08:00 · 2022-06-27 21:20:56 +08:00 · 395fccbc23
parent 0691e1b871
commit 395fccbc23
1 changed files with 1 additions and 0 deletions
--- a/setup.py
+++ b/setup.py
@ -21,6 +21,7 @@ def make_cuda_ext(
            "-D__CUDA_NO_HALF2_OPERATORS__",
            "-gencode=arch=compute_70,code=sm_70",
            "-gencode=arch=compute_75,code=sm_75",
+            "-gencode=arch=compute_80,code=sm_80",
            "-gencode=arch=compute_86,code=sm_86",
        ]
        sources += sources_cuda