Method GPUCounts

utils/spec_store.go:119–127

GPUCounts returns all valid GPU counts for a given GPU type, sorted.

(gpuType string)

// GPUCounts returns all valid GPU counts for a given GPU type, sorted.
func (s *SpecStore) GPUCounts(gpuType string) []int {
	var counts []int
	for gpuCount := 1; gpuCount <= 8; gpuCount++ {
		if s.Lookup(gpuType, gpuCount) != nil {
			counts = append(counts, gpuCount)
		}
	}
	return counts
}

// IsGPUTypeAvailable reports whether any count for this GPU type is available.
func (s *SpecStore) IsGPUTypeAvailable(gpuType string) bool {