Skip to content

GitLab

"tests/vscode:/vscode.git/clone" did not exist on "e41f06702cb6d6787ac4474832264108a6e28780"
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 ...@@ -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 @ 847f4654
...@@ -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