Skip to content

Releases: fanshiqing/grouped_gemm

Add streams sync to multi-stream cublas.

25 Apr 13:37
@fanshiqing fanshiqing
v1.1.1
8040918

Choose a tag to compare

View all tags
Add streams sync to multi-stream cublas. Pre-release
Pre-release
  • Add streams sync to multi-stream cublas.
Loading

Optimized permute/unpermute kernels for topk router.

19 Apr 07:43
@fanshiqing fanshiqing
v1.1
fa17d40

Choose a tag to compare

View all tags
Optimized permute/unpermute kernels for topk router. Pre-release
Pre-release

Optimized permute/unpermute kernels for topk router.

Loading

Initial release

01 Apr 09:51
@fanshiqing fanshiqing
v1.0
7a7f018

Choose a tag to compare

View all tags
Initial release Pre-release
Pre-release
  • Megablocks based gmm;
  • Multi-stream cublas gemm for sm90;
  • permute/unpermute kernel;
  • sinkhorn kernel.
Loading