Tong WU

icon indicating a website link https://rachmanino.github.io

icon company Peking University icon location Beijing icon location Senior undergraduate@PKU, beginner in MLSys@Tile-AI

Results 2 issues of Tong WU

[Enhancement] Enhance async proxy analyzing logic and GEMM dispatching

3 comment

- Added a mapping for GEMM instruction prefixes in `gemm.h`. - Renamed GEMM functions to include `mma_` prefix for clarity in `gemm_mma.h`, `gemm_sm70.h`, `gemm_sm90.h`, and `gemm_sp_sm80.h`. - Updated function signatures...

[Feature] Add `tl.infinity` operator for infinity handling of bfloat16

3 comment

As title, fix #1151 ## Summary by CodeRabbit * **New Features** * Support for producing Infinity values for float64, float32, float16, and bfloat16. * **Refactor** * Infinity construction now routes...