"...git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "06b01ea87e219c072e92db91dfda5f8f423b284a"
Unverified Commit fe3ec8db authored by Daniel Hiltgen's avatar Daniel Hiltgen Committed by GitHub
Browse files

Revert "Workaround broken NVIDIA iGPU free VRAM data (#12490)" (#12642)

The workaround has been moved into the underlying C++ code.

This reverts commit e4340667.
parent c7441342
...@@ -349,9 +349,6 @@ func GPUDevices(ctx context.Context, runners []FilteredRunnerDiscovery) []ml.Dev ...@@ -349,9 +349,6 @@ func GPUDevices(ctx context.Context, runners []FilteredRunnerDiscovery) []ml.Dev
} }
} }
// Apply any iGPU workarounds
iGPUWorkarounds(devices)
return devices return devices
} }
...@@ -601,32 +598,3 @@ func GetDevicesFromRunner(ctx context.Context, runner BaseRunner) ([]ml.DeviceIn ...@@ -601,32 +598,3 @@ func GetDevicesFromRunner(ctx context.Context, runner BaseRunner) ([]ml.DeviceIn
} }
} }
} }
func iGPUWorkarounds(devices []ml.DeviceInfo) {
// short circuit if we have no iGPUs
anyiGPU := false
for i := range devices {
if devices[i].Integrated {
anyiGPU = true
break
}
}
if !anyiGPU {
return
}
memInfo, err := GetCPUMem()
if err != nil {
slog.Debug("failed to fetch system memory information for iGPU", "error", err)
return
}
for i := range devices {
if !devices[i].Integrated {
continue
}
// NVIDIA iGPUs return useless free VRAM data which ignores system buff/cache
if devices[i].Library == "CUDA" {
devices[i].FreeMemory = memInfo.FreeMemory
}
}
}
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