Neil Zhu 4403a21d4b [Metax] refactor cutlass moe and optimize flash attention (#5361) ...

* [Metax] refactor moe and flash attention backend
---------

Co-authored-by: zhangchenyi_dl <16219492+zhangchenyidl@user.noreply.gitee.com>

2025-12-10 17:15:17 +08:00

..

air_top_p_sampling.cu

[Metax] modify wrapSize to WARP_SIZE (#5442)

2025-12-09 01:44:02 -08:00

min_p_sampling_from_probs.cu

[Feature] support min_p_sampling (#2872)

2025-07-20 23:17:59 -07:00

rejection_top_p_sampling.cu

[OPs] Universal optimization and Fix early_stop cuda 700 (#3375)

2025-08-14 22:40:44 +08:00

sampling.cuh

[Metax] refactor cutlass moe and optimize flash attention (#5361)

2025-12-10 17:15:17 +08:00

top_k_renorm_probs.cu

[Feature] support top_k_top_p sampling (#2753)

2025-07-09 20:58:58 -07:00

utils.cuh

[Metax] refactor cutlass moe and optimize flash attention (#5361)

2025-12-10 17:15:17 +08:00