[C] Suppress 128-D warning from cudnn-frontend (#1158)

suppress 128D warning from cudnn-frontend Signed-off-by: Charlene Yang <[email protected]>
NVIDIA · Sep 5, 2024 · 206c1d9 · 206c1d9
1 parent 215db88
commit 206c1d9
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/transformer_engine/common/CMakeLists.txt b/transformer_engine/common/CMakeLists.txt
@@ -135,6 +135,7 @@ set_source_files_properties(fused_softmax/scaled_masked_softmax.cu
                             COMPILE_OPTIONS "--use_fast_math")
 set(CMAKE_CUDA_FLAGS "${CMAKE_CUDA_FLAGS} --expt-relaxed-constexpr")
 set(CMAKE_CUDA_FLAGS "${CMAKE_CUDA_FLAGS} -O3")
+set(CMAKE_CUDA_FLAGS "${CMAKE_CUDA_FLAGS} -diag-suppress 128")
 
 # Number of parallel build jobs
 if(ENV{MAX_JOBS})