Commit 05fadb06 authored by sangwz's avatar sangwz
Browse files

Update CMakeLists.txt, add gfx938 target

parent 187b354a
...@@ -158,7 +158,7 @@ if(USE_HIP) ...@@ -158,7 +158,7 @@ if(USE_HIP)
project(dgl C CXX HIP) project(dgl C CXX HIP)
include(cmake/modules/ROCM.cmake) include(cmake/modules/ROCM.cmake)
# target_compile_features(dgl PRIVATE cxx_std_17) # target_compile_features(dgl PRIVATE cxx_std_17)
set(CMAKE_HIP_ARCHITECTURES gfx906;gfx928;gfx926;gfx936) set(CMAKE_HIP_ARCHITECTURES gfx906;gfx928;gfx926;gfx936;gfx938)
# see https://github.com/NVIDIA/thrust/issues/1401 # see https://github.com/NVIDIA/thrust/issues/1401
if(NOT DEFINED ENV{ROCM_PATH}) if(NOT DEFINED ENV{ROCM_PATH})
set(ROCM_PATH "/opt/dtk" CACHE PATH "Path to which RoCm has been installed") set(ROCM_PATH "/opt/dtk" CACHE PATH "Path to which RoCm has been installed")
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment