Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "c27acebba3b3c7d94209e0467b0a801db4af73ed"
Commit 847f4654 authored by wenjh's avatar wenjh
Browse files

Add nmz support 


Signed-off-by: wenjh's avatarwenjh <wenjh@sugon.com>
parent 0b03f913
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
tests/cpp/CMakeLists.txt
View file @ 847f4654
......@@ -19,7 +19,7 @@ if(((EXISTS "/opt/dtk/") OR (EXISTS $ENV{ROCM_PATH})) AND NOT (EXISTS "/bin/nvcc
add_definitions(-DNDEBUG)
add_definitions(-DUSE_ROCM)
if(NOT DEFINED ENV{NVTE_ROCM_ARCH})
SET(CMAKE_HIP_ARCHITECTURES gfx906;gfx926;gfx928;gfx936)
SET(CMAKE_HIP_ARCHITECTURES gfx906;gfx926;gfx928;gfx936;gfx938)
else()
SET(CMAKE_HIP_ARCHITECTURES $ENV{NVTE_ROCM_ARCH})
endif()
......
transformer_engine/common/CMakeLists.txt
View file @ 847f4654
......@@ -80,7 +80,7 @@ else()
SET(CMAKE_CXX_COMPILER "${ROCM_PATH}/bin/hipcc")
if(NOT DEFINED ENV{NVTE_ROCM_ARCH})
SET(CMAKE_HIP_ARCHITECTURES gfx906;gfx926;gfx928;gfx936)
SET(CMAKE_HIP_ARCHITECTURES gfx906;gfx926;gfx928;gfx936;gfx938)
else()
SET(CMAKE_HIP_ARCHITECTURES $ENV{NVTE_ROCM_ARCH})
endif()
......
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