4 Commits

Author	SHA1	Message	Date
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
Jinjie Liu	b7bf598fde	support softmax ... Signed-off-by: Jinjie Liu <jjliu@baai.ac.cn>	2026-02-04 16:38:33 +08:00