[Mlir-commits] [mlir] 67f60bc - [mlir] Expose `updateBoundsForCyclicDistribution` in Linalg/Utils.h.
Alexander Belyaev
llvmlistbot at llvm.org
Thu Apr 15 11:47:42 PDT 2021
Author: Alexander Belyaev
Date: 2021-04-15T20:47:37+02:00
New Revision: 67f60bcc75a37c3c6a3f0037959b212e449a1a64
URL: https://github.com/llvm/llvm-project/commit/67f60bcc75a37c3c6a3f0037959b212e449a1a64
DIFF: https://github.com/llvm/llvm-project/commit/67f60bcc75a37c3c6a3f0037959b212e449a1a64.diff
LOG: [mlir] Expose `updateBoundsForCyclicDistribution` in Linalg/Utils.h.
Differential Revision: https://reviews.llvm.org/D100580
Added:
Modified:
mlir/include/mlir/Dialect/Linalg/Utils/Utils.h
mlir/lib/Dialect/Linalg/Utils/Utils.cpp
Removed:
################################################################################
diff --git a/mlir/include/mlir/Dialect/Linalg/Utils/Utils.h b/mlir/include/mlir/Dialect/Linalg/Utils/Utils.h
index 4ee63a92875f3..ba41e86fb79b5 100644
--- a/mlir/include/mlir/Dialect/Linalg/Utils/Utils.h
+++ b/mlir/include/mlir/Dialect/Linalg/Utils/Utils.h
@@ -208,6 +208,11 @@ struct LinalgLoopDistributionOptions {
SmallVector<DistributionMethod, 0> 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
index 8c8b0cf1f7bf5..1aa47ea3ac19c 100644
--- a/mlir/lib/Dialect/Linalg/Utils/Utils.cpp
+++ b/mlir/lib/Dialect/Linalg/Utils/Utils.cpp
@@ -241,10 +241,9 @@ void GenerateLoopNest<AffineForOp>::doit(
}
/// 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);
More information about the Mlir-commits
mailing list