wuwenhai
/
bev-project
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #46 from zhiqi-li/a100_cuda 

support A100 GPU
This commit is contained in:
Zhijian Liu 2022-06-27 12:42:28 -04:00 committed by GitHub
parent 0691e1b871 395fccbc23
commit cb6cd789d1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
setup.py
View File

@ -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