llvm-project
d9d62331 - [mlir][linalg] Add a new helper hook: `hasVectorizationImpl` (#110708)

Commit
1 year ago
[mlir][linalg] Add a new helper hook: `hasVectorizationImpl` (#110708) The newly added hook simply returns `false` for Ops for which there's no "vectorization logic" in the Linalg Vectorizer (i.e. the `vectorize()` method). It's added so that the following two TD ops expose identical level of functionality (that's not the case ATM): * `transform.structured.vectorize_children_and_apply_patterns` * `transform.structured.vectorize` Specifically, ATM, the former works only for Linalg Ops, while the latter works for all Ops that the vectorizer supports (*). With this change, I am making sure that both TD will behave consistently. Note, this shouldn't affect any of the current uses of the vectorizer. (*) This is implemented via the `vectorize()` method in Vectorization.cpp.
Author
Parents
Loading