transformers
b9f0074f - XPU now supports MoE kernel(MegaBlocks) implementation (#43435)

Commit
25 days ago
XPU now supports MoE kernel(MegaBlocks) implementation (#43435) * Add moe kernel for XPU * Support xpu moe kernel tests * Style * Update tests/models/gpt_oss/test_modeling_gpt_oss.py Co-authored-by: Mohamed Mekkouri <93391238+MekkCyber@users.noreply.github.com> --------- Co-authored-by: Mohamed Mekkouri <93391238+MekkCyber@users.noreply.github.com>
Author
Parents
Loading