CODEWITHSUNDEEP
cluster-computinggpuslurm
user21096
user21096

Reputation: 11

code flow of slurm gpu allocation

Does any one know code flow how GPUs are allocated in SLURM? I have gone through and found that CUDA_VISIBLE_DEVICES environment variable is not updated any where in code. So how actually it is done in code?

My goal is to add new hardware support using environment variable for allocation.

Upvotes: 1

Views: 208

Answers (1)

Carles Fenoy
Carles Fenoy

Reputation: 5377

Take a look at the gpu gres plugin. There is where the code modifies the CUDA_VISIBLE_DEVICES environment variable. The file path is:

src/plugins/gres/gpu/gres_gpu.c

Upvotes: 1

Related Questions