Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
24 changes: 18 additions & 6 deletions examples/distributed/README.md
Original file line number Diff line number Diff line change
@@ -1,13 +1,11 @@
# Distributed Examples

This directory contains examples demonstrating distributed computing capabilities using TileLang.
These examples are sorted into two categories:
- Examples under `nvshmem` folder and inter-node examples depend on NVSHMEM library for distributed communication.
- Other examples have no external dependency and only rely on TileScale IPC

For example,
```
./tilelang/distributed/launch.sh examples/distributed/example_allgather.py
```

## Prerequisites
## `nvshmem` examples

Before running the examples, you need to build NVSHMEM library for device-side code generation.

Expand All @@ -28,3 +26,17 @@ Then you can test python import:
```bash
python -c "import pynvshmem"
```

Finally, run examples like this:
```bash
TILELANG_USE_DISTRIBUTED=1 bash ./tilelang/distributed/launch.sh examples/distributed/nvshmem/example_allgather.py
```

## IPC-based examples

Simply run via python:
```bash
TILELANG_USE_DISTRIBUTED=1 python examples/distributed/intranode/example_allgather_gemm_overlapped.py
```

> Tips: To disable annoying NCCL IB logs, consider running with: `NCCL_IB_DISABLE=1`
113 changes: 0 additions & 113 deletions examples/distributed/example_allgather_gemm.py

This file was deleted.

58 changes: 0 additions & 58 deletions examples/distributed/example_nvshmem.py

This file was deleted.

Loading
Loading