jinjieliu
|
213e4fc060
|
use templates to substitute parts of macros
Signed-off-by: jinjieliu <jinjie.liu@usc.edu>
|
2026-02-08 22:24:12 +08:00 |
|
jinjieliu
|
6a19a6b06d
|
put num_warps and num_stages in kwargs
Signed-off-by: jinjieliu <jinjie.liu@usc.edu>
|
2026-02-07 14:25:10 +08:00 |
|
jinjieliu
|
2298b6f8c8
|
support mm and autotune
Signed-off-by: jinjieliu <jinjie.liu@usc.edu>
|
2026-02-07 00:41:23 +08:00 |
|
JinjieLiu
|
b7bf598fde
|
support softmax
Signed-off-by: Jinjie Liu <jjliu@baai.ac.cn>
|
2026-02-04 16:38:33 +08:00 |
|