4 Commits

Author	SHA1	Message	Date
Jinjie Liu	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