Jelajahi Sumber

cuda : fix UMA detection on discrete GPUs. (#17537)

matt23654 1 bulan lalu
induk
melakukan
909072abcf
1 mengubah file dengan 1 tambahan dan 1 penghapusan
  1. 1 1
      ggml/src/ggml-cuda/ggml-cuda.cu

+ 1 - 1
ggml/src/ggml-cuda/ggml-cuda.cu

@@ -3837,7 +3837,7 @@ static void ggml_backend_cuda_device_get_memory(ggml_backend_dev_t dev, size_t *
 
     // Check if UMA is explicitly enabled via environment variable
     bool uma_env = getenv("GGML_CUDA_ENABLE_UNIFIED_MEMORY") != nullptr;
-    bool is_uma = prop.unifiedAddressing > 0 || uma_env;
+    bool is_uma = prop.integrated > 0 || uma_env;
 
     if (is_uma) {
         // For UMA systems (like DGX Spark), use system memory info