//===- TestLinalgRankReduceContractionOps.cpp -----------------------------===// // // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. // See https://llvm.org/LICENSE.txt for license information. // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception // //===----------------------------------------------------------------------===// // // This file implements a pass for testing rank reduing patterns for named // contraction ops with unit dims. // //===----------------------------------------------------------------------===// #include "mlir/Dialect/Affine/IR/AffineOps.h" #include "mlir/Dialect/Func/IR/FuncOps.h" #include "mlir/Dialect/Linalg/Transforms/Transforms.h" #include "mlir/Pass/Pass.h" #include "mlir/Pass/PassManager.h" #include "mlir/Transforms/GreedyPatternRewriteDriver.h" usingnamespacemlir; namespace { struct TestLinalgRankReduceContractionOps : public PassWrapper<TestLinalgRankReduceContractionOps, OperationPass<func::FuncOp>> { … }; } // namespace namespace mlir { namespace test { void registerTestLinalgRankReduceContractionOps() { … } } // namespace test } // namespace mlir