rasmith 1d75472626 [BugFix] [Kernel] Fix GPU SEGV occuring in fused_moe kernel (#10385) ...

Signed-off-by: Randall Smith <Randall.Smith@amd.com>

2024-11-16 09:55:05 +00:00

..

configs

[Kernel] adding fused moe kernel config for L40S TP4 (#9245)

2024-10-11 08:54:22 -07:00

__init__.py

[torch.compile] support moe models (#9632)

2024-10-27 21:58:04 -07:00

fused_marlin_moe.py

[torch.compile] directly register custom op (#9896)

2024-10-31 21:56:09 -07:00

fused_moe.py

[BugFix] [Kernel] Fix GPU SEGV occuring in fused_moe kernel (#10385)

2024-11-16 09:55:05 +00:00

layer.py

[torch.compile] support moe models (#9632)

2024-10-27 21:58:04 -07:00

moe_pallas.py

[Hardware][TPU] Support MoE with Pallas GMM kernel (#6457)

2024-07-16 09:56:28 -07:00