8000 NUMA binding integration with elastic agent and torchrun (#149334) · pytorch/pytorch@82c74e4 · GitHub
[go: up one dir, main page]

Skip to content

Commit 82c74e4

Browse files
raghavhrishipdesupinski
authored andcommitted
NUMA binding integration with elastic agent and torchrun (#149334)
Implements #148689 Pull Request resolved: #149334 Approved by: https://github.com/d4l3k Co-authored-by: Paul de Supinski <pdesupinski@gmail.com>
1 parent ba05238 commit 82c74e4

File tree

13 files changed

+1383
-5
lines changed

13 files changed

+1383
-5
lines changed

docs/source/distributed.elastic.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,7 @@ elastic/metrics
2929
elastic/events
3030
elastic/subprocess_handler
3131
elastic/control_plane
32+
elastic/numa
3233
```
3334

3435
```{toctree}

docs/source/elastic/numa.rst

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,10 @@
1+
.. _numa-api:
2+
3+
NUMA Binding Utilities
4+
======================
5+
6+
.. automodule:: torch.distributed.numa
7+
:members:
8+
9+
.. automodule:: torch.distributed.numa.binding
10+
:members:

0 commit comments

Comments
 (0)
0