Skip to content

Commit f6f24cf

Browse files
updating to allow for modules (#95)
* updating to allow for modules * hopefully fixed sphinx build
1 parent 78ebe96 commit f6f24cf

File tree

3 files changed

+5
-3
lines changed

3 files changed

+5
-3
lines changed

.github/workflows/SphinxBuild.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@ jobs:
1212
pre-build-command: "apt install -y pandoc"
1313
uses: ammaraskar/sphinx-action@master
1414
- name: Upload artifacts
15-
uses: actions/upload-artifact@v3
15+
uses: actions/upload-artifact@v4
1616
with:
1717
name: html-docs
1818
path: docs/_build/html/

cybergis_compute_client/UI.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -55,7 +55,7 @@ def __init__(self, compute, defaultJobName="hello_world", defaultDataFolder="./"
5555
'num_of_node', 'num_of_task', 'time',
5656
'cpu_per_task', 'memory_per_cpu', 'memory_per_gpu',
5757
'memory', 'gpus', 'gpus_per_node', 'gpus_per_socket',
58-
'gpus_per_task', 'partition']
58+
'gpus_per_task', 'partition', 'modules']
5959
self.slurm_integer_configs = [
6060
'num_of_node', 'num_of_task', 'time', 'cpu_per_task',
6161
'memory_per_cpu', 'memory_per_gpu', 'memory', 'gpus',
@@ -65,7 +65,7 @@ def __init__(self, compute, defaultJobName="hello_world", defaultDataFolder="./"
6565
self.slurm_integer_none_unit_config = [
6666
'cpu_per_task', 'num_of_node', 'num_of_task', 'gpus',
6767
'gpus_per_node', 'gpus_per_socket', 'gpus_per_task']
68-
self.slurm_string_option_configs = ['partition']
68+
self.slurm_string_option_configs = ['partition', 'modules']
6969
self.globus_filename = None
7070
self.jupyter_globus = None
7171

docs/model_contribution/develop_model.rst

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -125,6 +125,7 @@ Slurm parameters can be added by adding a “slurm_input_rules” to the manifes
125125
* **gpus_per_socket (integerRule):** The number of GPUs required for the job on each socket included in the job's resource allocation.
126126
* **gpus_per_task (integerRule):** The number of GPUs required for the job on each task to be spawned in the job's resource allocation.
127127
* **partition (stringOptionRule):** The partition name on the HPC.
128+
* **modules (stringOptionRule):** The modules you would like to load on the HPC
128129

129130
You can specify these SLURM parameters including a reasonable range for running your model. The UI will read from this manifest and populate widgets for users to specify SLURM settings. The widgets available for each SLURM parameter are::
130131

@@ -143,6 +144,7 @@ You can specify these SLURM parameters including a reasonable range for running
143144
"gpus_per_socket": integerRule, // number of GPU per socket, ie. SBATCH gpus-per-socket
144145
"gpus_per_task": integerRule, // number of GPU per task, ie. SBATCH gpus-per-task
145146
"partition": stringOptionRule // partition name on HPC, ie. SBATCH partition
147+
"modules": stringOptionRule // modules available on the HPC, i.e. "module load xxx yyy"
146148
}
147149
}
148150

0 commit comments

Comments
 (0)