d47dd11071
This allows for checking if a given operation may modify/reference/or both a given value. Right now this API is limited to Value based memory locations, but we should expand this to include attribute based values at some point. This is left for future work because the rest of the AliasAnalysis API also has this restriction. Differential Revision: https://reviews.llvm.org/D101673
98 lines
2.9 KiB
C++
98 lines
2.9 KiB
C++
//===- AliasAnalysis.cpp - Alias Analysis for MLIR ------------------------===//
|
|
//
|
|
// 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "mlir/Analysis/AliasAnalysis.h"
|
|
#include "mlir/Analysis/AliasAnalysis/LocalAliasAnalysis.h"
|
|
|
|
using namespace mlir;
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// AliasResult
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
/// Merge this alias result with `other` and return a new result that
|
|
/// represents the conservative merge of both results.
|
|
AliasResult AliasResult::merge(AliasResult other) const {
|
|
if (kind == other.kind)
|
|
return *this;
|
|
// A mix of PartialAlias and MustAlias is PartialAlias.
|
|
if ((isPartial() && other.isMust()) || (other.isPartial() && isMust()))
|
|
return PartialAlias;
|
|
// Otherwise, don't assume anything.
|
|
return MayAlias;
|
|
}
|
|
|
|
void AliasResult::print(raw_ostream &os) const {
|
|
switch (kind) {
|
|
case Kind::NoAlias:
|
|
os << "NoAlias";
|
|
break;
|
|
case Kind::MayAlias:
|
|
os << "MayAlias";
|
|
break;
|
|
case Kind::PartialAlias:
|
|
os << "PartialAlias";
|
|
break;
|
|
case Kind::MustAlias:
|
|
os << "MustAlias";
|
|
break;
|
|
}
|
|
}
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// ModRefResult
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
void ModRefResult::print(raw_ostream &os) const {
|
|
switch (kind) {
|
|
case Kind::NoModRef:
|
|
os << "NoModRef";
|
|
break;
|
|
case Kind::Ref:
|
|
os << "Ref";
|
|
break;
|
|
case Kind::Mod:
|
|
os << "Mod";
|
|
break;
|
|
case Kind::ModRef:
|
|
os << "ModRef";
|
|
break;
|
|
}
|
|
}
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// AliasAnalysis
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
AliasAnalysis::AliasAnalysis(Operation *op) {
|
|
addAnalysisImplementation(LocalAliasAnalysis());
|
|
}
|
|
|
|
AliasResult AliasAnalysis::alias(Value lhs, Value rhs) {
|
|
// Check each of the alias analysis implemenations for an alias result.
|
|
for (const std::unique_ptr<Concept> &aliasImpl : aliasImpls) {
|
|
AliasResult result = aliasImpl->alias(lhs, rhs);
|
|
if (!result.isMay())
|
|
return result;
|
|
}
|
|
return AliasResult::MayAlias;
|
|
}
|
|
|
|
ModRefResult AliasAnalysis::getModRef(Operation *op, Value location) {
|
|
// Compute the mod-ref behavior by refining a top `ModRef` result with each of
|
|
// the alias analysis implementations. We early exit at the point where we
|
|
// refine down to a `NoModRef`.
|
|
ModRefResult result = ModRefResult::getModAndRef();
|
|
for (const std::unique_ptr<Concept> &aliasImpl : aliasImpls) {
|
|
result = result.intersect(aliasImpl->getModRef(op, location));
|
|
if (result.isNoModRef())
|
|
return result;
|
|
}
|
|
return result;
|
|
}
|