Commit 332b783

committed

add all reduce warp group

1 parent 50319b2 commit 332b783Copy full SHA for 332b783

2 files changed

+1010

-17

lines changed

flashinfer/cute_dsl
- blockscaled_gemm.py
tests
- test_cute_dsl_blockscaled_gemm_allreduce_two_shot.py

2 files changed

+1010

-17

lines changed

Comments

(0)