d77067d08a
This adds support for using dominating conditions in computeKnownBits() when called from InstCombine. The implementation uses a DomConditionCache, which stores which branches may provide information that is relevant for a given value. DomConditionCache is similar to AssumptionCache, but does not try to do any kind of automatic tracking. Relevant branches have to be explicitly registered and invalidated values explicitly removed. The necessary tracking is done inside InstCombine. The reason why this doesn't just do exactly the same thing as AssumptionCache is that a lot more transforms touch branches and branch conditions than assumptions. AssumptionCache is an immutable analysis and mostly gets away with this because only a handful of places have to register additional assumptions (mostly as a result of cloning). This is very much not the case for branches. This change regresses compile-time by about ~0.2%. It also improves stage2-O0-g builds by about ~0.2%, which indicates that this change results in additional optimizations inside clang itself. Fixes https://github.com/llvm/llvm-project/issues/74242.
57 lines
1.9 KiB
C++
57 lines
1.9 KiB
C++
//===- llvm/Analysis/DomConditionCache.h ------------------------*- C++ -*-===//
|
|
//
|
|
// 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// Cache for branch conditions that affect a certain value for use by
|
|
// ValueTracking. Unlike AssumptionCache, this class does not perform any
|
|
// automatic analysis or invalidation. The caller is responsible for registering
|
|
// all relevant branches (and re-registering them if they change), and for
|
|
// removing invalidated values from the cache.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef LLVM_ANALYSIS_DOMCONDITIONCACHE_H
|
|
#define LLVM_ANALYSIS_DOMCONDITIONCACHE_H
|
|
|
|
#include "llvm/ADT/ArrayRef.h"
|
|
#include "llvm/ADT/DenseMap.h"
|
|
#include "llvm/ADT/DenseMapInfo.h"
|
|
#include "llvm/ADT/SmallVector.h"
|
|
#include "llvm/IR/ValueHandle.h"
|
|
|
|
namespace llvm {
|
|
|
|
class Value;
|
|
class BranchInst;
|
|
|
|
class DomConditionCache {
|
|
private:
|
|
/// A map of values about which a branch might be providing information.
|
|
using AffectedValuesMap = DenseMap<Value *, SmallVector<BranchInst *, 1>>;
|
|
AffectedValuesMap AffectedValues;
|
|
|
|
public:
|
|
/// Add a branch condition to the cache.
|
|
void registerBranch(BranchInst *BI);
|
|
|
|
/// Remove a value from the cache, e.g. because it will be erased.
|
|
void removeValue(Value *V) { AffectedValues.erase(V); }
|
|
|
|
/// Access the list of branches which affect this value.
|
|
ArrayRef<BranchInst *> conditionsFor(const Value *V) const {
|
|
auto AVI = AffectedValues.find_as(const_cast<Value *>(V));
|
|
if (AVI == AffectedValues.end())
|
|
return ArrayRef<BranchInst *>();
|
|
|
|
return AVI->second;
|
|
}
|
|
};
|
|
|
|
} // end namespace llvm
|
|
|
|
#endif // LLVM_ANALYSIS_DOMCONDITIONCACHE_H
|