a719e9e1bc
* nongnu/packages/nvidia.scm (cuda-python, nvidia-cutlass): Add variables. * nongnu/packages/patches: Add necessary patches.
74 lines
2.4 KiB
Diff
74 lines
2.4 KiB
Diff
From 7ee9ec4c2636cca833761d3466df27edc4e3f952 Mon Sep 17 00:00:00 2001
|
|
From: Nicolas Graves <ngraves@ngraves.fr>
|
|
Date: Tue, 30 Jul 2024 14:13:09 +0200
|
|
Subject: [PATCH] Add CUTLASS_BUILD_STATIC_LIBRARY option
|
|
|
|
---
|
|
tools/library/CMakeLists.txt | 22 ++++++++++++++++++----
|
|
1 file changed, 18 insertions(+), 4 deletions(-)
|
|
|
|
diff --git a/tools/library/CMakeLists.txt b/tools/library/CMakeLists.txt
|
|
index a11ebcf6..79f7ccd1 100644
|
|
--- a/tools/library/CMakeLists.txt
|
|
+++ b/tools/library/CMakeLists.txt
|
|
@@ -34,6 +34,7 @@ include(GNUInstallDirs)
|
|
|
|
set(CUTLASS_BUILD_MONO_LIBRARY OFF CACHE BOOL
|
|
"Determines whether the cutlass library is generated as a single file or multiple files.")
|
|
+option(CUTLASS_BUILD_STATIC_LIBRARY "Build static libary for CUTLASS" ON)
|
|
|
|
################################################################################
|
|
|
|
@@ -126,7 +127,9 @@ function(cutlass_add_cutlass_library)
|
|
# simply link the generated object files to the default library.
|
|
|
|
target_link_libraries(${DEFAULT_NAME} PRIVATE $<BUILD_INTERFACE:${__NAME}_objs>)
|
|
- target_link_libraries(${DEFAULT_NAME}_static PRIVATE $<BUILD_INTERFACE:${__NAME}_objs>)
|
|
+ if (CUTLASS_BUILD_STATIC_LIBRARY)
|
|
+ target_link_libraries(${DEFAULT_NAME}_static PRIVATE $<BUILD_INTERFACE:${__NAME}_objs>)
|
|
+ endif()
|
|
|
|
else()
|
|
|
|
@@ -152,7 +155,7 @@ function(cutlass_add_cutlass_library)
|
|
)
|
|
|
|
set_target_properties(${__NAME} PROPERTIES DEBUG_POSTFIX "${CUTLASS_LIBRARY_DEBUG_POSTFIX}")
|
|
-
|
|
+ if (CUTLASS_BUILD_STATIC_LIBRARY)
|
|
cutlass_add_library(
|
|
${__NAME}_static
|
|
STATIC
|
|
@@ -189,6 +192,15 @@ function(cutlass_add_cutlass_library)
|
|
LIBRARY DESTINATION ${CMAKE_INSTALL_LIBDIR}
|
|
ARCHIVE DESTINATION ${CMAKE_INSTALL_LIBDIR}
|
|
)
|
|
+ else()
|
|
+ install(
|
|
+ TARGETS ${__NAME}
|
|
+ EXPORT NvidiaCutlass
|
|
+ RUNTIME DESTINATION ${CMAKE_INSTALL_BINDIR}
|
|
+ LIBRARY DESTINATION ${CMAKE_INSTALL_LIBDIR}
|
|
+ ARCHIVE DESTINATION ${CMAKE_INSTALL_LIBDIR}
|
|
+ )
|
|
+ endif()
|
|
|
|
if (__SUFFIX)
|
|
|
|
@@ -197,8 +209,10 @@ function(cutlass_add_cutlass_library)
|
|
# commands to pull in all kernels by default.
|
|
|
|
target_link_libraries(${DEFAULT_NAME} INTERFACE ${__NAME})
|
|
- target_link_libraries(${DEFAULT_NAME}_static INTERFACE ${__NAME}_static)
|
|
-
|
|
+ if (CUTLASS_BUILD_STATIC_LIBRARY)
|
|
+ target_link_libraries(${DEFAULT_NAME}_static INTERFACE ${__NAME}_static)
|
|
+ endif()
|
|
+
|
|
endif()
|
|
|
|
endif()
|
|
--
|
|
2.45.2
|
|
|