search

iree-org / iree-turbine

IREE's PyTorch Frontend, based on Torch Dynamo.
Apache License 2.0
55 stars 25 forks source link

[TKW] Fix attention kernel to do non transposed output #245

Closed raikonenfnu closed 2 weeks ago

raikonenfnu commented 2 weeks ago

Turns out ordering of mapping matters