JinjieLiu
|
e41ec26329
|
support attention fwd
Signed-off-by: Jinjie Liu <jjliu@baai.ac.cn>
|
2026-02-09 17:15:51 +08:00 |
|
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 |
|