submit_job.py

2022-04-18 python

以下示例是关于Python中包含submit_job.py用法的示例代码，想了解submit_job.py的具体用法？submit_job.py怎么用？submit_job.py使用的例子？那么可以参考以下相关源代码片段来学习它的具体使用方法。

提交slurm工作由slurmer

文件名：submit_slurm.py[英]：submit_job.py源码类型：Python

import os
from typing import Literal, Optional

import click
from slurmer.job import SlurmJob

Partition = Literal["g1", "g2", "g3", "g4", "g5"]
partition_table = {"g1": 16, "g2": 20, "g3": 24, "g4": 96, "g5": 32}


def check_job_prepared(workdir):
    files_to_check = ("POSCAR", "INCAR", "KPOINTS", "POTCAR")
    for f in files_to_check:
        if not os.path.isfile(os.path.join(workdir, f)):
            return False
    return True


@click.command()
@click.option("-d", "--workdir", help="Working directory")
@click.option("-j", "--jobname", default=None, help="Job name")
@click.option("-p", "--partition", default="g1", help="Partition of nodes")
@click.option("-n", "--nodes", default=4, help="Number of nodes")
@click.option("--gamma", is_flag=True, default=False, help="Use gamma-only version")
def main(
    workdir: str,
    jobname: Optional[str],
    partition: Partition,
    nodes: int,
    gamma: bool,
):
    prepared = check_job_prepared(workdir)
    if not prepared:
        raise RuntimeError("Job is not prepared!")
    job = SlurmJob(
        workdir=workdir,
        template_file="vasp-6-gam.sh" if gamma else "vasp-6-std.sh",
        job_name=workdir if jobname is None else jobname,
        node_partition=partition,
        num_nodes=nodes,
        num_tasks=nodes * partition_table[partition],
    )
    job.submit(write_job_script=True)


if __name__ == "__main__":
    main()

本文地址：https://www.itbaoku.cn/snippets/785087.html

上一篇：创建新的终端

下一篇：不受时间限制地执行jupyter笔记本

相关教程

Web2Py教程

Web2py初学者教程 - 从基本到高级概念的简单简单步骤学习Web2py，其中包括简介，Pytho...

Unittest框架教程

UnitTest框架初学者教程 - 从基本概念到高级概念的简单易学步骤学习UnitTest Fram...

Python Web开发库教程

Python Web开发库初学者教程 - 从基本概念到高级概念，从简单简单的步骤学习Python W...

JCL 教程

JCL概述 - 从概述，环境设置，JOB语句，Exec语句，DD语句，基本库，过程，条件作业处理，定...

JCL教程

JCL概述 - 从概述，环境设置，JOB语句，Exec语句，DD语句，基本库，过程，条件作业处理，定...

Sqoop教程

Sqoop入门教程 - 从基本到高级概念的简单简单步骤学习Sqoop，其中包括简介，安装，导入，导入...

最新源代码片段分享

Shopify获得价值变体

嗯

旋转器中的旋转器

将模式添加为默认块

ConexiónConDB Mi...

在M系列Mac上安装CMS 2

重命名Git分支

bluefile_lead

带有复选框隐藏的字段

BASH脚本中的陷阱错误

Laravel Livewire...

地板骑行，完美，组合

在银色，现代或黎明框架上的粘性海...

watsontcp

WooCommerce终点标题

fb_leads_zap

枚举替代

现代化的线索

home_advisor

文档窗口宽度

server.js带有重写

使用Homebrew将PHP更新...

Swiper的示例具有分页圈边界...

截断日志文件

完全禁用评论

打字稿一般信息

不错的vim之路

更改序列化字段时如何更新对象?

Baoperg

统一的属性序列化

热门代码标签

abap abc actionscript ada apache_conf applescript asciidoc asl assembly assembly_x86 autohotkey basic batchfile c c_cpp cirru clojure coffee coldfusion cpp csharp css d dart delphi diff dockerfile dot ejs elixir elm erlang excel forth fortran gdscript gitignore glsl go golang graphqlschema groovy haml handlebars haskell haxe html html_ruby ini io jade java javascript json jsoniq jsp jsx julia kotlin latex

submit_job.py

相关源代码片段分享

相关文章

相关教程