(config string)
| 469 | } |
| 470 | |
| 471 | func isValidWorkloadConfig(config string) bool { |
| 472 | return config == gpuWorkloadConfigContainer || |
| 473 | config == gpuWorkloadConfigVMPassthrough || |
| 474 | config == gpuWorkloadConfigVMVgpu |
| 475 | } |
| 476 | |
| 477 | func getWorkloadConfig(ctx context.Context) (string, error) { |
| 478 | // check if default workload is overridden by flag |