Method GPUOptions

utils/spec_store.go:90–116

GPUOptions returns the GPU type identifiers available in any size, ordered by gpuDisplayOrder. Out-of-stock types are preserved in the list but pushed to the bottom.

()

// ordered by gpuDisplayOrder. Out-of-stock types are preserved in the list but
// pushed to the bottom.
func (s *SpecStore) GPUOptions() []string {
	seen := map[string]bool{}
	for key := range s.specs {
		gpuType := strings.Split(key, "_x")[0]
		seen[gpuType] = true
	}
	var ordered []string
	for _, gpu := range gpuDisplayOrder {
		if seen[gpu] {
			ordered = append(ordered, gpu)
		}
	}
	for gpuType := range seen {
		if !slices.Contains(gpuDisplayOrder, gpuType) {
			ordered = append(ordered, gpuType)
		}
	}
	var available, unavailable []string
	for _, gpu := range ordered {
		if s.IsGPUTypeAvailable(gpu) {
			available = append(available, gpu)
		} else {
			unavailable = append(unavailable, gpu)
		}
	}
	return append(available, unavailable...)
}

// GPUCounts returns all valid GPU counts for a given GPU type, sorted.
func (s *SpecStore) GPUCounts(gpuType string) []int {