Skip to content

GitLab

Commit 688c7ab9 authored by yuguo's avatar yuguo
Browse files

Merge branch 'develop_v2.8' of... 

Merge branch 'develop_v2.8' of http://10.16.6.30/dcutoolkit/deeplearing/TransformerEngine into release_v2.8
parents d685b559 aa62d24c
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
transformer_engine/common/gemm/cublaslt_gemm.cu
View file @ 688c7ab9
...@@ -29,7 +29,9 @@ ...@@ -29,7 +29,9 @@
#include "../util/logging.h" #include "../util/logging.h"
#include "../util/multi_stream.h" #include "../util/multi_stream.h"
#include "./config.h" #include "./config.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