iree-org / iree

A retargetable MLIR-based machine learning compiler and runtime toolkit.
http://iree.dev/
Apache License 2.0
2.57k stars 574 forks source link

Remove attention transform dialect e2e tests #17682

Closed Groverkss closed 3 months ago

Groverkss commented 3 months ago

With https://github.com/iree-org/iree/pull/17681 , the state of attention codegen for llvm-cpu is at a good enough place where we don't need to rely on a transform dialect spec (at least for llvm-cpu) anymore.

This patch removes e2e tests for the attention transform dialect spec as it adds more burden on a path that will probably not be maintained in the future. There are some e2e correctness tests in e2e/linalg_ext/ that check correctness for small test cases and pkgci running sdxl + attention tests.