4 Commits

Author	SHA1	Message	Date
hlu1	8207d5fd39	[None] [feat] Add model gpt-oss (#6645) ... Signed-off-by: Hao Lu <14827759+hlu1@users.noreply.github.com>	2025-08-07 03:04:18 -04:00
yunruis	b99c5ce8c1	Feat/ds r1 min latency opt round3, add router gemm, fused a gemm, PDL (#4560) ... Signed-off-by: yunruis <yunruis@nvidia.com> Signed-off-by: kduan <176893526+Kefeng-Duan@users.noreply.github.com> Signed-off-by: Kefeng-Duan <176893526+Kefeng-Duan@users.noreply.github.com> Co-authored-by: kduan <176893526+Kefeng-Duan@users.noreply.github.com>	2025-06-14 17:36:22 +08:00
Yukun He	0ae7017342	Unify two versions of AllReduce custom op (#3032) ... * Rewrite unit test for unified allreduce op. Removing the legacy unit test. * Revise formats, fusion_op bindings. Put all tensors as optional inputs. * Move the MoeAllreduceOp to a separate custom op. * Move all the fusion patterns to the new version of the AllReduce fusion kernel. Remove the AllReduce strategy config. Revise the AllReduce strategies and fusion pattern definitions. * Add more TODOs, fixing minor bugs, and remove legacy code. Signed-off-by: Yukun He <23156053+hyukn@users.noreply.github.com>	2025-04-22 21:58:42 +08:00
Void	316e5c3be3	feat: fix and improve allreduce and fusion kernels (#3064) ... Signed-off-by: Yilin Zhang <18275976+yilin-void@users.noreply.github.com>	2025-04-08 19:33:52 +08:00