- 27 Jun, 2025 1 commit
-
-
Chenggang Zhao authored
-
- 26 Jun, 2025 1 commit
-
-
Shangyan Zhou authored
-
- 25 Jun, 2025 2 commits
-
-
Shangyan Zhou authored
* Support bias. * Fix. * Fix style.
-
Shangyan Zhou authored
* Add `get_comm_stream`. * Fix style.
-
- 24 Jun, 2025 3 commits
-
-
Chenggang Zhao authored
-
Chenggang Zhao authored
* Add draft * Add fast-debugging flags * Fix several bugs * Add sender timeout checks * Fix stuck * Fix bugs * Fix bugs
-
Shangyan Zhou authored
* Increase the test round. * Add warp synchronization. * Shuffle the send warps. * Add time elapsed into bench result.
-
- 23 Jun, 2025 2 commits
- 20 Jun, 2025 1 commit
-
-
Chenggang Zhao authored
-
- 18 Jun, 2025 6 commits
-
-
Chenggang Zhao authored
-
Chenggang Zhao authored
-
Shangyan Zhou authored
Set `device_id` to suppress pytorch warning.
-
Shangyan Zhou authored
-
Shangyan Zhou authored
-
Shangyan Zhou authored
* Fix the tail loading issue. * Modify the sync offset.
-
- 16 Jun, 2025 4 commits
-
-
Shangyan Zhou authored
* Fix warp synchronization. * Another fix.
-
Chenggang Zhao authored
-
Chenggang Zhao authored
* Add automatic warp count control for low-latency dispatch * Add automatic warp count control for low-latency combine * More assertions
-
fzyzcjy authored
-
- 13 Jun, 2025 2 commits
-
-
Shangyan Zhou authored
-
Zhicheng Wu authored
let the sender SM use the channel_id, and the receiver SM use channel_id + num_channels
-
- 12 Jun, 2025 1 commit
-
-
Shifang Xu authored
-
- 11 Jun, 2025 4 commits
-
-
Chenggang Zhao authored
* Use `pynvml` to detect NVLink connections * Add a TODO * Add shutdown * Fix comments
-
Chenggang Zhao authored
* Update README * Update `setup.py` * Fix headers * Add `DISABLE_NVSHMEM` for APIs * Fix launch * Fix TMA settings * Fix TMA usages * Fix dlink * Separate layout kernels * Update version * Add `is_sm90_compiled` * Fix tests * Add NVLink connection checks * Update README * Fix tests * Add some comments * Minor fix * Minor fix * Fix bugs
-
Chenggang Zhao authored
-
Chenggang Zhao authored
-
- 10 Jun, 2025 4 commits
-
-
Chenggang Zhao authored
* Fully remove FIFO slots * Fully remove FIFO buffers * Minor fix styles * Fix some typos * Bugs fixed * Cleanup `ibgda_poll_cq`
-
Shangyan Zhou authored
remove the dependency of gdrcopy
-
youkaichao authored
Signed-off-by:youkaichao <youkaichao@gmail.com>
-
youkaichao authored
Signed-off-by:youkaichao <youkaichao@gmail.com>
-
- 09 Jun, 2025 4 commits
-
-
Chenggang Zhao authored
-
Chenggang Zhao authored
-
Chenggang Zhao authored
* Add low-latency kernel usage flag * Update comments
-
Chenggang Zhao authored
-
- 08 Jun, 2025 1 commit
-
-
Shangyan Zhou authored
Allow using MNNVL
-
- 07 Jun, 2025 1 commit
-
-
fzyzcjy authored
-
- 06 Jun, 2025 2 commits
-
-
Chenggang Zhao authored
* Update CMake files * Use TMA instead of LD/ST for intranode dispatch * Use TMA instead of LD/ST for intranode combine * Adjust configs * Test default configs as well * More warps for combine * Add inter-thread fence * Enable more warps * Do not use TMA for senders * Update configs * Remove useless wait
-
Shangyan Zhou authored
Co-authored-by:Shangyan Zhou <sy.zhou@deepseek.com>
-
- 05 Jun, 2025 1 commit
-
-
Chenggang Zhao authored
-