Skip to content

GitLab

"...SevenNet/git@developer.sourcefind.cn:nivren/ict-csp.git" did not exist on "2189e8bbf46c5dfd6cae004fff6929c351d4ae85"
Commit e908bd30 authored by wenjh's avatar wenjh
Browse files

Add nmz support 


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