Yang Chen 95460fc513 [Kernel] port sgl moe_align_block_size kernels (#12574) ...

sgl_moe_align_block_size is based on:


ded9fcd09a

moe_align_block_size is based on:


ba5112ff69

Signed-off-by: Yang Chen <yangche@fb.com>

2025-02-03 13:09:50 +08:00

..

marlin_kernels

Update pre-commit hooks (#12475)

2025-01-27 17:23:08 -07:00

marlin_moe_ops.cu

[Bugfix] Fix support for dimension like integers and ScalarType (#9299)

2024-10-17 19:08:34 +00:00

moe_align_sum_kernels.cu

[Kernel] port sgl moe_align_block_size kernels (#12574)

2025-02-03 13:09:50 +08:00

moe_ops.h

[Kernel] port sgl moe_align_block_size kernels (#12574)

2025-02-03 13:09:50 +08: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] port sgl moe_align_block_size kernels (#12574)

2025-02-03 13:09:50 +08:00