Skip to content

CLI Missing GPU Support (H200, B200, A10G) #6

@sqdshguy

Description

@sqdshguy

The CLI only supports 6 GPU types while the website supports 8.

Supported GPUs:

Website (https://github.com/tensara/tensara/blob/main/src/constants/gpu.ts#L1-L11):

export const GPU_DISPLAY_NAMES: Record<string, string> = {
  all: "All GPUs",
  T4: "Tesla T4",
  H100: "NVIDIA H100",
  H200: "NVIDIA H200",
  B200: "NVIDIA B200",
  "A100-80GB": "NVIDIA A100",
  A10G: "NVIDIA A10G",
  L40S: "NVIDIA L40S",
  L4: "NVIDIA L4",
}

CLI (https://github.com/tensara/cli/blob/main/src/parser.rs#L7-L14):

pub enum GPU {
    T4,
    A100,
    A100_80GB,
    H100,
    L4,
    L40s,
}

By the way, the CLI uses L40s (lowercase 's') while the website uses L40S (uppercase 'S'). Seems like this might cause some issues potentially, although I haven't tested it to be honest.

The fix is obvious and straightforward. Add the missing GPU variants to the enum in https://github.com/tensara/cli/blob/main/src/parser.rs#L7-L14. Also update the Display implementation (https://github.com/tensara/cli/blob/main/src/parser.rs#L25-L36) and parser (https://github.com/tensara/cli/blob/main/src/parser.rs#L71-L77) to handle these new cases.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions