Tyler Michael Smith d66ac62854

[Kernel][Bugfix] Delete some more useless code in marlin_moe_ops.cu (#8643)

2024-09-21 23:45:02 +00:00

..

marlin_moe_ops.cu

[Kernel][Bugfix] Delete some more useless code in marlin_moe_ops.cu (#8643)

2024-09-21 23:45:02 +00:00

marlin_moe_ops.h

[Kernel] Enable 8-bit weights in Fused Marlin MoE (#8032)

2024-09-16 09:47:19 -06: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] Enable 8-bit weights in Fused Marlin MoE (#8032)

2024-09-16 09:47:19 -06:00