================ @@ -13,6 +13,8 @@ #include "mlir/Dialect/Linalg/Utils/Utils.h" #include "mlir/Transforms/DialectConversion.h" +#include "llvm/ADT/SmallVectorExtras.h" + ---------------- hanhanW wrote: ditto https://github.com/llvm/llvm-project/pull/178526