Snakepit.Hardware.ROCmDetector (Snakepit v0.8.7)

AMD ROCm GPU hardware detection.

Detects AMD GPUs with ROCm support using rocm-smi when available.

Summary

rocm_device()

rocm_info()

detect()

Detects ROCm GPU information.

@type rocm_device() :: %{
  id: non_neg_integer(),
  name: String.t(),
  memory_total_mb: non_neg_integer(),
  memory_free_mb: non_neg_integer()
}

@type rocm_info() :: %{version: String.t(), devices: [rocm_device()]}

@spec detect() :: rocm_info() | nil

Detects ROCm GPU information.

Returns nil if ROCm is not available, or a map with: