[slurm-dev] Add support for AMD/ATI GPU

[Dominik Bartkiewicz]

Hi
I add support for selecting amd gpu to slurm gpu gres.
All i have done was additionally set GPU_DEVICE_ORDINAL env.
GPU_DEVICE_ORDINAL env have analogous function to CUDA_VISIBLE_DEVICES.

Dominik Bartkiewicz
ICM
University of Warsaw

--- gres_gpu.c.orig	2014-02-05 14:27:12.311629000 +0100
+++ gres_gpu.c	2014-02-05 13:52:32.921540000 +0100
@@ -250,6 +250,8 @@
 	if (dev_list) {
 		env_array_overwrite(job_env_ptr,"CUDA_VISIBLE_DEVICES",
 				    dev_list);
+		env_array_overwrite(job_env_ptr,"GPU_DEVICE_ORDINAL",
+				    dev_list);
 		xfree(dev_list);
 	}
 }
@@ -293,6 +295,8 @@
 	if (dev_list) {
 		env_array_overwrite(job_env_ptr,"CUDA_VISIBLE_DEVICES",
 				    dev_list);
+		env_array_overwrite(job_env_ptr,"GPU_DEVICE_ORDINAL",
+				    dev_list);
 		xfree(dev_list);
 	}
 }