Skip to content

GitLab

Commit 47d6a78f authored by wenjh's avatar wenjh
Browse files

[DTK-25.04.1] Add support of __shfl_*sync apis

parent 521f8d3b
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
transformer_engine/common/CMakeLists.txt
View file @ 47d6a78f
......@@ -73,6 +73,7 @@ else()
# Disable Asserts In Code (Can't use asserts on HIP stack.)
add_definitions(-DNDEBUG)
add_definitions(-DUSE_ROCM)
add_definitions(-DHIP_ENABLE_WARP_SYNC_BUILTINS)
# Change clang++ to hipcc
SET(CMAKE_CXX_COMPILER "${ROCM_PATH}/bin/hipcc")
......
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