Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
TransformerEngine
Commits
5be81251
Unverified
Commit
5be81251
authored
Oct 03, 2025
by
Tim Moon
Committed by
GitHub
Oct 03, 2025
Browse files
Fix bug where CUTLASS kernel was not being compiled for SM90a (#2235)
Signed-off-by:
Tim Moon
<
tmoon@nvidia.com
>
parent
dfe5b7df
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
6 additions
and
12 deletions
+6
-12
transformer_engine/common/CMakeLists.txt
transformer_engine/common/CMakeLists.txt
+6
-12
No files found.
transformer_engine/common/CMakeLists.txt
View file @
5be81251
...
@@ -155,18 +155,12 @@ add_library(transformer_engine SHARED ${transformer_engine_SOURCES})
...
@@ -155,18 +155,12 @@ add_library(transformer_engine SHARED ${transformer_engine_SOURCES})
target_include_directories
(
transformer_engine PUBLIC
target_include_directories
(
transformer_engine PUBLIC
"
${
CMAKE_CURRENT_SOURCE_DIR
}
/include"
)
"
${
CMAKE_CURRENT_SOURCE_DIR
}
/include"
)
if
(
${
CMAKE_CUDA_COMPILER_VERSION
}
VERSION_GREATER 12.0
)
# CUTLASS kernels require SM90a and cause hang in debug build
set_source_files_properties
(
set_property
(
"gemm/cutlass_grouped_gemm.cu"
SOURCE gemm/cutlass_grouped_gemm.cu
PROPERTIES
APPEND
COMPILE_FLAGS
PROPERTY
"-gencode arch=compute_90a,code=sm_90a"
)
COMPILE_OPTIONS
"--generate-code=arch=compute_90a,code=sm_90a;-g0"
)
else
()
message
(
FATAL_ERROR
"cutlass gemm/cutlass_grouped_gemm.cu kernel required sm 90a"
)
endif
()
# Disable debug build for cutlass due to hang.
set_source_files_properties
(
"gemm/cutlass_grouped_gemm.cu"
PROPERTIES COMPILE_FLAGS
"-g0"
)
# Configure dependencies
# Configure dependencies
target_link_libraries
(
transformer_engine PUBLIC
target_link_libraries
(
transformer_engine PUBLIC
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment