//===- TestCFGLoopInfo.cpp - Test CFG loop info analysis ------------------===// // // 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 logic for testing the CFGLoopInfo analysis. // //===----------------------------------------------------------------------===// #include "mlir/Analysis/CFGLoopInfo.h" #include "mlir/Interfaces/FunctionInterfaces.h" #include "mlir/Pass/Pass.h" using namespace mlir; namespace { /// A testing pass that applies the CFGLoopInfo analysis on a region and prints /// the information it collected to llvm::errs(). struct TestCFGLoopInfo : public PassWrapper> { MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(TestCFGLoopInfo) StringRef getArgument() const final { return "test-cfg-loop-info"; } StringRef getDescription() const final { return "Test the loop info analysis."; } void runOnOperation() override; }; } // namespace void TestCFGLoopInfo::runOnOperation() { auto func = getOperation(); DominanceInfo &domInfo = getAnalysis(); Region ®ion = func.getFunctionBody(); // Prints the label of the test. llvm::errs() << "Testing : " << func.getNameAttr() << "\n"; if (region.empty()) { llvm::errs() << "empty region\n"; return; } // Print all the block identifiers first such that the tests can match them. llvm::errs() << "Blocks : "; region.front().printAsOperand(llvm::errs()); for (auto &block : region.getBlocks()) { llvm::errs() << ", "; block.printAsOperand(llvm::errs()); } llvm::errs() << "\n"; if (region.getBlocks().size() == 1) { llvm::errs() << "no loops\n"; return; } llvm::DominatorTreeBase &domTree = domInfo.getDomTree(®ion); mlir::CFGLoopInfo loopInfo(domTree); if (loopInfo.getTopLevelLoops().empty()) llvm::errs() << "no loops\n"; else loopInfo.print(llvm::errs()); } namespace mlir { namespace test { void registerTestCFGLoopInfoPass() { PassRegistration(); } } // namespace test } // namespace mlir