3df1c6f88b
Add an alternative to test VPlan in more isolation via a new `vplan-test-transform` option, which builds VPlan0 for each loop in the input IR and then can invoke a set of transforms on it. In order to allow different recipe types to be created, a new widen-from-metadata transform is added, which transforms VPInstructions to different recipes, based on custom !vplan.widen metadata. Currently this supports creating widen & replicate recipes, but can easily be extended in the future. Currently the handling is intentionally bare-bones, to be extended gradually as needed. PR: https://github.com/llvm/llvm-project/pull/178522
54 lines
1.8 KiB
C++
54 lines
1.8 KiB
C++
//===- VPlanTestPass.cpp - Test VPlan transforms -------------------------===//
|
|
//
|
|
// 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 pass is a lightweight testing harness for VPlan transforms. It builds
|
|
// VPlan0 for loops and runs specified transforms.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "llvm/Transforms/Vectorize/VPlanTestPass.h"
|
|
#include "VPlan.h"
|
|
#include "VPlanTransforms.h"
|
|
#include "llvm/Analysis/LoopInfo.h"
|
|
#include "llvm/Analysis/ScalarEvolution.h"
|
|
#include "llvm/Analysis/TargetLibraryInfo.h"
|
|
#include "llvm/IR/DebugLoc.h"
|
|
#include "llvm/IR/PassManager.h"
|
|
|
|
using namespace llvm;
|
|
|
|
PreservedAnalyses VPlanTestPass::run(Function &F, FunctionAnalysisManager &AM) {
|
|
if (TransformPipeline.empty())
|
|
return PreservedAnalyses::all();
|
|
|
|
auto &LI = AM.getResult<LoopAnalysis>(F);
|
|
auto &SE = AM.getResult<ScalarEvolutionAnalysis>(F);
|
|
auto &TLI = AM.getResult<TargetLibraryAnalysis>(F);
|
|
|
|
for (Loop *L : LI) {
|
|
PredicatedScalarEvolution PSE(SE, *L);
|
|
|
|
auto Plan = VPlanTransforms::buildVPlan0(
|
|
L, LI, Type::getInt64Ty(F.getContext()), DebugLoc(), PSE);
|
|
|
|
VPlanTransforms::runTestTransforms(*Plan, TransformPipeline, &TLI);
|
|
outs() << *Plan << "\n";
|
|
}
|
|
|
|
return PreservedAnalyses::all();
|
|
}
|
|
|
|
void VPlanTestPass::printPipeline(
|
|
raw_ostream &OS, function_ref<StringRef(StringRef)> MapClassName2PassName) {
|
|
static_cast<PassInfoMixin<VPlanTestPass> *>(this)->printPipeline(
|
|
OS, MapClassName2PassName);
|
|
OS << '<';
|
|
OS << TransformPipeline;
|
|
OS << '>';
|
|
}
|