Wang Yixuan c68ddc11ce [OPS] add bmm_transpose ops (#3990) ...

### What this PR does / why we need it?
Add a new fusion ops to custom_op, which can cobime the torch.bmm() and
transpsose to achieve better peformance. This ops is used in mla_v1 to
replace the bmm and transpose

### Does this PR introduce _any_ user-facing change?
No

### How was this patch tested?


- vLLM version: v0.11.2

---------

Signed-off-by: hust17yixuan <303660421@qq.com>

2025-12-01 09:09:51 +08:00

..

tiling_data.cpp

[OPS] add bmm_transpose ops (#3990)

2025-12-01 09:09:51 +08:00

tiling_data.h

[OPS] add bmm_transpose ops (#3990)

2025-12-01 09:09:51 +08:00