From 294b1796885c07b579a3d6fcf5e1b6060d2fdea4 Mon Sep 17 00:00:00 2001 From: Inforithmics Date: Sat, 4 Oct 2025 15:20:33 +0200 Subject: [PATCH] merge fixes --- discover/runner.go | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/discover/runner.go b/discover/runner.go index 15fac2f17..d173959ad 100644 --- a/discover/runner.go +++ b/discover/runner.go @@ -123,13 +123,15 @@ func GPUDevices(ctx context.Context, runners []FilteredRunnerDiscovery) []ml.Dev go func(i int) { defer wg.Done() var envVar string + if devices[i].Library == "ROCm" { if runtime.GOOS != "linux" { envVar = "HIP_VISIBLE_DEVICES" } else { envVar = "ROCR_VISIBLE_DEVICES" } + } else if devices[i].Library == "CUDA" { envVar = "CUDA_VISIBLE_DEVICES" - } else if devices[i].Library == "VULKAN" { + } else if devices[i].Library == "Vulkan" { envVar = "GGML_VK_VISIBLE_DEVICES" }