Add gpu-search command to search and filter GPU instance types#275
Closed
theFong wants to merge 3 commits into
Closed
Add gpu-search command to search and filter GPU instance types#275theFong wants to merge 3 commits into
theFong wants to merge 3 commits into
Conversation
Introduces a new `brev gpu-search` command (also aliased as `brev gpu`, `brev gpus`, and `brev gpu-list`) that allows users to search and filter GPU instance types from the Brev API. Features include: - Filter by GPU name (case-insensitive partial match) - Filter by minimum VRAM per GPU (in GB) - Filter by minimum total VRAM (GPU count * VRAM) - Filter by minimum GPU compute capability (e.g., 8.0 for Ampere) - Sort by price, gpu-count, vram, total-vram, vcpu, type, or capability - Support for ascending/descending sort order The command displays results in a formatted table showing instance type, GPU name, count, VRAM per GPU, total VRAM, compute capability, vCPUs, and hourly price. Includes comprehensive unit tests for filtering, sorting, and data processing functionality.
Member
Author
|
$ brev gpu-search ✅ Successfully fetched 536 GPU instance types from the API✅ Results displayed in formatted table sorted by price (default)$ brev gpu --gpu-name A100 ✅ Returned only A100 instances (partial match, case-insensitive)$ brev gpu -g H100 ✅ Returned only H100 instances$ brev gpu --gpu-name L4 ✅ Returned L4 instances (correctly not matching L40/L40S)$ brev gpu --sort price ✅ Sorted by price ascending (cheapest first)$ brev gpu --sort total-vram --desc ✅ Sorted by total VRAM descending (B200x8 1536GB first)$ brev gpu --sort gpu-count -d ✅ Sorted by GPU count descending (8-GPU configs first)$ brev gpu --sort capability ✅ Sorted by compute capability ascending |
Add capability mappings for: - RTXPro6000 (12.0), B200 and RTX5090 (10.0 Blackwell) - RTX6000Ada, RTX4000Ada (8.9 Ada Lovelace) - A6000, A5000, A4000 (8.6 Ampere) - RTX6000 (7.5 Turing) - M60 (5.2 Maxwell) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
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>
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Introduces a new
brev gpu-searchcommand (also aliased asbrev gpu,brev gpus, andbrev gpu-list) that allows users to search and filterGPU instance types from the Brev API. Features include:
The command displays results in a formatted table showing instance type,
GPU name, count, VRAM per GPU, total VRAM, compute capability, vCPUs,
and hourly price.
Includes comprehensive unit tests for filtering, sorting, and data
processing functionality.