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

This commit is contained in:
matt23654
2025-12-12 17:53:10 +02:00
committed by Georgi Gerganov
parent 93f6cdb9c0
commit e682af7886
+1 -1
View File
@@ -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