Commits

Commits on Oct 16, 2024

Update __init__.py (#45 )
zheng-ningxin
authored

Commits on Sep 18, 2024

Update __init__.py (#41 )
zheng-ningxin
authored

Commits on Jul 25, 2024

Commits on Jul 22, 2024

add torch version to the whl name (#24 )
zheng-ningxin
authored

Commits on Jul 19, 2024

Support performance tunning for gemm-rs kernel on sm80l (#23 )
zheng-ningxin
authored

Commits on Jul 18, 2024

Remove pynvshmem import in gemm_rs_80.py

tlrmchlsmth
authored and
houqi
committed

Commits on Jul 16, 2024

Tune the AG performance for the llama-8b (#21 )
zheng-ningxin
authored

Commits on Jul 15, 2024

feat: fix tuning for the all-gather gemm && move the reset-signal() to the forward critical path (#19 )
zheng-ningxin
authored

Commits on Jul 9, 2024

zero out all the allocated shm buffer (#18 )
zheng-ningxin
authored

Commits on Jul 4, 2024

Commits on Jul 1, 2024

fix the _allgather_base backend issue(issue11) (#12 )
zheng-ningxin
authored

Commits on Jun 27, 2024

Support IPC && SM90 version of AG-GEMM, GEMM-RS (#9 )

authored

Commits on Jun 14, 2024

Update README.md
liwenchangbdbz
authored

Commits on Jun 13, 2024

Update bibtex (#6 )
wenlei-bao
authored

Commits on Jun 12, 2024

Add arxiv paper link (#5 )
wenlei-bao
authored

Commits on Jun 7, 2024

Reorgnize and deduplicate files (#4 )
wenlei-bao
authored

Commits on Jun 6, 2024

Sm80 Allgather-gemm && Gemm-Reducescatter

authored

Commits on Mar 2, 2024

add cutlass submodule and patches (#2 )
liwenchangbdbz
authored

Commits on Mar 1, 2024