Commit 129d7526 authored by yuguo's avatar yuguo
Browse files

Merge branch 'develop_v2.3' of...

Merge branch 'develop_v2.3' of http://10.16.6.30/dcutoolkit/deeplearing/TransformerEngine into develop_v2.3
parents f6937668 7e4e1e40
...@@ -73,6 +73,7 @@ else() ...@@ -73,6 +73,7 @@ else()
# Disable Asserts In Code (Can't use asserts on HIP stack.) # Disable Asserts In Code (Can't use asserts on HIP stack.)
add_definitions(-DNDEBUG) add_definitions(-DNDEBUG)
add_definitions(-DUSE_ROCM) add_definitions(-DUSE_ROCM)
add_definitions(-DHIP_ENABLE_WARP_SYNC_BUILTINS)
# Change clang++ to hipcc # Change clang++ to hipcc
SET(CMAKE_CXX_COMPILER "${ROCM_PATH}/bin/hipcc") SET(CMAKE_CXX_COMPILER "${ROCM_PATH}/bin/hipcc")
......
...@@ -13,6 +13,7 @@ ...@@ -13,6 +13,7 @@
#ifndef __HIPCC_RTC__ #ifndef __HIPCC_RTC__
#include <cstdint> #include <cstdint>
#else #else
#include <hip/amd_detail/hip_assert.h>
using namespace __hip_internal; using namespace __hip_internal;
#endif #endif
#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