348 lines
13 KiB
C++
348 lines
13 KiB
C++
// Copyright (c) 2017 Google Inc.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
// This file implements conditional constant propagation as described in
|
|
//
|
|
// Constant propagation with conditional branches,
|
|
// Wegman and Zadeck, ACM TOPLAS 13(2):181-210.
|
|
|
|
#include "source/opt/ccp_pass.h"
|
|
|
|
#include <algorithm>
|
|
#include <limits>
|
|
|
|
#include "source/opt/fold.h"
|
|
#include "source/opt/function.h"
|
|
#include "source/opt/module.h"
|
|
#include "source/opt/propagator.h"
|
|
|
|
namespace spvtools {
|
|
namespace opt {
|
|
|
|
namespace {
|
|
|
|
// This SSA id is never defined nor referenced in the IR. It is a special ID
|
|
// which represents varying values. When an ID is found to have a varying
|
|
// value, its entry in the |values_| table maps to kVaryingSSAId.
|
|
const uint32_t kVaryingSSAId = std::numeric_limits<uint32_t>::max();
|
|
|
|
} // namespace
|
|
|
|
bool CCPPass::IsVaryingValue(uint32_t id) const { return id == kVaryingSSAId; }
|
|
|
|
SSAPropagator::PropStatus CCPPass::MarkInstructionVarying(Instruction* instr) {
|
|
assert(instr->result_id() != 0 &&
|
|
"Instructions with no result cannot be marked varying.");
|
|
values_[instr->result_id()] = kVaryingSSAId;
|
|
return SSAPropagator::kVarying;
|
|
}
|
|
|
|
SSAPropagator::PropStatus CCPPass::VisitPhi(Instruction* phi) {
|
|
uint32_t meet_val_id = 0;
|
|
|
|
// Implement the lattice meet operation. The result of this Phi instruction is
|
|
// interesting only if the meet operation over arguments coming through
|
|
// executable edges yields the same constant value.
|
|
for (uint32_t i = 2; i < phi->NumOperands(); i += 2) {
|
|
if (!propagator_->IsPhiArgExecutable(phi, i)) {
|
|
// Ignore arguments coming through non-executable edges.
|
|
continue;
|
|
}
|
|
uint32_t phi_arg_id = phi->GetSingleWordOperand(i);
|
|
auto it = values_.find(phi_arg_id);
|
|
if (it != values_.end()) {
|
|
// We found an argument with a constant value. Apply the meet operation
|
|
// with the previous arguments.
|
|
if (it->second == kVaryingSSAId) {
|
|
// The "constant" value is actually a placeholder for varying. Return
|
|
// varying for this phi.
|
|
return MarkInstructionVarying(phi);
|
|
} else if (meet_val_id == 0) {
|
|
// This is the first argument we find. Initialize the result to its
|
|
// constant value id.
|
|
meet_val_id = it->second;
|
|
} else if (it->second == meet_val_id) {
|
|
// The argument is the same constant value already computed. Continue
|
|
// looking.
|
|
continue;
|
|
} else {
|
|
// We either found a varying value, or another constant value different
|
|
// from the previous computed meet value. This Phi will never be
|
|
// constant.
|
|
return MarkInstructionVarying(phi);
|
|
}
|
|
} else {
|
|
// The incoming value has no recorded value and is therefore not
|
|
// interesting. A not interesting value joined with any other value is the
|
|
// other value.
|
|
continue;
|
|
}
|
|
}
|
|
|
|
// If there are no incoming executable edges, the meet ID will still be 0. In
|
|
// that case, return not interesting to evaluate the Phi node again.
|
|
if (meet_val_id == 0) {
|
|
return SSAPropagator::kNotInteresting;
|
|
}
|
|
|
|
// All the operands have the same constant value represented by |meet_val_id|.
|
|
// Set the Phi's result to that value and declare it interesting.
|
|
values_[phi->result_id()] = meet_val_id;
|
|
return SSAPropagator::kInteresting;
|
|
}
|
|
|
|
SSAPropagator::PropStatus CCPPass::VisitAssignment(Instruction* instr) {
|
|
assert(instr->result_id() != 0 &&
|
|
"Expecting an instruction that produces a result");
|
|
|
|
// If this is a copy operation, and the RHS is a known constant, assign its
|
|
// value to the LHS.
|
|
if (instr->opcode() == SpvOpCopyObject) {
|
|
uint32_t rhs_id = instr->GetSingleWordInOperand(0);
|
|
auto it = values_.find(rhs_id);
|
|
if (it != values_.end()) {
|
|
if (IsVaryingValue(it->second)) {
|
|
return MarkInstructionVarying(instr);
|
|
} else {
|
|
values_[instr->result_id()] = it->second;
|
|
return SSAPropagator::kInteresting;
|
|
}
|
|
}
|
|
return SSAPropagator::kNotInteresting;
|
|
}
|
|
|
|
// Instructions with a RHS that cannot produce a constant are always varying.
|
|
if (!instr->IsFoldable()) {
|
|
return MarkInstructionVarying(instr);
|
|
}
|
|
|
|
// See if the RHS of the assignment folds into a constant value.
|
|
auto map_func = [this](uint32_t id) {
|
|
auto it = values_.find(id);
|
|
if (it == values_.end() || IsVaryingValue(it->second)) {
|
|
return id;
|
|
}
|
|
return it->second;
|
|
};
|
|
Instruction* folded_inst =
|
|
context()->get_instruction_folder().FoldInstructionToConstant(instr,
|
|
map_func);
|
|
|
|
if (folded_inst != nullptr) {
|
|
// We do not want to change the body of the function by adding new
|
|
// instructions. When folding we can only generate new constants.
|
|
assert(folded_inst->IsConstant() && "CCP is only interested in constant.");
|
|
values_[instr->result_id()] = folded_inst->result_id();
|
|
return SSAPropagator::kInteresting;
|
|
}
|
|
|
|
// Conservatively mark this instruction as varying if any input id is varying.
|
|
if (!instr->WhileEachInId([this](uint32_t* op_id) {
|
|
auto iter = values_.find(*op_id);
|
|
if (iter != values_.end() && IsVaryingValue(iter->second)) return false;
|
|
return true;
|
|
})) {
|
|
return MarkInstructionVarying(instr);
|
|
}
|
|
|
|
// If not, see if there is a least one unknown operand to the instruction. If
|
|
// so, we might be able to fold it later.
|
|
if (!instr->WhileEachInId([this](uint32_t* op_id) {
|
|
auto it = values_.find(*op_id);
|
|
if (it == values_.end()) return false;
|
|
return true;
|
|
})) {
|
|
return SSAPropagator::kNotInteresting;
|
|
}
|
|
|
|
// Otherwise, we will never be able to fold this instruction, so mark it
|
|
// varying.
|
|
return MarkInstructionVarying(instr);
|
|
}
|
|
|
|
SSAPropagator::PropStatus CCPPass::VisitBranch(Instruction* instr,
|
|
BasicBlock** dest_bb) const {
|
|
assert(instr->IsBranch() && "Expected a branch instruction.");
|
|
|
|
*dest_bb = nullptr;
|
|
uint32_t dest_label = 0;
|
|
if (instr->opcode() == SpvOpBranch) {
|
|
// An unconditional jump always goes to its unique destination.
|
|
dest_label = instr->GetSingleWordInOperand(0);
|
|
} else if (instr->opcode() == SpvOpBranchConditional) {
|
|
// For a conditional branch, determine whether the predicate selector has a
|
|
// known value in |values_|. If it does, set the destination block
|
|
// according to the selector's boolean value.
|
|
uint32_t pred_id = instr->GetSingleWordOperand(0);
|
|
auto it = values_.find(pred_id);
|
|
if (it == values_.end() || IsVaryingValue(it->second)) {
|
|
// The predicate has an unknown value, either branch could be taken.
|
|
return SSAPropagator::kVarying;
|
|
}
|
|
|
|
// Get the constant value for the predicate selector from the value table.
|
|
// Use it to decide which branch will be taken.
|
|
uint32_t pred_val_id = it->second;
|
|
const analysis::Constant* c = const_mgr_->FindDeclaredConstant(pred_val_id);
|
|
assert(c && "Expected to find a constant declaration for a known value.");
|
|
// Undef values should have returned as varying above.
|
|
assert(c->AsBoolConstant() || c->AsNullConstant());
|
|
if (c->AsNullConstant()) {
|
|
dest_label = instr->GetSingleWordOperand(2u);
|
|
} else {
|
|
const analysis::BoolConstant* val = c->AsBoolConstant();
|
|
dest_label = val->value() ? instr->GetSingleWordOperand(1)
|
|
: instr->GetSingleWordOperand(2);
|
|
}
|
|
} else {
|
|
// For an OpSwitch, extract the value taken by the switch selector and check
|
|
// which of the target literals it matches. The branch associated with that
|
|
// literal is the taken branch.
|
|
assert(instr->opcode() == SpvOpSwitch);
|
|
if (instr->GetOperand(0).words.size() != 1) {
|
|
// If the selector is wider than 32-bits, return varying. TODO(dnovillo):
|
|
// Add support for wider constants.
|
|
return SSAPropagator::kVarying;
|
|
}
|
|
uint32_t select_id = instr->GetSingleWordOperand(0);
|
|
auto it = values_.find(select_id);
|
|
if (it == values_.end() || IsVaryingValue(it->second)) {
|
|
// The selector has an unknown value, any of the branches could be taken.
|
|
return SSAPropagator::kVarying;
|
|
}
|
|
|
|
// Get the constant value for the selector from the value table. Use it to
|
|
// decide which branch will be taken.
|
|
uint32_t select_val_id = it->second;
|
|
const analysis::Constant* c =
|
|
const_mgr_->FindDeclaredConstant(select_val_id);
|
|
assert(c && "Expected to find a constant declaration for a known value.");
|
|
// TODO: support 64-bit integer switches.
|
|
uint32_t constant_cond = 0;
|
|
if (const analysis::IntConstant* val = c->AsIntConstant()) {
|
|
constant_cond = val->words()[0];
|
|
} else {
|
|
// Undef values should have returned varying above.
|
|
assert(c->AsNullConstant());
|
|
constant_cond = 0;
|
|
}
|
|
|
|
// Start assuming that the selector will take the default value;
|
|
dest_label = instr->GetSingleWordOperand(1);
|
|
for (uint32_t i = 2; i < instr->NumOperands(); i += 2) {
|
|
if (constant_cond == instr->GetSingleWordOperand(i)) {
|
|
dest_label = instr->GetSingleWordOperand(i + 1);
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
assert(dest_label && "Destination label should be set at this point.");
|
|
*dest_bb = context()->cfg()->block(dest_label);
|
|
return SSAPropagator::kInteresting;
|
|
}
|
|
|
|
SSAPropagator::PropStatus CCPPass::VisitInstruction(Instruction* instr,
|
|
BasicBlock** dest_bb) {
|
|
*dest_bb = nullptr;
|
|
if (instr->opcode() == SpvOpPhi) {
|
|
return VisitPhi(instr);
|
|
} else if (instr->IsBranch()) {
|
|
return VisitBranch(instr, dest_bb);
|
|
} else if (instr->result_id()) {
|
|
return VisitAssignment(instr);
|
|
}
|
|
return SSAPropagator::kVarying;
|
|
}
|
|
|
|
bool CCPPass::ReplaceValues() {
|
|
// Even if we make no changes to the function's IR, propagation may have
|
|
// created new constants. Even if those constants cannot be replaced in
|
|
// the IR, the constant definition itself is a change. To reflect this,
|
|
// we check whether the next ID to be given by the module is different than
|
|
// the original bound ID. If that happens, new instructions were added to the
|
|
// module during propagation.
|
|
//
|
|
// See https://github.com/KhronosGroup/SPIRV-Tools/issues/3636 and
|
|
// https://github.com/KhronosGroup/SPIRV-Tools/issues/3991 for details.
|
|
bool changed_ir = (context()->module()->IdBound() > original_id_bound_);
|
|
|
|
for (const auto& it : values_) {
|
|
uint32_t id = it.first;
|
|
uint32_t cst_id = it.second;
|
|
if (!IsVaryingValue(cst_id) && id != cst_id) {
|
|
context()->KillNamesAndDecorates(id);
|
|
changed_ir |= context()->ReplaceAllUsesWith(id, cst_id);
|
|
}
|
|
}
|
|
|
|
return changed_ir;
|
|
}
|
|
|
|
bool CCPPass::PropagateConstants(Function* fp) {
|
|
if (fp->IsDeclaration()) {
|
|
return false;
|
|
}
|
|
|
|
// Mark function parameters as varying.
|
|
fp->ForEachParam([this](const Instruction* inst) {
|
|
values_[inst->result_id()] = kVaryingSSAId;
|
|
});
|
|
|
|
const auto visit_fn = [this](Instruction* instr, BasicBlock** dest_bb) {
|
|
return VisitInstruction(instr, dest_bb);
|
|
};
|
|
|
|
propagator_ =
|
|
std::unique_ptr<SSAPropagator>(new SSAPropagator(context(), visit_fn));
|
|
|
|
if (propagator_->Run(fp)) {
|
|
return ReplaceValues();
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
void CCPPass::Initialize() {
|
|
const_mgr_ = context()->get_constant_mgr();
|
|
|
|
// Populate the constant table with values from constant declarations in the
|
|
// module. The values of each OpConstant declaration is the identity
|
|
// assignment (i.e., each constant is its own value).
|
|
for (const auto& inst : get_module()->types_values()) {
|
|
// Record compile time constant ids. Treat all other global values as
|
|
// varying.
|
|
if (inst.IsConstant()) {
|
|
values_[inst.result_id()] = inst.result_id();
|
|
} else {
|
|
values_[inst.result_id()] = kVaryingSSAId;
|
|
}
|
|
}
|
|
|
|
original_id_bound_ = context()->module()->IdBound();
|
|
}
|
|
|
|
Pass::Status CCPPass::Process() {
|
|
Initialize();
|
|
|
|
// Process all entry point functions.
|
|
ProcessFunction pfn = [this](Function* fp) { return PropagateConstants(fp); };
|
|
bool modified = context()->ProcessReachableCallTree(pfn);
|
|
return modified ? Pass::Status::SuccessWithChange
|
|
: Pass::Status::SuccessWithoutChange;
|
|
}
|
|
|
|
} // namespace opt
|
|
} // namespace spvtools
|