================ @@ -8,6 +8,7 @@ #include "mlir/Dialect/Utils/ReshapeOpsUtils.h" +#include "mlir/Dialect/Arith/IR/Arith.h" ---------------- Shukla-Gaurav wrote: Moved to Arith/Utils. Thanks :) https://github.com/llvm/llvm-project/pull/90040