[Mlir-commits] [mlir] 12c3bca - [mlir] Rename linalg.pooling operations to have a FOp postfix for floating point

Rob Suderman llvmlistbot at llvm.org
Fri Apr 2 12:00:17 PDT 2021


Author: Rob Suderman
Date: 2021-04-02T11:54:20-07:00
New Revision: 12c3bca1934a5eb0e23f6265d8fcc11ad6f4dea5

URL: https://github.com/llvm/llvm-project/commit/12c3bca1934a5eb0e23f6265d8fcc11ad6f4dea5
DIFF: https://github.com/llvm/llvm-project/commit/12c3bca1934a5eb0e23f6265d8fcc11ad6f4dea5.diff

LOG: [mlir] Rename linalg.pooling operations to have a FOp postfix for floating point

Linalg pooling operations only support floating point currently but integer
variants will soon be needed. Renaming to uncluse a FOp postfix to clarify.

Differential Revision: https://reviews.llvm.org/D99779

Added: 
    

Modified: 
    mlir/include/mlir/Dialect/Linalg/IR/LinalgNamedStructuredOpsSpec.tc

Removed: 
    


################################################################################
diff  --git a/mlir/include/mlir/Dialect/Linalg/IR/LinalgNamedStructuredOpsSpec.tc b/mlir/include/mlir/Dialect/Linalg/IR/LinalgNamedStructuredOpsSpec.tc
index 66c7978d4252..ae29a1fe6fd9 100644
--- a/mlir/include/mlir/Dialect/Linalg/IR/LinalgNamedStructuredOpsSpec.tc
+++ b/mlir/include/mlir/Dialect/Linalg/IR/LinalgNamedStructuredOpsSpec.tc
@@ -341,7 +341,7 @@ order of (`N`, `F`, `D`, `H`, `W`, `KD`, `KH`, `KW`, `C`).
                                  K(kd, kh, kw, c, f)));
 }
 
-ods_def<PoolingNHWCSumOp>:
+ods_def<PoolingNHWCSumFOp>:
 def pooling_nhwc_sum
     (I: f32(N, H, W, C), K: f32(KH, KW))
   -> (O: f32(N, OH, OW, C))
@@ -352,7 +352,7 @@ def pooling_nhwc_sum
                                        ow * strides[1] + kw * dilations[1], c));
 }
 
-ods_def<PoolingNHWCMaxOp>:
+ods_def<PoolingNHWCMaxFOp>:
 def pooling_nhwc_max
     (I: f32(N, H, W, C), K: f32(KH, KW))
   -> (O: f32(N, OH, OW, C))
@@ -367,7 +367,7 @@ def pooling_nhwc_max
                          O(n, oh, ow, c));
 }
 
-ods_def<PoolingNHWCMinOp>:
+ods_def<PoolingNHWCMinFOp>:
 def pooling_nhwc_min
     (I: f32(N, H, W, C), K: f32(KH, KW))
   -> (O: f32(N, OH, OW, C))


        


More information about the Mlir-commits mailing list