Skip to content

GitLab

"...git@developer.sourcefind.cn:kecinstone/2024-pra-vllm.git" did not exist on "8a25d3a71ac8b0b06c37935ed67e7c35aa901bf5"
Commit dc86f372 authored by wenjh's avatar wenjh
Browse files

Add nmz support 


Signed-off-by: wenjh's avatarwenjh <wenjh@sugon.com>
parent 08be824c
Show 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 ...@@ -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 @ dc86f372
...@@ -125,7 +125,7 @@ else() ...@@ -125,7 +125,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