diff --git a/mlir/include/mlir/Dialect/Linalg/Utils/Utils.h b/mlir/include/mlir/Dialect/Linalg/Utils/Utils.h --- a/mlir/include/mlir/Dialect/Linalg/Utils/Utils.h +++ b/mlir/include/mlir/Dialect/Linalg/Utils/Utils.h @@ -208,6 +208,11 @@ SmallVector distributionMethod = {}; }; +/// Update the `lb`, `ub` and `step` to get per processor `lb`, `ub` and `step`. +void updateBoundsForCyclicDistribution(OpBuilder &builder, Location loc, + Value procId, Value nprocs, Value &lb, + Value &ub, Value &step); + //===----------------------------------------------------------------------===// // Generic op region utilities //===----------------------------------------------------------------------===// diff --git a/mlir/lib/Dialect/Linalg/Utils/Utils.cpp b/mlir/lib/Dialect/Linalg/Utils/Utils.cpp --- a/mlir/lib/Dialect/Linalg/Utils/Utils.cpp +++ b/mlir/lib/Dialect/Linalg/Utils/Utils.cpp @@ -241,10 +241,9 @@ } /// Update the `lb`, `ub` and `step` to get per processor `lb`, `ub` and `step`. -static void updateBoundsForCyclicDistribution(OpBuilder &builder, Location loc, - Value procId, Value nprocs, - Value &lb, Value &ub, - Value &step) { +void updateBoundsForCyclicDistribution(OpBuilder &builder, Location loc, + Value procId, Value nprocs, Value &lb, + Value &ub, Value &step) { using edsc::op::operator+; using edsc::op::operator*; lb = lb + (procId * step);