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
}
}
}