1. 11 Jun, 2025 2 commits
    • Chenggang Zhao's avatar
      Use `pynvml` to detect NVLink connections (#205) · 9ec06120
      Chenggang Zhao authored
      * Use `pynvml` to detect NVLink connections
      
      * Add a TODO
      
      * Add shutdown
      
      * Fix comments
      9ec06120
    • Chenggang Zhao's avatar
      Support Ampere architecture (#204) · b8d90fb7
      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
      b8d90fb7
  2. 25 Feb, 2025 1 commit