tree 503a2a39c7e4bec448361755b335b38713b9cf62
parent 6b995b95694fa97cb13765cd556b0d450bf894c2
author Han-Chung Wang <hanhan0912@gmail.com> 1709055338 -0800
committer GitHub <noreply@github.com> 1709055338 -0800
gpgsig -----BEGIN PGP SIGNATURE-----
 
 wsFcBAABCAAQBQJl3h1qCRC1aQ7uu5UhlAAAXo8QABiY6c9VeC55i4otuYD5jFi7
 6okMaezFaHSz1BRThqPQlc/SBs3RI3lVEE3ulV4H6LCQcJUl/231oiE3iqMlw4fS
 tBnZMs9GODjxgF5HktjN/Q4BU9NiHNOrfknM6muNdVaVC1etTCVtSxi9zYvaGDPh
 GNoF1RnGHCDMFaZtPz+l1CWWlP9eqEu9quP1hRn/DLHJua5wCJw4JZRcOyny67v8
 tbZZ2UQLpnAvbo/1Yn3I1Zi4JHm+4mc8NcbcjS1xQ2Nq+/Qj6jlLO0uH6pVTGSWA
 3YP9d4vsgYEcqkuVPb1eaMt32ggzCgyB3kYuQHGlFCEddj76P0S1G2F4jzTq0ckT
 wheOAWPYEOBrAbOLZ9zZHC35pPdNOUWymDZqBjC+/aiicS5NOMszHjaNTL0Ml0nj
 4ZcI1+T5amAOfodt80j984F9lVh6tiCP7BXzFR3V1l5/Fh7DlZiS5zqXs/HH4gFs
 59hIPeQXrrWJ6l+23RLqLXOPamXhV7tDY5HprKcD7AU3YFaDY0lWgqCNxchJ5s+I
 5W7ou6o6ZQPofuY9YGrh43zzrKhWq3LeHiwtI08KQlPP3EVf38KhjVV4Q7PBq/ok
 hSp2YM5XwSC7fJDFH9m1ffzjbkyR5D81ZtEJbAaTRYm8THfzm5k/ChfV4ttGbMWa
 SkaaQ3kyCYBInBYPFPnC
 =hQRF
 -----END PGP SIGNATURE-----
 

[CPU] Add a specialized pipeline for LinalgExt::AttentionOp. (#16577)

The revision adds a new pipeline for LinalgExt ops. It is an
experimental pipeline, and should eventually get merged into
MultiTilingPipeline.

The new pipeline introduces vector level of tiling to LinalgExt, and
vectorization. Some dimension of attention op is not able to tile at
this moment, so we set all the tile sizes to 1 which avoids huge
vectors. Because the reduction dimension of matmuls is not tiled. Here
is selected IR dump:
https://gist.githubusercontent.com/hanhanW/db4511da681d4932cb81dd68cc98976f/raw/08c3cc42c9d7fb86b769f60dc712fecb9fb10700/dump.mlir

Towards https://github.com/openxla/iree/issues/16421