将任务到队列调度系统Slurm中

[perillaroc]

提交Slurm作业，编写 TAKLER_JOB_CMD，两种方式：

• 直接运行 sbatch 命令
• 编写提交脚本，在脚本中运行 sbatch 命令

[perillaroc]

方案一：直接运行

将 sbatch 和 scancel 写入到 TAKLER_SHELL_JOB_CMD 和 TAKLER_SHELL_KILL_CMD 变量中

job_cmd = "sbatch {{ TAKLER_JOB }}"
kill_cmd = "scancel {{ TAKLER_RID }}"


def slurm_serial_job(node: Node, partition: str = "serial"):
    node.add_parameter("TAKLER_SHELL_JOB_CMD", job_cmd)
    node.add_parameter("TAKLER_SHELL_KILL_CMD", kill_cmd)
    node.add_parameter("PARTITION", partition)


def slurm_para_job(node: Node, nodes: int, tasks_per_node: int = 32, partition: str = "serial"):
    node.add_parameter("TAKLER_SHELL_JOB_CMD", job_cmd)
    node.add_parameter("TAKLER_SHELL_KILL_CMD", kill_cmd)
    node.add_parameter("PARTITION", partition)
    node.add_parameter("NODES", nodes)
    node.add_parameter("TASKS_PER_NODE", tasks_per_node)

[perillaroc]

方案二：编写工具脚本

仿照业务系统中 slsubmit6 和 slcancel4 编写提交工具脚本，便于在提交失败时将任务节点设为 aborted，并将提交失败记录写到作业日志中。