[examples][CuTeDSL] init commit for distirbuted examples (#2806)

Browse Source

* init commit for distirbuted examples

* better OOB protection

* and try import to nvshmem for better error message and a READMME.md to introduce nvshmem and multimem instructions

* add some lamport explanation

* enhance f8 output and warn that f8 output can have nan in it

* tell user why we need complicate data conversions in ref check part

* tell user we don't support nvshmem device function

---------

Co-authored-by: bangyus <bangyus@nvidia.com>

...

This commit is contained in:

bangyu shen

2025-12-02 11:25:40 +08:00

committed by GitHub

parent 5e847d37c4

commit 52ae719eda

7 changed files with 8745 additions and 0 deletions

Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL

Download Patch File Download Diff File Expand all files Collapse all files

2455

examples/python/CuTeDSL/distributed/distributed_gemm_all_reduce_blackwell.py Normal file

File diff suppressed because it is too large Load Diff