//= TestAffineLoopParametricTiling.cpp -- Parametric Affine loop tiling pass =// // // 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 test pass to test parametric tiling of perfectly // nested affine for loops. // //===----------------------------------------------------------------------===// #include "mlir/Dialect/Affine/IR/AffineOps.h" #include "mlir/Dialect/Affine/LoopUtils.h" #include "mlir/Dialect/Affine/Passes.h" #include "mlir/Dialect/Func/IR/FuncOps.h" usingnamespacemlir; usingnamespacemlir::affine; #define DEBUG_TYPE … namespace { struct TestAffineLoopParametricTiling : public PassWrapper<TestAffineLoopParametricTiling, OperationPass<func::FuncOp>> { … }; } // namespace /// Checks if the function enclosing the loop nest has any arguments passed to /// it, which can be used as tiling parameters. Assumes that atleast 'n' /// arguments are passed, where 'n' is the number of loops in the loop nest. static LogicalResult checkIfTilingParametersExist(ArrayRef<AffineForOp> band) { … } /// Captures tiling parameters, which are expected to be passed as arguments /// to the function enclosing the loop nest. Also checks if the required /// parameters are of index type. This approach is temporary for testing /// purposes. static LogicalResult getTilingParameters(ArrayRef<AffineForOp> band, SmallVectorImpl<Value> &tilingParameters) { … } void TestAffineLoopParametricTiling::runOnOperation() { … } namespace mlir { namespace test { void registerTestAffineLoopParametricTilingPass() { … } } // namespace test } // namespace mlir