 Chenggang Zhao
|
6db7e1863b
|
Solve STSM bank conflict via padding and 3D TMA
|
2025-04-03 15:39:35 +08:00 |
|
|
Chenggang Zhao
|
25db8de345
|
Better performance
|
2025-03-25 17:34:06 +08:00 |
|
|
Chenggang Zhao
|
1999d553e5
|
Lower TMA requirement
|
2025-03-25 17:18:53 +08:00 |
|
|
Chenggang Zhao
|
046fab64b7
|
Fix grouped GEMM cases
|
2025-03-25 16:41:44 +08:00 |
|
|
Chenggang Zhao
|
3497428a5e
|
Minor fix
|
2025-03-25 15:16:26 +08:00 |
|
|
Chenggang Zhao
|
7ffb118e54
|
Support multicasting on B
|
2025-03-25 14:56:42 +08:00 |
|
|
Chenggang Zhao
|
b922e64cb2
|
Support block size 160
|
2025-03-25 13:37:59 +08:00 |
|
|
sazc
|
46eb0d08fb
|
Performance: Larger BlockTile optimizations enable 1470+ TFLOPS FP8 performance on the H800-SXM platform
|
2025-03-25 10:44:57 +08:00 |
|
|
Liang
|
e1c070fbef
|
Merge pull request #65 from Z-NAVY/main
Fix get_col_major_tma_aligned_tensor to handle 2-dimensional inputs
|
2025-03-14 13:50:08 +08:00 |
|
|
z-navy
|
3f92607b98
|
Fix get_col_major_tma_aligned_tensor to handle 2-dimensional inputs
|
2025-03-13 22:15:16 +08:00 |
|
|
fzyzcjy
|
e7fff7ef0a
|
Update m_grouped_gemm.py
|
2025-03-13 22:09:15 +08:00 |
|
|
Chenggang Zhao
|
5e4badc577
|
Fix type lint
|
2025-03-10 13:10:16 +08:00 |
|
|
sazc
|
bed67b234c
|
Minor fix
|
2025-03-10 13:02:02 +08:00 |
|
|
sazc
|
ed278eddd3
|
formats: Optimize get_best_configs implementation
|
2025-03-10 12:56:14 +08:00 |
|
|
sazc
|
50cf26cc7c
|
Performance: Configuration algorithms tuned to minimize the impact of tail effects, now up to 1402 TFLOPS
|
2025-03-10 11:45:05 +08:00 |
|
|
Liang
|
ded740f736
|
Fix documentation of m_grouped_gemm_fp8_fp8_bf16_nt_contiguous in m_grouped_gemm.py
|
2025-03-04 11:26:23 +08:00 |
|
|
Chenggang Zhao
|
6c5da03ba9
|
Support more shapes
|
2025-02-28 10:04:59 +08:00 |
|
|
Chenggang Zhao
|
b69f630b91
|
Minor fix util function
|
2025-02-28 09:46:38 +08:00 |
|
|
Chenggang Zhao
|
6e10cba207
|
Minor fix
|
2025-02-28 09:21:35 +08:00 |
|
|
Liang
|
fbec9e5eee
|
Update get_best_configs
a better strategy to choose config
|
2025-02-27 23:18:52 +08:00 |
|
|
Chenggang Zhao
|
6da94d2d36
|
Add extra TMA checks
|
2025-02-27 18:20:57 +08:00 |
|
|
AcraeaTerpsicore
|
96b31fd6bb
|
fix typo
|
2025-02-26 18:37:22 +08:00 |
|
|
Chenggang Zhao
|
a6d97a1c1b
|
Initial commit
|
2025-02-25 22:52:41 +08:00 |
|