fix cuda compule error (#52654)

1ad943dd · risemeup1 · GitHub · a7707efb · 1ad943dd
显示空白变更内容
内联并排

Showing with 2 addition and 1 deletion

cmake/cuda.cmake cmake/cuda.cmake +2 -1

未找到文件。
--- a/cmake/cuda.cmake
+++ b/cmake/cuda.cmake
@@ -58,7 +58,8 @@ function(detect_installed_gpus out_variable)
      "}\n")

    execute_process(
-      COMMAND "${CUDA_NVCC_EXECUTABLE}" "--run" "${cufile}"
+      COMMAND "${CUDA_NVCC_EXECUTABLE}" "-ccbin=${CMAKE_C_COMPILER}" "--run"
+              "${cufile}"
      WORKING_DIRECTORY "${PROJECT_BINARY_DIR}/CMakeFiles/"
      RESULT_VARIABLE nvcc_res
      OUTPUT_VARIABLE nvcc_out