planit/example_vsc_args.py at main · WPoelman/planit · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
"""
Example planit configurations for VSC clusters (Wice / Genius).

The cpu counts for GPU partitions are *not* arbitrary, these are required per node.
The counts for the batch jobs are the max.

See: https://docs.vscentrum.be/leuven/tier2_hardware.html

Make sure to fill in "account" and "mail_user" with actual values!
"""

from planit import MailType, SlurmArgs

MAIL = [MailType.BEGIN, MailType.END, MailType.FAIL]


WICE_BATCH = SlurmArgs(
    time="01:00:00",
    partition="batch",
    cpus_per_task=72,
    account="my-hpc-account",
    cluster="wice",
    mail_user="me@uni.edu",
    mail_type=MAIL,
)

WICE_A100 = SlurmArgs(
    time="01:00:00",
    partition="gpu_a100",
    gpus_per_node=1,
    cpus_per_gpu=18,
    account="my-hpc-account",
    cluster="wice",
    mail_user="me@uni.edu",
    mail_type=MAIL,
)

WICE_H100 = SlurmArgs(
    time="01:00:00",
    partition="gpu_h100",
    gpus_per_node=1,
    cpus_per_gpu=16,
    account="my-hpc-account",
    cluster="wice",
    mail_user="me@uni.edu",
    mail_type=MAIL,
)

WICE_A100_DEBUG = SlurmArgs(
    time="01:00:00",
    partition="gpu_a100_debug",
    gpus_per_node=1,
    cpus_per_gpu=64,
    account="my-hpc-account",
    cluster="wice",
    mail_user="me@uni.edu",
    mail_type=MAIL,
)

GENIUS_BATCH = SlurmArgs(
    time="01:00:00",
    partition="batch",
    cpus_per_task=36,
    account="my-hpc-account",
    cluster="genius",
    mail_user="me@uni.edu",
    mail_type=MAIL,
)

GENIUS_V100 = SlurmArgs(
    time="01:00:00",
    partition="gpu_v100",
    gpus_per_node=1,
    cpus_per_gpu=4,
    account="my-hpc-account",
    cluster="genius",
    mail_user="me@uni.edu",
    mail_type=MAIL,
)