Requesting Either Or in GRES option

HPC Technical
1

Hi,

I was wondering if there is an easy way to have an “either this or that” gres requirement in the sbatch submission.
For example, if I am ambivolent to whether I get a single titan card or RTX card, but don’t want a Pascal card, is it possible ot make this request?
I have tried various logical combinations, but none seem to function.

--gres=gpu:titan:1,gpu:rtx:1
--gres=gpu:titan|rtx:1
--gres=gpu:titan:1/gpu:rtx:1
--gres=gpu:titan:1|gpu:rtx:1

For now I will use --gres=gpu:1 and specify the nodes, but it would be great to know if there is an easier catch all way!

Cheers,
Johnny

2

Dear @John.Raine,

I read the doc and it doesn’t seems it’s possible. As as workaround, I’ve added the following “Feature” to the gpu nodes:

COMPUTE_TYPE_TITAN
COMPUTE_TYPE_PASCAL
COMPUTE_TYPE_RTX

You can then submit your job as usual, specifying the gres with only the number of gpu and select the compute nodes you want to select like this:

#SBATCH --gres=gpu:1
#SBATCH --constraint="COMPUTE_TYPE_TITAN|COMPUTE_TYPE_RTX"

Let us know if this works for you as I didn’t tested it:)

1 Like
3

Hi there,

FWIW, the above seems redundant to me, given that for each GPU type we already have the corresponding COMPUTE_CAPABILITY feature (cf. https://baobab.unige.ch/enduser/src/enduser/applications.html#nvidia-cuda ), thus the selection would be:

#SBATCH --constraint="COMPUTE_CAPABILITY_6_1|COMPUTE_CAPABILITY_7_5"

Thx, bye,
Luca