7 Commits

Author	SHA1	Message	Date
Chenggang Zhao	b8d90fb753	Support Ampere architecture (#204) ... * 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	2025-06-11 15:48:18 +08:00
Chenggang Zhao	8da2d7b38d	Fully remove barrier FIFO designs (#200) ... * Fully remove FIFO slots * Fully remove FIFO buffers * Minor fix styles * Fix some typos * Bugs fixed * Cleanup `ibgda_poll_cq`	2025-06-10 16:23:20 +08:00
Shangyan Zhou	9fe9021f29	Use IBGDA only (#177)	2025-05-28 16:40:14 +08:00
Shangyan Zhou	3e54b78fd7	Normal kernels always use IBGDA mode.	2025-04-22 10:36:24 +08:00
moningchen	5ab80c28f3	In the Internode Normal Kernel, when using nvshmem ibrc for RDMA data transmission, a single QP is used for data transfer between two GPUs, which limits kernel performance in network card dual-port and RoCE network scenarios. ... In our optimized Internode Normal Kernel, we implemented multiple QPs for data transmission between two GPUs, setting a different QP for each channel. Additionally, we modified the transmission method from IBRC to IBGAD. Through these optimizations, the Internode Normal Kernel achieves optimal performance in both H800 and H20 environments, with RDMA transmission performance nearly reaching the physical network performance limit. Using the current default statistical method, in 4-node H800 and H20 environments, RDMA performance can reach 60GB/s+.	2025-04-21 15:50:39 +08:00
Shangyan Zhou	2d0cf41dd1	Low latency kernels use rdma atomic to support AR.	2025-03-14 11:04:57 +08:00
Chenggang Zhao	ebfe47e46f	Initial commit	2025-02-25 09:07:53 +08:00