[CI/Build][Bugfix] Fix CUTLASS header-only line (#7034)

6a11fdfb · Tyler Michael Smith · GitHub · 805a8a75 · 6a11fdfb
Unverified Commit 6a11fdfb authored Aug 01, 2024 by Tyler Michael Smith Committed by GitHub Aug 01, 2024
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 1 deletion

CMakeLists.txt CMakeLists.txt +4 -1

No files found.
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -156,12 +156,15 @@ set(VLLM_EXT_SRC

 if(VLLM_GPU_LANG STREQUAL "CUDA")
  include(FetchContent)
-  SET(CUTLASS_ENABLE_HEADERS_ONLY=ON)
+  SET(CUTLASS_ENABLE_HEADERS_ONLY ON CACHE BOOL "Enable only the header library")
  FetchContent_Declare(
        cutlass
        GIT_REPOSITORY https://github.com/nvidia/cutlass.git
        # CUTLASS 3.5.0
        GIT_TAG 7d49e6c7e2f8896c47f586706e67e1fb215529dc
+        # Shallow clone with depth 1
+        GIT_SHALLOW TRUE
+        GIT_PROGRESS TRUE
  )
  FetchContent_MakeAvailable(cutlass)