torchx-nightly 2024.5.13__py3-none-any.whl → 2024.5.16__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 torchx-nightly might be problematic. Click here for more details.

@@ -109,6 +109,16 @@ def aws_p4de_24xlarge() -> Resource:
109
109
  )
110
110
 
111
111
 
112
+ def aws_p5_48xlarge() -> Resource:
113
+ return Resource(
114
+ cpu=192,
115
+ gpu=8,
116
+ memMB=2048 * GiB,
117
+ capabilities={K8S_ITYPE: "p5.48xlarge"},
118
+ devices={EFA_DEVICE: 32},
119
+ )
120
+
121
+
112
122
  def aws_t3_medium() -> Resource:
113
123
  return Resource(cpu=2, gpu=0, memMB=4 * GiB, capabilities={K8S_ITYPE: "t3.medium"})
114
124
 
@@ -268,6 +278,7 @@ NAMED_RESOURCES: Mapping[str, Callable[[], Resource]] = {
268
278
  "aws_p3dn.24xlarge": aws_p3dn_24xlarge,
269
279
  "aws_p4d.24xlarge": aws_p4d_24xlarge,
270
280
  "aws_p4de.24xlarge": aws_p4de_24xlarge,
281
+ "aws_p5.48xlarge": aws_p5_48xlarge,
271
282
  "aws_g4dn.xlarge": aws_g4dn_xlarge,
272
283
  "aws_g4dn.2xlarge": aws_g4dn_2xlarge,
273
284
  "aws_g4dn.4xlarge": aws_g4dn_4xlarge,
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: torchx-nightly
3
- Version: 2024.5.13
3
+ Version: 2024.5.16
4
4
  Summary: TorchX SDK and Components
5
5
  Home-page: https://github.com/pytorch/torchx
6
6
  Author: TorchX Devs
@@ -87,7 +87,7 @@ torchx/specs/api.py,sha256=73jJOYQ-H-jX3Yvbc9hJjolavNQdF8bW9Q0oQnWzLL0,34885
87
87
  torchx/specs/builders.py,sha256=QDcQrnCO4bdSaiP0216XbCgTsnLutO_1_FW5jDiEIWI,9939
88
88
  torchx/specs/file_linter.py,sha256=IeiomB1BgHUlT-ZsvGxar3llY63NOupfLBrOrD_---A,11860
89
89
  torchx/specs/finder.py,sha256=MnwxG_UC4a-3X2wQ37ANEQR6D1TvriCLyuVYBh_-wuI,16249
90
- torchx/specs/named_resources_aws.py,sha256=8jsbFZFTa1ZkdpZmBrqhVfSVuZonrXbADhTPmIRdtEQ,7743
90
+ torchx/specs/named_resources_aws.py,sha256=gmMbzzaYy3WzwDFo5iyd28icX7zRbUOseLl8yudY13Q,7988
91
91
  torchx/specs/named_resources_generic.py,sha256=Sg4tAdqiiWDrDz2Lj_pnfsjzGIXKTou73wPseh6j55w,2646
92
92
  torchx/specs/test/components/__init__.py,sha256=J8qjUOysmcMAek2KFN13mViOXZxTYc5vCrF02t3VuFU,223
93
93
  torchx/specs/test/components/a/__init__.py,sha256=kdxEgnI8QBSBiuTjaB4qDD7JX84hWowyPWU4B2Cqe9A,561
@@ -113,9 +113,9 @@ torchx/workspace/__init__.py,sha256=FqN8AN4VhR1C_SBY10MggQvNZmyanbbuPuE-JCjkyUY,
113
113
  torchx/workspace/api.py,sha256=1heBmPgB-W5Zf9gwViM7NrqvHpZlVYeMN7jpY8Qkytc,5479
114
114
  torchx/workspace/dir_workspace.py,sha256=npNW_IjUZm_yS5r-8hrRkH46ndDd9a_eApT64m1S1T4,2268
115
115
  torchx/workspace/docker_workspace.py,sha256=PFu2KQNVC-0p2aKJ-W_BKA9ZOmXdCY2ABEkCExp3udQ,10269
116
- torchx_nightly-2024.5.13.dist-info/LICENSE,sha256=WVHfXhFC0Ia8LTKt_nJVYobdqTJVg_4J3Crrfm2A8KQ,1721
117
- torchx_nightly-2024.5.13.dist-info/METADATA,sha256=1Mz12fs68xabgZj_x2IONjmr1Dg7Z0ZiIGuztllXUT8,6053
118
- torchx_nightly-2024.5.13.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
119
- torchx_nightly-2024.5.13.dist-info/entry_points.txt,sha256=3JYZFlX9aWzR-Gs_qsx1zq7mlqbFz6Mi9rQUULW8caI,170
120
- torchx_nightly-2024.5.13.dist-info/top_level.txt,sha256=pxew3bc2gsiViS0zADs0jb6kC5v8o_Yy_85fhHj_J1A,7
121
- torchx_nightly-2024.5.13.dist-info/RECORD,,
116
+ torchx_nightly-2024.5.16.dist-info/LICENSE,sha256=WVHfXhFC0Ia8LTKt_nJVYobdqTJVg_4J3Crrfm2A8KQ,1721
117
+ torchx_nightly-2024.5.16.dist-info/METADATA,sha256=G_brjaroLDfy83kMu3f1bU_41kHtb7eymtCU8sXj5iU,6053
118
+ torchx_nightly-2024.5.16.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
119
+ torchx_nightly-2024.5.16.dist-info/entry_points.txt,sha256=3JYZFlX9aWzR-Gs_qsx1zq7mlqbFz6Mi9rQUULW8caI,170
120
+ torchx_nightly-2024.5.16.dist-info/top_level.txt,sha256=pxew3bc2gsiViS0zADs0jb6kC5v8o_Yy_85fhHj_J1A,7
121
+ torchx_nightly-2024.5.16.dist-info/RECORD,,