update compiler support for Ampere architecture devices

denghuilu · denghuilu · commit fd660595a1a2 · 2021-02-04T11:05:32.000+08:00
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")

Original file line number	Diff line number	Diff line change
`@@ -28,6 +28,8 @@ if (${CUDA_VERSION_MAJOR} GREATER "10")`
`28`	`28`	`-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`
`29`	`29`	`-gencode arch=compute_70,code=sm_70; # Volta - GV100/Tesla V100, GTX 1180 (GV104)`
`30`	`30`	`-gencode arch=compute_75,code=sm_75; # Turing - RTX 2080, Titan RTX, Quadro R8000`
	`31`	`+ -gencode arch=compute_80,code=sm_80; # Anpere - A100`
	`32`	`+ -gencode arch=compute_86,code=sm_86; # Anpere - RTX 3090`
`31`	`33`	`-O3; -Xcompiler -fPIC;`
`32`	`34`	`)`
`33`	`35`	`elseif (${CUDA_VERSION_MAJOR} STREQUAL "10")`