mirrors/ollama
1
0
Fork 0
mirror of https://github.com/ollama/ollama.git synced 2025-05-11 18:36:41 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

discover: make unique_id check optional for AMD GPU detection

Browse source
This commit is contained in:
Valentin Tunev 2025-04-06 16:12:17 +03:00
parent 0f3f9e353d
commit 0a0752e714
1 changed files with 9 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

12
discover/amd_linux.go
View file

@ -216,9 +216,15 @@ func AMDGetGPUInfo() ([]RocmGPUInfo, error) {
filename := filepath.Join(devDir, m.filename)
buf, err := os.ReadFile(filename)
if err != nil {
slog.Debug("failed to read sysfs node", "file", filename, "error", err)
matched = false
break
// Only treat it as a mismatch if the file is required
if m.filename == DRMUniqueIDFile {
slog.Debug("unique_id not found, skipping this check", "file", filename)
continue
} else {
slog.Debug("failed to read required sysfs node", "file", filename, "error", err)
matched = false
break
}
}
// values here are in hex, strip off the lead 0x and parse so we can compare the numeric (decimal) values in amdgpu
cmp, err := strconv.ParseUint(strings.TrimPrefix(strings.TrimSpace(string(buf)), "0x"), 16, 64)