romtools.hpc.scheduler#

Functions

create_slurm_script(job_name, num_nodes, ...)

Create a SLURM job script with the specified parameters.

romtools.hpc.scheduler.create_slurm_script(job_name, num_nodes, tasks_per_node, wall_time, wcid, partition, command)[source]#

Create a SLURM job script with the specified parameters.

Parameters: - job_name: Name of the SLURM job - num_nodes: Number of nodes to request - tasks_per_node: Number of tasks to run on each node - wall_time: Maximum wall time for the job (format: HH:MM:SS) - wcid: The account to charge for the job - partition: The partition to submit the job to (“batch”, “short”) - command: The command to execute in the job

Returns: A string containing the SLURM job script.

Parameters:
  • job_name (str)

  • num_nodes (int)

  • tasks_per_node (int)

  • wall_time (str)

  • wcid (str)

  • partition (str)

  • command (str)

Return type:

str