Преглед на файлове

build: add -DGGML_CUDA_NO_PEER_COPY=ON for rocm builds on windows (#9060)

Jeffrey Morgan преди 2 месеца
родител
ревизия
a4f69a0191
променени са 1 файла, в които са добавени 4 реда и са изтрити 0 реда
  1. 4 0
      CMakeLists.txt

+ 4 - 0
CMakeLists.txt

@@ -104,6 +104,10 @@ if(CMAKE_HIP_COMPILER)
     if(AMDGPU_TARGETS)
         add_subdirectory(${CMAKE_CURRENT_SOURCE_DIR}/ml/backend/ggml/ggml/src/ggml-hip)
 
+        if (WIN32)
+            target_compile_definitions(ggml-hip PRIVATE GGML_CUDA_NO_PEER_COPY=1)
+        endif()
+
         set(OLLAMA_HIP_INSTALL_DIR ${OLLAMA_INSTALL_DIR}/rocm)
         install(TARGETS ggml-hip
             RUNTIME_DEPENDENCIES