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

Signed-off-by: youkaichao <youkaichao@gmail.com>

2024-10-31 21:56:09 -07: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

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

2024-10-31 21:56:09 -07: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