Skip to content

GitLab

"...AutoBuildImmortalWrt.git" did not exist on "606f0c6c5dfd89920202f2501d58ac686ee73c0f"
Commit aa62d24c authored by yuguo's avatar yuguo
Browse files

[DCU] fix compile issues

parent 8d5cd8c6
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
transformer_engine/common/CMakeLists.txt
View file @ aa62d24c
...@@ -206,6 +206,7 @@ else() ...@@ -206,6 +206,7 @@ else()
dropout/dropout.cu dropout/dropout.cu
activation/relu.cu activation/relu.cu
activation/swiglu.cu activation/swiglu.cu
gemm/cublaslt_gemm.cu
gemm/hipblas_gemm.cu gemm/hipblas_gemm.cu
normalization/common.cpp normalization/common.cpp
normalization/layernorm/ln_api.cpp normalization/layernorm/ln_api.cpp
......
transformer_engine/common/gemm/cublaslt_gemm.cu
View file @ aa62d24c
...@@ -25,7 +25,9 @@ ...@@ -25,7 +25,9 @@
#include "../util/logging.h" #include "../util/logging.h"
#include "../util/multi_stream.h" #include "../util/multi_stream.h"
#include "common/util/cuda_runtime.h" #include "common/util/cuda_runtime.h"
#ifndef __HIP_PLATFORM_AMD__
#include "cutlass_grouped_gemm.cuh" #include "cutlass_grouped_gemm.cuh"
#endif
#ifndef __HIP_PLATFORM_AMD__ #ifndef __HIP_PLATFORM_AMD__
namespace { namespace {
......
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