.. |
align_block_size_kernel.cu
|
935027bdcc
feat: dynamic shared memory allocation for moe align block size (#372)
|
há 11 meses atrás |
moe_ops.cpp
|
7d6ba53602
feat: fused top-k kernels for MoE (#273)
|
há 1 ano atrás |
moe_ops.h
|
7d6ba53602
feat: fused top-k kernels for MoE (#273)
|
há 1 ano atrás |
softmax.cu
|
7d6ba53602
feat: fused top-k kernels for MoE (#273)
|
há 1 ano atrás |