[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