Dipika Sikka 8678a69ab5

[Kernel] Expand MoE weight loading + Add Fused Marlin MoE Kernel (#7527) ...

Co-authored-by: ElizaWszola <eliza@neuralmagic.com>

2024-08-21 16:17:10 -07:00

..

marlin_moe_ops.cu

[Kernel] Expand MoE weight loading + Add Fused Marlin MoE Kernel (#7527)

2024-08-21 16:17:10 -07:00

marlin_moe_ops.h

[Kernel] Expand MoE weight loading + Add Fused Marlin MoE Kernel (#7527)

2024-08-21 16:17:10 -07:00

moe_ops.h

[Kernel][Misc] Use TORCH_LIBRARY instead of PYBIND11_MODULE for custom ops (#5047)

2024-06-09 16:23:30 -04:00

topk_softmax_kernels.cu

[Kernel][Misc] Use TORCH_LIBRARY instead of PYBIND11_MODULE for custom ops (#5047)

2024-06-09 16:23:30 -04:00

torch_bindings.cpp

[Kernel] Expand MoE weight loading + Add Fused Marlin MoE Kernel (#7527)

2024-08-21 16:17:10 -07:00