Skip to content

GitLab

Commit dc86f372 authored by wenjh's avatar wenjh
Browse files

Add nmz support 


Signed-off-by: wenjh's avatarwenjh <wenjh@sugon.com>
parent 08be824c
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
tests/cpp/CMakeLists.txt
View file @ dc86f372
......@@ -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 @ dc86f372
......@@ -125,7 +125,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