def _generate_amd_devices()

in azure-slurm/slurmcc/cli.py [0:0]

• 5 lines of code
• 4 McCabe index (conditional complexity)

def _generate_amd_devices(gpu_count: int) -> str:
    if gpu_count == 1:
        return "/dev/dri/renderD128"
    # NOTE: AMD GPU devices should be comma-separated and may not have spaces in the list
    amd_gpu_list = ",".join([f"{128+8*index}" for index in range(0, gpu_count)])
    return "/dev/dri/renderD[{}]".format(amd_gpu_list)