1//===- RegionsFromBBs.cpp - A helper to test RegionPasses -----------------===//
2//
3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4// See https://llvm.org/LICENSE.txt for license information.
5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6//
7//===----------------------------------------------------------------------===//
8
9#include "llvm/Transforms/Vectorize/SandboxVectorizer/Passes/RegionsFromBBs.h"
10#include "llvm/SandboxIR/Function.h"
11#include "llvm/Transforms/Vectorize/SandboxVectorizer/RegionWithScore.h"
12#include "llvm/Transforms/Vectorize/SandboxVectorizer/SandboxVectorizerPassBuilder.h"
13
14namespace llvm::sandboxir {
15
16RegionsFromBBs::RegionsFromBBs(StringRef Pipeline, StringRef AuxArg)
17 : FunctionPass("regions-from-bbs"),
18 RPM("rpm", Pipeline, SandboxVectorizerPassBuilder::createRegionPass) {}
19
20bool RegionsFromBBs::runOnFunction(Function &F, const Analyses &A) {
21 SmallVector<std::unique_ptr<RegionWithScore>, 16> Regions;
22 // Create a region for each BB.
23 for (BasicBlock &BB : F) {
24 Regions.push_back(
25 Elt: std::make_unique<RegionWithScore>(args&: F.getContext(), args&: A.getTTI()));
26 auto &RgnPtr = Regions.back();
27 for (Instruction &I : BB)
28 RgnPtr->add(I: &I);
29 }
30 // For each region run the region pass pipeline.
31 for (auto &RgnPtr : Regions)
32 RPM.runOnRegion(R&: *RgnPtr, A);
33 return false;
34}
35
36} // namespace llvm::sandboxir
37