func readNvidiaSKUConfig()

in pkg/utils/gpu.go [55:72]


func readNvidiaSKUConfig() {
	var nvidiaSKUConfig NvidiaSKUConfig

	err := yaml.Unmarshal(configFile, &nvidiaSKUConfig)
	if err != nil {
		panic(err)
	}
	for _, skus := range nvidiaSKUConfig.NvidiaEnabledSKUFamilies {
		for _, sku := range skus {
			nvidiaEnabledSKUs[sku] = true
		}
	}
	for _, skus := range nvidiaSKUConfig.MarinerNvidiaEnabledSKUFamilies {
		for _, sku := range skus {
			marinerNvidiaEnabledSKUs[sku] = true
		}
	}
}