xmanager-slurm 0.4.12__py3-none-any.whl → 0.4.13__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.

This version of xmanager-slurm might be problematic. Click here for more details.

Files changed (6) hide show
  1. xm_slurm/contrib/clusters/drac.py +52 -1
  2. {xmanager_slurm-0.4.12.dist-info → xmanager_slurm-0.4.13.dist-info}/METADATA +1 -1
  3. {xmanager_slurm-0.4.12.dist-info → xmanager_slurm-0.4.13.dist-info}/RECORD +6 -6
  4. {xmanager_slurm-0.4.12.dist-info → xmanager_slurm-0.4.13.dist-info}/WHEEL +0 -0
  5. {xmanager_slurm-0.4.12.dist-info → xmanager_slurm-0.4.13.dist-info}/entry_points.txt +0 -0
  6. {xmanager_slurm-0.4.12.dist-info → xmanager_slurm-0.4.13.dist-info}/licenses/LICENSE.md +0 -0
xm_slurm/contrib/clusters/drac.py CHANGED
@@ -4,7 +4,7 @@ from typing import Literal
4
4
  from xm_slurm import config
5
5
  from xm_slurm.resources import FeatureType, ResourceType
6
6
 
7
- __all__ = ["narval", "beluga", "cedar", "graham"]
7
+ __all__ = ["beluga", "cedar", "fir", "graham", "narval"]
8
8
 
9
9
 
10
10
  def _drac_cluster(
@@ -121,6 +121,36 @@ def beluga(
121
121
  )
122
122
 
123
123
 
124
+ def rorqual(
125
+ *,
126
+ user: str | None = None,
127
+ account: str | None = None,
128
+ proxy: Literal["submission-host"] | str | None = None,
129
+ mounts: dict[os.PathLike[str] | str, os.PathLike[str] | str] | None = None,
130
+ ) -> config.SlurmClusterConfig:
131
+ """DRAC Beluga Cluster (https://docs.alliancecan.ca/wiki/Rorqual/en)."""
132
+ modules = []
133
+ if proxy != "submission-host":
134
+ modules.append("httpproxy")
135
+
136
+ return _drac_cluster(
137
+ name="rorqual",
138
+ host="robot.rorqual.alliancecan.ca",
139
+ host_public_key=config.PublicKey(
140
+ "ssh-ed25519", "AAAAC3NzaC1lZDI1NTE5AAAAINME5e9bifKZbuKKOQSpe3xrvC4g1b0QLMYj+AXBQGJe"
141
+ ),
142
+ user=user,
143
+ account=account,
144
+ mounts=mounts,
145
+ proxy=proxy,
146
+ modules=modules,
147
+ resources={ResourceType.H100: "h100"},
148
+ features={
149
+ FeatureType.NVIDIA_NVLINK: "nvlink",
150
+ },
151
+ )
152
+
153
+
124
154
  def cedar(
125
155
  *,
126
156
  user: str | None = None,
@@ -146,6 +176,27 @@ def cedar(
146
176
  )
147
177
 
148
178
 
179
+ def fir(
180
+ *,
181
+ user: str | None = None,
182
+ account: str | None = None,
183
+ mounts: dict[os.PathLike[str] | str, os.PathLike[str] | str] | None = None,
184
+ ) -> config.SlurmClusterConfig:
185
+ """DRAC Fir Cluster (https://docs.alliancecan.ca/wiki/Fir/en)."""
186
+ return _drac_cluster(
187
+ name="fir",
188
+ host="robot.fir.alliancecan.ca",
189
+ host_public_key=config.PublicKey(
190
+ "ssh-ed25519",
191
+ "AAAAC3NzaC1lZDI1NTE5AAAAIJtenyJz+inwobvlJntWYFNu+ANcVWNcOHRKcEN6zmDo",
192
+ ),
193
+ user=user,
194
+ account=account,
195
+ mounts=mounts,
196
+ resources={ResourceType.H100: "h100"},
197
+ )
198
+
199
+
149
200
  def graham(
150
201
  *,
151
202
  user: str | None = None,
{xmanager_slurm-0.4.12.dist-info → xmanager_slurm-0.4.13.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: xmanager-slurm
3
- Version: 0.4.12
3
+ Version: 0.4.13
4
4
  Summary: Slurm backend for XManager.
5
5
  Project-URL: GitHub, https://github.com/jessefarebro/xm-slurm
6
6
  Author-email: Jesse Farebrother <jfarebro@cs.mcgill.ca>
{xmanager_slurm-0.4.12.dist-info → xmanager_slurm-0.4.13.dist-info}/RECORD RENAMED
@@ -23,7 +23,7 @@ xm_slurm/api/sqlite/client.py,sha256=jAesCKDuYwnNcAxwJk_1b1TB8cT_QGbSjo1UE3mZjEQ
23
23
  xm_slurm/api/web/client.py,sha256=uO67Y7fnQ-w__Vm_A5BEuy7Qi8wQcWk3vIsBGEBkyfk,6261
24
24
  xm_slurm/contrib/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
25
25
  xm_slurm/contrib/clusters/__init__.py,sha256=XFCVnkThiU3_8uA_tUgDByOBanXNHrxDvfmuptmQ2KE,2214
26
- xm_slurm/contrib/clusters/drac.py,sha256=ViLYerYBMSuZXnWVbz9RDIPPV7JA8BgBpgTfj1wPP28,5881
26
+ xm_slurm/contrib/clusters/drac.py,sha256=vY3dxrNUk12H9Gq-tuCcqo2YcdTGq-4LJnQF6DzD4_k,7431
27
27
  xm_slurm/experimental/parameter_controller.py,sha256=b5LfglHV307F6QcPrHeZX5GJBtyOK9aQydke_SZ3Wto,8457
28
28
  xm_slurm/packaging/__init__.py,sha256=dh307yLpUT9KN7rJ1e9fYC6hegGKfZcGboUq9nGpDVQ,233
29
29
  xm_slurm/packaging/docker.py,sha256=-DWcB9qqbeHmIEqyfF0-v6xOT25ae90u2x-QZ7kluOw,13579
@@ -44,8 +44,8 @@ xm_slurm/templates/slurm/fragments/monitor.bash.j2,sha256=BJ1brSjhESOe9VX_OYaPyy
44
44
  xm_slurm/templates/slurm/fragments/proxy.bash.j2,sha256=VJLglZo-Nvx9R-qe3rHTxr07CylTQ6Z9NwBzvIpAZrA,814
45
45
  xm_slurm/templates/slurm/runtimes/apptainer.bash.j2,sha256=lE2EWVCK2O-n08RL4_MJYIikVTvODjcYKuv7Eh73Q2w,1932
46
46
  xm_slurm/templates/slurm/runtimes/podman.bash.j2,sha256=3j7K5eyXt_WhXK0EoMlxnhlmFVJ2JyxRKbsMRaDqzSs,1148
47
- xmanager_slurm-0.4.12.dist-info/METADATA,sha256=ttPSMz6bQs8BEQVCurQ9lFWV7MuzLOHWU7lISOOoufA,1007
48
- xmanager_slurm-0.4.12.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
49
- xmanager_slurm-0.4.12.dist-info/entry_points.txt,sha256=_HLGmLgxuQLOPmF2gOFYDVq2HqtMVD_SzigHvUh8TCY,49
50
- xmanager_slurm-0.4.12.dist-info/licenses/LICENSE.md,sha256=IxstXr3MPHwTJ5jMrByHrQsR1ZAGQ2U_uz_4qzI_15Y,11756
51
- xmanager_slurm-0.4.12.dist-info/RECORD,,
47
+ xmanager_slurm-0.4.13.dist-info/METADATA,sha256=E8QV408HDZOHu4OhsV8Xq4pon_1A5Qi3RLABBAS9iDM,1007
48
+ xmanager_slurm-0.4.13.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
49
+ xmanager_slurm-0.4.13.dist-info/entry_points.txt,sha256=_HLGmLgxuQLOPmF2gOFYDVq2HqtMVD_SzigHvUh8TCY,49
50
+ xmanager_slurm-0.4.13.dist-info/licenses/LICENSE.md,sha256=IxstXr3MPHwTJ5jMrByHrQsR1ZAGQ2U_uz_4qzI_15Y,11756
51
+ xmanager_slurm-0.4.13.dist-info/RECORD,,