68 lines
2.4 KiB
C++
68 lines
2.4 KiB
C++
|
//===- ReduceArguments.cpp - Specialized Delta 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 function which calls the Generic Delta pass in order
|
||
|
|
// to reduce uninteresting Arguments from defined functions.
|
||
|
|
//
|
||
|
|
//===----------------------------------------------------------------------===//
|
||
|
|
|
||
|
|
#include "ReduceInstructions.h"
|
||
|
|
|
||
|
|
using namespace llvm;
|
||
|
|
|
||
|
|
/// Removes out-of-chunk arguments from functions, and modifies their calls
|
||
|
|
/// accordingly. It also removes allocations of out-of-chunk arguments.
|
||
|
|
static void extractInstrFromModule(std::vector<Chunk> ChunksToKeep,
|
||
|
|
Module *Program) {
|
||
|
|
Oracle O(ChunksToKeep);
|
||
|
|
|
||
|
|
std::set<Instruction *> InstToKeep;
|
||
|
|
|
||
|
|
for (auto &F : *Program)
|
||
|
|
for (auto &BB : F) {
|
||
|
|
// Removing the terminator would make the block invalid. Only iterate over
|
||
|
|
// instructions before the terminator.
|
||
|
|
InstToKeep.insert(BB.getTerminator());
|
||
|
|
for (auto &Inst : make_range(BB.begin(), std::prev(BB.end())))
|
||
|
|
if (O.shouldKeep())
|
||
|
|
InstToKeep.insert(&Inst);
|
||
|
|
}
|
||
|
|
|
||
|
|
std::vector<Instruction *> InstToDelete;
|
||
|
|
for (auto &F : *Program)
|
||
|
|
for (auto &BB : F)
|
||
|
|
for (auto &Inst : BB)
|
||
|
|
if (!InstToKeep.count(&Inst)) {
|
||
|
|
Inst.replaceAllUsesWith(UndefValue::get(Inst.getType()));
|
||
|
|
InstToDelete.push_back(&Inst);
|
||
|
|
}
|
||
|
|
|
||
|
|
for (auto &I : InstToDelete)
|
||
|
|
I->eraseFromParent();
|
||
|
|
}
|
||
|
|
|
||
|
|
/// Counts the amount of basic blocks and prints their name & respective index
|
||
|
|
static unsigned countInstructions(Module *Program) {
|
||
|
|
// TODO: Silence index with --quiet flag
|
||
|
|
outs() << "----------------------------\n";
|
||
|
|
int InstCount = 0;
|
||
|
|
for (auto &F : *Program)
|
||
|
|
for (auto &BB : F)
|
||
|
|
// Well-formed blocks have terminators, which we cannot remove.
|
||
|
|
InstCount += BB.getInstList().size() - 1;
|
||
|
|
outs() << "Number of instructions: " << InstCount << "\n";
|
||
|
|
|
||
|
|
return InstCount;
|
||
|
|
}
|
||
|
|
|
||
|
|
void llvm::reduceInstructionsDeltaPass(TestRunner &Test) {
|
||
|
|
outs() << "*** Reducing Instructions...\n";
|
||
|
|
unsigned InstCount = countInstructions(Test.getProgram());
|
||
|
|
runDeltaPass(Test, InstCount, extractInstrFromModule);
|
||
|
|
}
|