70-real;80-real;86-real;90-real

Oneflow-Inc · Sep 1, 2023 · 32c6dc7 · 32c6dc7
1 parent a90a300
commit 32c6dc7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cmake/caches/ci/release/cuda-86.cmake b/cmake/caches/ci/release/cuda-86.cmake
@@ -7,7 +7,7 @@ set(THIRD_PARTY_MIRROR aliyun CACHE STRING "")
 set(PIP_INDEX_MIRROR "https://pypi.tuna.tsinghua.edu.cn/simple" CACHE STRING "")
 set(CMAKE_BUILD_TYPE Release CACHE STRING "")
 set(CMAKE_GENERATOR Ninja CACHE STRING "")
-set(CMAKE_CUDA_ARCHITECTURES "75-real;80-real;86-real" CACHE STRING "")
+set(CMAKE_CUDA_ARCHITECTURES "70-real;80-real;86-real;90-real" CACHE STRING "")
 set(CUDNN_STATIC OFF CACHE BOOL "")
 set(WITH_MLIR ON CACHE BOOL "")
 set(BUILD_CPP_API OFF CACHE BOOL "")