Skip to content

Commit 89be4b6

Browse files
theFongclaude
andcommitted
Filter out non-NVIDIA GPUs from gpu-search results
Only show NVIDIA GPUs (exclude AMD Radeon, Intel Gaudi, etc.) since compute capability is NVIDIA-specific. Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
1 parent 701e232 commit 89be4b6

1 file changed

Lines changed: 5 additions & 0 deletions

File tree

pkg/cmd/gpusearch/gpusearch.go

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -344,6 +344,11 @@ func filterInstances(instances []GPUInstanceInfo, gpuName string, minVRAM, minTo
344344
var filtered []GPUInstanceInfo
345345

346346
for _, inst := range instances {
347+
// Filter out non-NVIDIA GPUs (AMD, Intel/Habana, etc.)
348+
if !strings.Contains(strings.ToUpper(inst.Manufacturer), "NVIDIA") {
349+
continue
350+
}
351+
347352
// Filter by GPU name (case-insensitive partial match)
348353
if gpuName != "" && !strings.Contains(strings.ToLower(inst.GPUName), strings.ToLower(gpuName)) {
349354
continue

0 commit comments

Comments
 (0)