Skip to content

GitLab

Unverified Commit 311f8e0c authored by Daniel Hiltgen's avatar Daniel Hiltgen Committed by GitHub
Browse files

Merge pull request #2243 from dhiltgen/harden_zero_gpus 

Harden for zero detected GPUs
parents e02ecfb6 f07f8b7a
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
gpu/gpu.go
View file @ 311f8e0c
...@@ -135,7 +135,7 @@ func GetGPUInfo() GpuInfo { ...@@ -135,7 +135,7 @@ func GetGPUInfo() GpuInfo {
if memInfo.err != nil { if memInfo.err != nil {
slog.Info(fmt.Sprintf("error looking up CUDA GPU memory: %s", C.GoString(memInfo.err))) slog.Info(fmt.Sprintf("error looking up CUDA GPU memory: %s", C.GoString(memInfo.err)))
C.free(unsafe.Pointer(memInfo.err)) C.free(unsafe.Pointer(memInfo.err))
} else { } else if memInfo.count > 0 {
// Verify minimum compute capability // Verify minimum compute capability
var cc C.cuda_compute_capability_t var cc C.cuda_compute_capability_t
C.cuda_compute_capability(*gpuHandles.cuda, &cc) C.cuda_compute_capability(*gpuHandles.cuda, &cc)
...@@ -157,7 +157,7 @@ func GetGPUInfo() GpuInfo { ...@@ -157,7 +157,7 @@ func GetGPUInfo() GpuInfo {
} else if memInfo.igpu_index >= 0 && memInfo.count == 1 { } else if memInfo.igpu_index >= 0 && memInfo.count == 1 {
// Only one GPU detected and it appears to be an integrated GPU - skip it // Only one GPU detected and it appears to be an integrated GPU - skip it
slog.Info("ROCm unsupported integrated GPU detected") slog.Info("ROCm unsupported integrated GPU detected")
} else { } else if memInfo.count > 0 {
if memInfo.igpu_index >= 0 { if memInfo.igpu_index >= 0 {
// We have multiple GPUs reported, and one of them is an integrated GPU // We have multiple GPUs reported, and one of them is an integrated GPU
// so we have to set the env var to bypass it // so we have to set the env var to bypass it
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment