[CUDA] Add Ada architecture to DGL architecture list. (#7282)

e9661d3b · Muhammed Fatih BALIN · GitHub · 7b470eb0 · e9661d3b
Unverified Commit e9661d3b authored Apr 08, 2024 by Muhammed Fatih BALIN Committed by GitHub Apr 08, 2024
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

cmake/modules/CUDA.cmake cmake/modules/CUDA.cmake +1 -1

No files found.
--- a/cmake/modules/CUDA.cmake
+++ b/cmake/modules/CUDA.cmake
@@ -17,7 +17,7 @@ if (CUDA_VERSION_MAJOR GREATER_EQUAL "11")
  set(dgl_cuda_arch_ptx "80" "86")
 endif()
 if (CUDA_VERSION VERSION_GREATER_EQUAL "11.8")
-  list(APPEND dgl_known_gpu_archs "90")
+  list(APPEND dgl_known_gpu_archs "89" "90")
  set(dgl_cuda_arch_ptx "90")
 endif()
 if (CUDA_VERSION VERSION_GREATER_EQUAL "12.0")