DeepSeek/DeepEP
1
0
Fork 0
mirror of https://github.com/deepseek-ai/DeepEP synced 2025-06-26 18:28:11 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity
122 Commits 8 Branches 0 Tags 4.5 MiB
bd429ffefc
Commit Graph

3 Commits

Author SHA1 Message Date
Chenggang Zhao
9ec061204e
Use pynvml to detect NVLink connections (#205) 
* Use `pynvml` to detect NVLink connections

* Add a TODO

* Add shutdown

* Fix comments
 2025-06-11 17:29:00 +08:00
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
ebfe47e46f Initial commit 2025-02-25 09:07:53 +08:00