gpu: validate if index exists by checking length

vinegarhq · Oct 16, 2023 · 8cddff0 · 8cddff0
1 parent 9d20c2c
commit 8cddff0
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/gpu/pick.go b/gpu/pick.go
@@ -23,12 +23,12 @@ func HandleGpu(target target.TargetGPU, env map[string]string, isVulkan bool) (m
 		}
 	}
 
-	gpu := gpus[target.Id]
-
-	if gpu == nil {
+	if len(gpus) < target.Id+1 {
 		return env, errors.New("gpu not found")
 	}
 
+	gpu := gpus[target.Id]
+
 	env = gpu.ApplyEnv(env)
 
 	return env, nil