revert library changes

Signed-off-by: youkaichao <[email protected]>
vllm-project · Jan 24, 2025 · 3771aca · 3771aca
1 parent caefcff
commit 3771aca
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cmake/utils.cmake b/cmake/utils.cmake
@@ -431,7 +431,7 @@ function (define_gpu_extension_target GPU_MOD_NAME)
   # Don't use `TORCH_LIBRARIES` for CUDA since it pulls in a bunch of
   # dependencies that are not necessary and may not be installed.
   if (GPU_LANGUAGE STREQUAL "CUDA")
-    target_link_libraries(${GPU_MOD_NAME} PRIVATE CUDA::cudart CUDA::cuda_driver torch_cpu torch_cuda c10_cuda c10)
+    target_link_libraries(${GPU_MOD_NAME} PRIVATE CUDA::cudart CUDA::cuda_driver)
   else()
     target_link_libraries(${GPU_MOD_NAME} PRIVATE ${TORCH_LIBRARIES})
   endif()