Skip to content

Commit 909072a

Browse files
authored
cuda : fix UMA detection on discrete GPUs. (#17537)
1 parent cd8370b commit 909072a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

ggml/src/ggml-cuda/ggml-cuda.cu

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3837,7 +3837,7 @@ static void ggml_backend_cuda_device_get_memory(ggml_backend_dev_t dev, size_t *
38373837

38383838
// Check if UMA is explicitly enabled via environment variable
38393839
bool uma_env = getenv("GGML_CUDA_ENABLE_UNIFIED_MEMORY") != nullptr;
3840-
bool is_uma = prop.unifiedAddressing > 0 || uma_env;
3840+
bool is_uma = prop.integrated > 0 || uma_env;
38413841

38423842
if (is_uma) {
38433843
// For UMA systems (like DGX Spark), use system memory info

0 commit comments

Comments
 (0)