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
tsoc
superbenchmark
Commits
3e090482
Unverified
Commit
3e090482
authored
Apr 30, 2025
by
pdr
Committed by
GitHub
May 01, 2025
Browse files
cuda arch flag for cublaslt (#701)
adding gb200 cuda arch flag for cublaslt compilation
parent
330c68aa
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
6 additions
and
1 deletion
+6
-1
superbench/benchmarks/micro_benchmarks/cublaslt_gemm/CMakeLists.txt
.../benchmarks/micro_benchmarks/cublaslt_gemm/CMakeLists.txt
+6
-1
No files found.
superbench/benchmarks/micro_benchmarks/cublaslt_gemm/CMakeLists.txt
View file @
3e090482
...
...
@@ -9,6 +9,10 @@ find_package(CUDAToolkit QUIET)
if
(
CUDAToolkit_FOUND AND NOT CUDAToolkit_VERSION VERSION_LESS 11.8
)
set
(
CMAKE_CUDA_STANDARD 17
)
include
(
../cuda_common.cmake
)
set
(
CUDA_ARCH_LIST
"80;86;90"
)
if
(
CUDAToolkit_VERSION VERSION_GREATER_EQUAL
"12.8.0"
)
list
(
APPEND CUDA_ARCH_LIST
"100"
)
endif
()
add_library
(
cublaslt_utils SHARED cublaslt_utils.cc
)
target_link_libraries
(
cublaslt_utils CUDA::cublas CUDA::cublasLt
)
...
...
@@ -17,6 +21,7 @@ if(CUDAToolkit_FOUND AND NOT CUDAToolkit_VERSION VERSION_LESS 11.8)
add_executable
(
cublaslt_gemm cublaslt_gemm.cu
)
target_link_libraries
(
cublaslt_gemm cublaslt_utils
)
set_target_properties
(
cublaslt_gemm PROPERTIES CUDA_ARCHITECTURES
"80;86;90"
)
set_target_properties
(
cublaslt_gemm PROPERTIES CUDA_ARCHITECTURES
"
${
CUDA_ARCH_LIST
}
"
)
install
(
TARGETS cublaslt_gemm RUNTIME DESTINATION bin
)
endif
()
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