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

..

test_internode.py

Refactor some code.

2025-04-22 10:22:30 +08:00

test_intranode.py

Support Ampere architecture (#204)

2025-06-11 15:48:18 +08:00

test_low_latency.py

Support statistics tensor for low-latency kernels (#196)

2025-06-09 15:50:56 +08:00

utils.py

Initial commit

2025-02-25 09:07:53 +08:00