13 Commits

Author	SHA1	Message	Date
bnellnm	c1909e7e8c	[Kernels] MoE refactor (#19636) ... Signed-off-by: Bill Nell <bnell@redhat.com> Signed-off-by: ElizaWszola <ewszola@redhat.com> Co-authored-by: ElizaWszola <ewszola@redhat.com>	2025-07-02 06:08:27 -07:00
bnellnm	015fab8c2f	[Kernels][Bugfix] Use torch op for all kernels in FusedMoE forward. Add additional testing for cudagraphs. (#19717) ... Signed-off-by: Bill Nell <bnell@redhat.com>	2025-06-24 23:22:58 -07:00
bnellnm	29fa5cac1c	[Kernels] Add activation chunking logic to FusedMoEModularKernel (#19168) ... Signed-off-by: Bill Nell <bnell@redhat.com>	2025-06-11 12:53:10 -04:00
Simon Mo	02f0c7b220	[Misc] Add SPDX-FileCopyrightText (#19100) ... Signed-off-by: simon-mo <simon.mo@hey.com>	2025-06-03 11:20:17 -07:00
bnellnm	c6c10ca920	[Bugfix] Reduce moe_sum test size to avoid OOM (#18484) ... Signed-off-by: Bill Nell <bnell@redhat.com>	2025-05-21 06:46:39 -07:00
bnellnm	92247c522e	[Bug] Fix moe_sum signature (#18440) ... Signed-off-by: Bill Nell <bnell@redhat.com>	2025-05-20 22:37:08 -07:00
bnellnm	f9c069c85e	Modularize fused experts and integrate PPLX kernels (#15956)	2025-05-14 13:11:54 -07:00
TJian	612c2edb4f	[FEAT] [ROCm]: Add AITER CK 2 Stages MoE support (#17110) ... Signed-off-by: tjtanaa <tunjian.tan@embeddedllm.com> Co-authored-by: Gregory Shtrasberg <Gregory.Shtrasberg@amd.com>	2025-05-14 03:03:11 -07:00
Jinzhen Lin	d74e5f37bc	[Kernel] fp4 marlin kernel (#17687) ... Signed-off-by: Jinzhen Lin <linjinzhen@hotmail.com>	2025-05-10 19:58:49 -07:00
Cyrus Leung	6e5595ca39	[CI/Build] Automatically retry flaky tests (#17856) ... Signed-off-by: DarkLight1337 <tlleungac@connect.ust.hk>	2025-05-09 09:55:17 -06:00
Jinzhen Lin	1d0c9d6b2d	[Kernel] some optimizations for dense marlin and moe marlin (#16850) ... Signed-off-by: Jinzhen Lin <linjinzhen@hotmail.com>	2025-05-05 09:39:30 -07:00
Caleb_Du	3e887d2e0c	permute/unpermute kernel for moe optimization (#14568) ... Signed-off-by: Caleb_Du <Caleb_Du@zju.edu.cn>	2025-05-02 11:31:55 -07:00
Michael Goin	6317a5174a	Categorize `tests/kernels/` based on kernel type (#16799) ... Signed-off-by: mgoin <mgoin64@gmail.com>	2025-04-23 09:21:07 -04:00