torchx-nightly 2024.5.6__py3-none-any.whl → 2024.5.9__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.

Files changed (8) hide show
  1. torchx/examples/torchx_out_of_sync_training.py +11 -0
  2. torchx/schedulers/docker_scheduler.py +1 -1
  3. {torchx_nightly-2024.5.6.dist-info → torchx_nightly-2024.5.9.dist-info}/METADATA +1 -1
  4. {torchx_nightly-2024.5.6.dist-info → torchx_nightly-2024.5.9.dist-info}/RECORD +8 -7
  5. {torchx_nightly-2024.5.6.dist-info → torchx_nightly-2024.5.9.dist-info}/LICENSE +0 -0
  6. {torchx_nightly-2024.5.6.dist-info → torchx_nightly-2024.5.9.dist-info}/WHEEL +0 -0
  7. {torchx_nightly-2024.5.6.dist-info → torchx_nightly-2024.5.9.dist-info}/entry_points.txt +0 -0
  8. {torchx_nightly-2024.5.6.dist-info → torchx_nightly-2024.5.9.dist-info}/top_level.txt +0 -0
torchx/examples/torchx_out_of_sync_training.py ADDED
@@ -0,0 +1,11 @@
1
+ #!/usr/bin/env python3
2
+ # Copyright (c) Meta Platforms, Inc. and affiliates.
3
+ # All rights reserved.
4
+ #
5
+ # This source code is licensed under the BSD-style license found in the
6
+ # LICENSE file in the root directory of this source tree.
7
+
8
+ # pyre-strict
9
+
10
+ # This file is for lex team to ramp up on torchX OSS.
11
+ # Please do not modify this file unless you know what you are doing.
torchx/schedulers/docker_scheduler.py CHANGED
@@ -314,7 +314,7 @@ class DockerScheduler(DockerWorkspaceMixin, Scheduler[DockerOpts]):
314
314
  c.kwargs["device_requests"] = [
315
315
  DeviceRequest(
316
316
  count=resource.gpu,
317
- capabilities=[["compute"]],
317
+ capabilities=[["compute", "utility"]],
318
318
  )
319
319
  ]
320
320
  req.containers.append(c)
{torchx_nightly-2024.5.6.dist-info → torchx_nightly-2024.5.9.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: torchx-nightly
3
- Version: 2024.5.6
3
+ Version: 2024.5.9
4
4
  Summary: TorchX SDK and Components
5
5
  Home-page: https://github.com/pytorch/torchx
6
6
  Author: TorchX Devs
{torchx_nightly-2024.5.6.dist-info → torchx_nightly-2024.5.9.dist-info}/RECORD RENAMED
@@ -36,6 +36,7 @@ torchx/components/integration_tests/component_provider.py,sha256=cFNGqmclcZTJlOW
36
36
  torchx/components/integration_tests/integ_tests.py,sha256=O8jd8Jq5O0mns7xzIFsHexBDHkIIAIfELQkWCzNPzRw,5165
37
37
  torchx/distributed/__init__.py,sha256=pkB_V4eX0DnupRsmOGnINL2XGDM1oRLG0-fH7-feeNI,10280
38
38
  torchx/examples/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
39
+ torchx/examples/torchx_out_of_sync_training.py,sha256=4kwAtPKT7H5FRSgLwXc0KJKv6bo8ZioXCyUNZrErQzo,370
39
40
  torchx/examples/apps/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
40
41
  torchx/examples/apps/datapreproc/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
41
42
  torchx/examples/apps/datapreproc/datapreproc.py,sha256=cu88O_WZgqZ6g7jVIG2kagAVbJ4oPMzTH03_H65w8RU,4317
@@ -68,7 +69,7 @@ torchx/schedulers/api.py,sha256=s2hI87uAWtU2SHMNBKjAqelzQU_GKp_BjcxdtjVVDDk,1415
68
69
  torchx/schedulers/aws_batch_scheduler.py,sha256=7qxy3UFRq0F731-kTjEi6VABWKD60o0req6CBMsTohU,27975
69
70
  torchx/schedulers/aws_sagemaker_scheduler.py,sha256=dPah3yaKFUVm-ZZrzFbyM_abP-LCTd-AcAjZ6t2iycU,20699
70
71
  torchx/schedulers/devices.py,sha256=BnjZnbXGTWiZKLmMKE3zPDOkb4Vao3jgPVa01aV6vyY,1367
71
- torchx/schedulers/docker_scheduler.py,sha256=jQIa6Hxor3N3-lISpbIm1xae4oNyZeTOewEbukBd6nU,16193
72
+ torchx/schedulers/docker_scheduler.py,sha256=xgTStwLrf9hKiz2mbaSCuHWeUBZKJBG30awo7F2rCTE,16204
72
73
  torchx/schedulers/gcp_batch_scheduler.py,sha256=dlUfvjfMuQiRcSXQAdwxqdadwPhOf82L5u-ejRWtFgE,16226
73
74
  torchx/schedulers/ids.py,sha256=3E-_vwVYC-8Tv8kjuY9-W7TbOe_-Laqd8a65uIN3hQY,1798
74
75
  torchx/schedulers/kubernetes_mcad_scheduler.py,sha256=xAt-on3K8HwS2kzWasn0zXd2q4IDQzo2N5A5Ehh9NII,42885
@@ -112,9 +113,9 @@ torchx/workspace/__init__.py,sha256=FqN8AN4VhR1C_SBY10MggQvNZmyanbbuPuE-JCjkyUY,
112
113
  torchx/workspace/api.py,sha256=1heBmPgB-W5Zf9gwViM7NrqvHpZlVYeMN7jpY8Qkytc,5479
113
114
  torchx/workspace/dir_workspace.py,sha256=npNW_IjUZm_yS5r-8hrRkH46ndDd9a_eApT64m1S1T4,2268
114
115
  torchx/workspace/docker_workspace.py,sha256=PFu2KQNVC-0p2aKJ-W_BKA9ZOmXdCY2ABEkCExp3udQ,10269
115
- torchx_nightly-2024.5.6.dist-info/LICENSE,sha256=WVHfXhFC0Ia8LTKt_nJVYobdqTJVg_4J3Crrfm2A8KQ,1721
116
- torchx_nightly-2024.5.6.dist-info/METADATA,sha256=w8Z5nndX5EJU9vZx9k3qqO3otVWZD7sc5jP7ROniVGc,6052
117
- torchx_nightly-2024.5.6.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
118
- torchx_nightly-2024.5.6.dist-info/entry_points.txt,sha256=3JYZFlX9aWzR-Gs_qsx1zq7mlqbFz6Mi9rQUULW8caI,170
119
- torchx_nightly-2024.5.6.dist-info/top_level.txt,sha256=pxew3bc2gsiViS0zADs0jb6kC5v8o_Yy_85fhHj_J1A,7
120
- torchx_nightly-2024.5.6.dist-info/RECORD,,
116
+ torchx_nightly-2024.5.9.dist-info/LICENSE,sha256=WVHfXhFC0Ia8LTKt_nJVYobdqTJVg_4J3Crrfm2A8KQ,1721
117
+ torchx_nightly-2024.5.9.dist-info/METADATA,sha256=gVqoaLL4qCoh_jeMy0YkkaB2JxjojU94dbwHnRSXfSo,6052
118
+ torchx_nightly-2024.5.9.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
119
+ torchx_nightly-2024.5.9.dist-info/entry_points.txt,sha256=3JYZFlX9aWzR-Gs_qsx1zq7mlqbFz6Mi9rQUULW8caI,170
120
+ torchx_nightly-2024.5.9.dist-info/top_level.txt,sha256=pxew3bc2gsiViS0zADs0jb6kC5v8o_Yy_85fhHj_J1A,7
121
+ torchx_nightly-2024.5.9.dist-info/RECORD,,