Hook GridwiseGemmRewrite up, prepare to ditch GridwiseGemmEx.

llvm · Jun 6, 2020 · 47917d6 · 47917d6
1 parent 54287ab
commit 47917d6
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/mlir/include/mlir/Dialect/MIOpenOps/LowerMIOpenOps.h b/mlir/include/mlir/Dialect/MIOpenOps/LowerMIOpenOps.h
@@ -852,10 +852,10 @@ template struct Conv2DRewritePattern<miopen::Conv2DBwdDataOp>;
 // GridwiseGemm lowering.
 //===----------------------------------------------------------------------===//
 
-struct GridwiseGemmRewritePattern : public OpRewritePattern<miopen::GridwiseGemmExOp> {
-  using OpRewritePattern<miopen::GridwiseGemmExOp>::OpRewritePattern;
+struct GridwiseGemmRewritePattern : public OpRewritePattern<miopen::GridwiseGemmOp> {
+  using OpRewritePattern<miopen::GridwiseGemmOp>::OpRewritePattern;
 
-  PatternMatchResult matchAndRewrite(miopen::GridwiseGemmExOp op, PatternRewriter &b) const override {
+  PatternMatchResult matchAndRewrite(miopen::GridwiseGemmOp op, PatternRewriter &b) const override {
     // Prepare some useful constants.
     auto zeroConstantIndexOp = b.create<ConstantIndexOp>(op.getLoc(), 0);
     auto oneConstantIndexOp = b.create<ConstantIndexOp>(op.getLoc(), 1);