(vllm) Add 9999-build-debug.patch

sasha0552 · Aug 21, 2024 · 3ff196c · 3ff196c
1 parent a34e7d2
commit 3ff196c
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/patches/vllm_pascal/main/9999-build-debug.patch b/patches/vllm_pascal/main/9999-build-debug.patch
@@ -0,0 +1,15 @@
+--- a/CMakeLists.txt
++++ b/CMakeLists.txt
+@@ -242,6 +242,12 @@ if(VLLM_GPU_LANG STREQUAL "CUDA")
+   )
+
+   if (NOT machete_generation_result EQUAL 0)
++    execute_process(
++      COMMAND ${CMAKE_COMMAND} -E echo env 
++      PYTHONPATH=${CMAKE_CURRENT_SOURCE_DIR}/csrc/cutlass_extensions/:${CUTLASS_DIR}/python/:$PYTHONPATH 
++        ${Python_EXECUTABLE} ${CMAKE_CURRENT_SOURCE_DIR}/csrc/quantization/machete/generate.py
++    )
++    execute_process(COMMAND ${CMAKE_COMMAND} -E cat "${CMAKE_CURRENT_BINARY_DIR}/machete_generation.log")
+     message(FATAL_ERROR "Machete generation failed."
+                         " Result: \"${machete_generation_result}\"" 
+                         "\nCheck the log for details: "