Skip to content

Commit 0e3697c

Browse files
Add GPU shared mode support to quickrun.sbatch
- Enabled shared GPU compute mode by adding the `#SBATCH --gpu_cmode=shared` directive. - Included comments to clarify the purpose of the new setting and provide a reference for users regarding GPU compute modes.
1 parent 6a5f098 commit 0e3697c

1 file changed

Lines changed: 4 additions & 0 deletions

File tree

scripts/openfe/quickrun.sbatch

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,10 +9,14 @@
99
#SBATCH --ntasks-per-node=1
1010
#SBATCH --cpus-per-task=6
1111
#SBATCH --gpus=1
12+
#SBATCH --gpu_cmode=shared
1213
#SBATCH --mem=12G
1314
#SBATCH --constraint="GPU_SKU:A100_PCIE|GPU_SKU:A100_SXM4|GPU_SKU:A40|GPU_SKU:H100_SXM5|GPU_SKU:H200_SXM5|GPU_SKU:L40S|GPU_SKU:V100_PCIE|GPU_SKU:V100S_PCIE|GPU_SKU:V100_SXM2"
1415
#SBATCH --time=2-00:00:00
1516

17+
# Here we enable default mode so that multiple contexts are allowed per device
18+
# See https://www.sherlock.stanford.edu/docs/user-guide/gpu/#gpu-compute-modes
19+
1620
set -euo pipefail
1721

1822
OPENFE_CONTAINER=/scratch/groups/ayting/containers/openfe/openfe_latest-apptainer.sif

0 commit comments

Comments
 (0)