-
Notifications
You must be signed in to change notification settings - Fork 51
Open
Labels
help wantedExtra attention is neededExtra attention is needed
Description
We would like to cover more operators from the following sources:
- Tilelang benchmark: https://github.com/tile-ai/tilelang/tree/main/examples, deepseek_deepgemm, deepseek_mla, linear_attention
- gemm_sm100
- deepseek_deepgemm
- deepseek_mla
- linear_attention: https://github.com/meta-pytorch/tritonbench/pull/567/files
-
Mojo mha_sm100: https://github.com/modular/modular/blob/main/max/kernels/src/nn/mha_sm100.mojo
-
vLLM custom Triton kernels from MoE
Metadata
Metadata
Assignees
Labels
help wantedExtra attention is neededExtra attention is needed