5 Commits

Author SHA1 Message Date
d7ba0be4ed 修复了一个无关紧要的warning 2026-02-15 03:48:44 +08:00
Vanka0051
a963ca3d0b
Merge branch 'main' into perf/gumbel_softmax_sampler 2025-11-07 15:32:22 +08:00
PAN
a3b884ff6f feat: gumbel_softmax_sampler
Signed-off-by: PAN <1162953505@qq.com>
2025-11-06 13:12:11 +08:00
storyicon
3c360273da feat: add batch support
Signed-off-by: storyicon <storyicon@foxmail.com>
2025-11-06 05:00:28 +00:00
storyicon
c1ef4148af feat: achieve inference acceleration for the gpt2 stage
Signed-off-by: storyicon <storyicon@foxmail.com>
2025-10-24 08:15:00 +00:00