Function testSpecStore

cmd/create_test.go:16–32

()

)

func testSpecStore() *utils.SpecStore {
	return utils.NewSpecStore(map[string]api.GpuSpecConfig{
		"a100xl_x1": {DisplayName: "NVIDIA A100 (80GB)", VramGB: 80, GpuCount: 1, VcpuOptions: []int{4, 8, 12}, RamPerVCPUGiB: 8, StorageGB: api.StorageRange{Min: 100, Max: 500}},
		"a100xl_x2": {DisplayName: "NVIDIA A100 (80GB)", VramGB: 80, GpuCount: 2, VcpuOptions: []int{8, 12, 16, 20, 24}, RamPerVCPUGiB: 8, StorageGB: api.StorageRange{Min: 100, Max: 1000}},
		"a100xl_x4": {DisplayName: "NVIDIA A100 (80GB)", VramGB: 80, GpuCount: 4, VcpuOptions: []int{60}, RamPerVCPUGiB: 8, StorageGB: api.StorageRange{Min: 100, Max: 2000}},
		"a100xl_x8": {DisplayName: "NVIDIA A100 (80GB)", VramGB: 80, GpuCount: 8, VcpuOptions: []int{120}, RamPerVCPUGiB: 8, StorageGB: api.StorageRange{Min: 100, Max: 4000}},
		"a6000_x1":  {DisplayName: "RTX A6000", VramGB: 48, GpuCount: 1, VcpuOptions: []int{4, 6}, RamPerVCPUGiB: 8, StorageGB: api.StorageRange{Min: 100, Max: 500}},
		"h100_x1":   {DisplayName: "NVIDIA H100", VramGB: 80, GpuCount: 1, VcpuOptions: []int{4, 8, 12, 16}, RamPerVCPUGiB: 8, StorageGB: api.StorageRange{Min: 100, Max: 500}},
		"h100_x2":   {DisplayName: "NVIDIA H100", VramGB: 80, GpuCount: 2, VcpuOptions: []int{8, 12, 16, 20, 24}, RamPerVCPUGiB: 8, StorageGB: api.StorageRange{Min: 100, Max: 1000}},
		"h100_x4":   {DisplayName: "NVIDIA H100", VramGB: 80, GpuCount: 4, VcpuOptions: []int{60}, RamPerVCPUGiB: 8, StorageGB: api.StorageRange{Min: 100, Max: 2000}},
		"h100_x8":   {DisplayName: "NVIDIA H100", VramGB: 80, GpuCount: 8, VcpuOptions: []int{120}, RamPerVCPUGiB: 8, StorageGB: api.StorageRange{Min: 100, Max: 4000}},
		"l40_x1":    {DisplayName: "NVIDIA L40", VramGB: 48, GpuCount: 1, VcpuOptions: []int{4, 8}, RamPerVCPUGiB: 8, StorageGB: api.StorageRange{Min: 100, Max: 500}},
		"l40_x2":    {DisplayName: "NVIDIA L40", VramGB: 48, GpuCount: 2, VcpuOptions: []int{8, 12, 16}, RamPerVCPUGiB: 8, StorageGB: api.StorageRange{Min: 100, Max: 1000}},
		"l40_x4":    {DisplayName: "NVIDIA L40", VramGB: 48, GpuCount: 4, VcpuOptions: []int{40}, RamPerVCPUGiB: 8, StorageGB: api.StorageRange{Min: 100, Max: 2000}},
		"l40_x8":    {DisplayName: "NVIDIA L40", VramGB: 48, GpuCount: 8, VcpuOptions: []int{80}, RamPerVCPUGiB: 8, StorageGB: api.StorageRange{Min: 100, Max: 4000}},
	})
}

func tmplEntry(key, displayName string) api.TemplateEntry {
	return api.TemplateEntry{Key: key, Template: types.EnvironmentTemplate{DisplayName: displayName}}