Merge pull request deepmodeling#354 from denghuilu/master

update compiler support for Ampere architecture devices
HPC-AI-Team · Feb 4, 2021 · a814a61 · a814a61
2 parents a7e3bb3 + 455b441
commit a814a61
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/source/op/cuda/CMakeLists.txt b/source/op/cuda/CMakeLists.txt
@@ -28,6 +28,8 @@ if (${CUDA_VERSION_MAJOR} GREATER "10")
                         -gencode arch=compute_61,code=sm_61; # Pascal - GTX 1080, GTX 1070, GTX 1060, GTX 1050, GTX 1030, Titan Xp, Tesla P40, Tesla P4, Discrete GPU on the NVIDIA Drive PX2
                         -gencode arch=compute_70,code=sm_70; # Volta  - GV100/Tesla V100, GTX 1180 (GV104)
                         -gencode arch=compute_75,code=sm_75; # Turing - RTX 2080, Titan RTX, Quadro R8000
+                        -gencode arch=compute_80,code=sm_80; # Anpere - A100
+                        -gencode arch=compute_86,code=sm_86; # Anpere - RTX 3090
                         -O3; -Xcompiler -fPIC;
         )
 elseif (${CUDA_VERSION_MAJOR} STREQUAL "10")