338 lines
14 KiB
C++
338 lines
14 KiB
C++
|
//===- llvm/unittest/CodeGen/SelectionDAGAddressAnalysisTest.cpp ---------===//
|
||
|
|
//
|
||
|
|
// 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 "llvm/CodeGen/SelectionDAGAddressAnalysis.h"
|
||
|
|
#include "llvm/Analysis/MemoryLocation.h"
|
||
|
|
#include "llvm/Analysis/OptimizationRemarkEmitter.h"
|
||
|
|
#include "llvm/AsmParser/Parser.h"
|
||
|
|
#include "llvm/CodeGen/MachineModuleInfo.h"
|
||
|
|
#include "llvm/CodeGen/SelectionDAG.h"
|
||
|
|
#include "llvm/CodeGen/TargetLowering.h"
|
||
|
|
#include "llvm/Support/SourceMgr.h"
|
||
|
|
#include "llvm/Support/TargetRegistry.h"
|
||
|
|
#include "llvm/Support/TargetSelect.h"
|
||
|
|
#include "llvm/Target/TargetMachine.h"
|
||
|
|
#include "gtest/gtest.h"
|
||
|
|
|
||
|
|
namespace llvm {
|
||
|
|
|
||
|
|
class SelectionDAGAddressAnalysisTest : public testing::Test {
|
||
|
|
protected:
|
||
|
|
static void SetUpTestCase() {
|
||
|
|
InitializeAllTargets();
|
||
|
|
InitializeAllTargetMCs();
|
||
|
|
}
|
||
|
|
|
||
|
|
void SetUp() override {
|
||
|
|
StringRef Assembly = "@g = global i32 0\n"
|
||
|
|
"define i32 @f() {\n"
|
||
|
|
" %1 = load i32, i32* @g\n"
|
||
|
|
" ret i32 %1\n"
|
||
|
|
"}";
|
||
|
|
|
||
|
|
Triple TargetTriple("aarch64--");
|
||
|
|
std::string Error;
|
||
|
|
const Target *T = TargetRegistry::lookupTarget("", TargetTriple, Error);
|
||
|
|
// FIXME: These tests do not depend on AArch64 specifically, but we have to
|
||
|
|
// initialize a target. A skeleton Target for unittests would allow us to
|
||
|
|
// always run these tests.
|
||
|
|
if (!T)
|
||
|
|
return;
|
||
|
|
|
||
|
|
TargetOptions Options;
|
||
|
|
TM = std::unique_ptr<LLVMTargetMachine>(static_cast<LLVMTargetMachine *>(
|
||
|
|
T->createTargetMachine("AArch64", "", "+sve", Options, None, None,
|
||
|
|
CodeGenOpt::Aggressive)));
|
||
|
|
if (!TM)
|
||
|
|
return;
|
||
|
|
|
||
|
|
SMDiagnostic SMError;
|
||
|
|
M = parseAssemblyString(Assembly, SMError, Context);
|
||
|
|
if (!M)
|
||
|
|
report_fatal_error(SMError.getMessage());
|
||
|
|
M->setDataLayout(TM->createDataLayout());
|
||
|
|
|
||
|
|
F = M->getFunction("f");
|
||
|
|
if (!F)
|
||
|
|
report_fatal_error("F?");
|
||
|
|
G = M->getGlobalVariable("g");
|
||
|
|
if (!G)
|
||
|
|
report_fatal_error("G?");
|
||
|
|
|
||
|
|
MachineModuleInfo MMI(TM.get());
|
||
|
|
|
||
|
|
MF = std::make_unique<MachineFunction>(*F, *TM, *TM->getSubtargetImpl(*F),
|
||
|
|
0, MMI);
|
||
|
|
|
||
|
|
DAG = std::make_unique<SelectionDAG>(*TM, CodeGenOpt::None);
|
||
|
|
if (!DAG)
|
||
|
|
report_fatal_error("DAG?");
|
||
|
|
OptimizationRemarkEmitter ORE(F);
|
||
|
|
DAG->init(*MF, ORE, nullptr, nullptr, nullptr, nullptr, nullptr);
|
||
|
|
}
|
||
|
|
|
||
|
|
TargetLoweringBase::LegalizeTypeAction getTypeAction(EVT VT) {
|
||
|
|
return DAG->getTargetLoweringInfo().getTypeAction(Context, VT);
|
||
|
|
}
|
||
|
|
|
||
|
|
EVT getTypeToTransformTo(EVT VT) {
|
||
|
|
return DAG->getTargetLoweringInfo().getTypeToTransformTo(Context, VT);
|
||
|
|
}
|
||
|
|
|
||
|
|
LLVMContext Context;
|
||
|
|
std::unique_ptr<LLVMTargetMachine> TM;
|
||
|
|
std::unique_ptr<Module> M;
|
||
|
|
Function *F;
|
||
|
|
GlobalVariable *G;
|
||
|
|
std::unique_ptr<MachineFunction> MF;
|
||
|
|
std::unique_ptr<SelectionDAG> DAG;
|
||
|
|
};
|
||
|
|
|
||
|
|
TEST_F(SelectionDAGAddressAnalysisTest, sameFrameObject) {
|
||
|
|
if (!TM)
|
||
|
|
return;
|
||
|
|
SDLoc Loc;
|
||
|
|
auto Int8VT = EVT::getIntegerVT(Context, 8);
|
||
|
|
auto VecVT = EVT::getVectorVT(Context, Int8VT, 4);
|
||
|
|
SDValue FIPtr = DAG->CreateStackTemporary(VecVT);
|
||
|
|
int FI = cast<FrameIndexSDNode>(FIPtr.getNode())->getIndex();
|
||
|
|
MachinePointerInfo PtrInfo = MachinePointerInfo::getFixedStack(*MF, FI);
|
||
|
|
TypeSize Offset = TypeSize::Fixed(0);
|
||
|
|
SDValue Value = DAG->getConstant(0, Loc, VecVT);
|
||
|
|
SDValue Index = DAG->getMemBasePlusOffset(FIPtr, Offset, Loc);
|
||
|
|
SDValue Store = DAG->getStore(DAG->getEntryNode(), Loc, Value, Index,
|
||
|
|
PtrInfo.getWithOffset(Offset));
|
||
|
|
Optional<int64_t> NumBytes = MemoryLocation::getSizeOrUnknown(
|
||
|
|
cast<StoreSDNode>(Store)->getMemoryVT().getStoreSize());
|
||
|
|
|
||
|
|
bool IsAlias;
|
||
|
|
bool IsValid = BaseIndexOffset::computeAliasing(
|
||
|
|
Store.getNode(), NumBytes, Store.getNode(), NumBytes, *DAG, IsAlias);
|
||
|
|
|
||
|
|
EXPECT_TRUE(IsValid);
|
||
|
|
EXPECT_TRUE(IsAlias);
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST_F(SelectionDAGAddressAnalysisTest, noAliasingFrameObjects) {
|
||
|
|
if (!TM)
|
||
|
|
return;
|
||
|
|
SDLoc Loc;
|
||
|
|
auto Int8VT = EVT::getIntegerVT(Context, 8);
|
||
|
|
// <4 x i8>
|
||
|
|
auto VecVT = EVT::getVectorVT(Context, Int8VT, 4);
|
||
|
|
// <2 x i8>
|
||
|
|
auto SubVecVT = EVT::getVectorVT(Context, Int8VT, 2);
|
||
|
|
SDValue FIPtr = DAG->CreateStackTemporary(VecVT);
|
||
|
|
int FI = cast<FrameIndexSDNode>(FIPtr.getNode())->getIndex();
|
||
|
|
MachinePointerInfo PtrInfo = MachinePointerInfo::getFixedStack(*MF, FI);
|
||
|
|
SDValue Value = DAG->getConstant(0, Loc, SubVecVT);
|
||
|
|
TypeSize Offset0 = TypeSize::Fixed(0);
|
||
|
|
TypeSize Offset1 = SubVecVT.getStoreSize();
|
||
|
|
SDValue Index0 = DAG->getMemBasePlusOffset(FIPtr, Offset0, Loc);
|
||
|
|
SDValue Index1 = DAG->getMemBasePlusOffset(FIPtr, Offset1, Loc);
|
||
|
|
SDValue Store0 = DAG->getStore(DAG->getEntryNode(), Loc, Value, Index0,
|
||
|
|
PtrInfo.getWithOffset(Offset0));
|
||
|
|
SDValue Store1 = DAG->getStore(DAG->getEntryNode(), Loc, Value, Index1,
|
||
|
|
PtrInfo.getWithOffset(Offset1));
|
||
|
|
Optional<int64_t> NumBytes0 = MemoryLocation::getSizeOrUnknown(
|
||
|
|
cast<StoreSDNode>(Store0)->getMemoryVT().getStoreSize());
|
||
|
|
Optional<int64_t> NumBytes1 = MemoryLocation::getSizeOrUnknown(
|
||
|
|
cast<StoreSDNode>(Store1)->getMemoryVT().getStoreSize());
|
||
|
|
|
||
|
|
bool IsAlias;
|
||
|
|
bool IsValid = BaseIndexOffset::computeAliasing(
|
||
|
|
Store0.getNode(), NumBytes0, Store1.getNode(), NumBytes1, *DAG, IsAlias);
|
||
|
|
|
||
|
|
EXPECT_TRUE(IsValid);
|
||
|
|
EXPECT_FALSE(IsAlias);
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST_F(SelectionDAGAddressAnalysisTest, unknownSizeFrameObjects) {
|
||
|
|
if (!TM)
|
||
|
|
return;
|
||
|
|
SDLoc Loc;
|
||
|
|
auto Int8VT = EVT::getIntegerVT(Context, 8);
|
||
|
|
// <vscale x 4 x i8>
|
||
|
|
auto VecVT = EVT::getVectorVT(Context, Int8VT, 4, true);
|
||
|
|
// <vscale x 2 x i8>
|
||
|
|
auto SubVecVT = EVT::getVectorVT(Context, Int8VT, 2, true);
|
||
|
|
SDValue FIPtr = DAG->CreateStackTemporary(VecVT);
|
||
|
|
int FI = cast<FrameIndexSDNode>(FIPtr.getNode())->getIndex();
|
||
|
|
MachinePointerInfo PtrInfo = MachinePointerInfo::getFixedStack(*MF, FI);
|
||
|
|
SDValue Value = DAG->getConstant(0, Loc, SubVecVT);
|
||
|
|
TypeSize Offset0 = TypeSize::Fixed(0);
|
||
|
|
TypeSize Offset1 = SubVecVT.getStoreSize();
|
||
|
|
SDValue Index0 = DAG->getMemBasePlusOffset(FIPtr, Offset0, Loc);
|
||
|
|
SDValue Index1 = DAG->getMemBasePlusOffset(FIPtr, Offset1, Loc);
|
||
|
|
SDValue Store0 = DAG->getStore(DAG->getEntryNode(), Loc, Value, Index0,
|
||
|
|
PtrInfo.getWithOffset(Offset0));
|
||
|
|
SDValue Store1 = DAG->getStore(DAG->getEntryNode(), Loc, Value, Index1,
|
||
|
|
PtrInfo.getWithOffset(Offset1));
|
||
|
|
Optional<int64_t> NumBytes0 = MemoryLocation::getSizeOrUnknown(
|
||
|
|
cast<StoreSDNode>(Store0)->getMemoryVT().getStoreSize());
|
||
|
|
Optional<int64_t> NumBytes1 = MemoryLocation::getSizeOrUnknown(
|
||
|
|
cast<StoreSDNode>(Store1)->getMemoryVT().getStoreSize());
|
||
|
|
|
||
|
|
bool IsAlias;
|
||
|
|
bool IsValid = BaseIndexOffset::computeAliasing(
|
||
|
|
Store0.getNode(), NumBytes0, Store1.getNode(), NumBytes1, *DAG, IsAlias);
|
||
|
|
|
||
|
|
EXPECT_FALSE(IsValid);
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST_F(SelectionDAGAddressAnalysisTest, globalWithFrameObject) {
|
||
|
|
if (!TM)
|
||
|
|
return;
|
||
|
|
SDLoc Loc;
|
||
|
|
auto Int8VT = EVT::getIntegerVT(Context, 8);
|
||
|
|
// <vscale x 4 x i8>
|
||
|
|
auto VecVT = EVT::getVectorVT(Context, Int8VT, 4, true);
|
||
|
|
SDValue FIPtr = DAG->CreateStackTemporary(VecVT);
|
||
|
|
int FI = cast<FrameIndexSDNode>(FIPtr.getNode())->getIndex();
|
||
|
|
MachinePointerInfo PtrInfo = MachinePointerInfo::getFixedStack(*MF, FI);
|
||
|
|
SDValue Value = DAG->getConstant(0, Loc, VecVT);
|
||
|
|
TypeSize Offset = TypeSize::Fixed(0);
|
||
|
|
SDValue Index = DAG->getMemBasePlusOffset(FIPtr, Offset, Loc);
|
||
|
|
SDValue Store = DAG->getStore(DAG->getEntryNode(), Loc, Value, Index,
|
||
|
|
PtrInfo.getWithOffset(Offset));
|
||
|
|
Optional<int64_t> NumBytes = MemoryLocation::getSizeOrUnknown(
|
||
|
|
cast<StoreSDNode>(Store)->getMemoryVT().getStoreSize());
|
||
|
|
EVT GTy = DAG->getTargetLoweringInfo().getValueType(DAG->getDataLayout(),
|
||
|
|
G->getType());
|
||
|
|
SDValue GValue = DAG->getConstant(0, Loc, GTy);
|
||
|
|
SDValue GAddr = DAG->getGlobalAddress(G, Loc, GTy);
|
||
|
|
SDValue GStore = DAG->getStore(DAG->getEntryNode(), Loc, GValue, GAddr,
|
||
|
|
MachinePointerInfo(G, 0));
|
||
|
|
Optional<int64_t> GNumBytes = MemoryLocation::getSizeOrUnknown(
|
||
|
|
cast<StoreSDNode>(GStore)->getMemoryVT().getStoreSize());
|
||
|
|
|
||
|
|
bool IsAlias;
|
||
|
|
bool IsValid = BaseIndexOffset::computeAliasing(
|
||
|
|
Store.getNode(), NumBytes, GStore.getNode(), GNumBytes, *DAG, IsAlias);
|
||
|
|
|
||
|
|
EXPECT_TRUE(IsValid);
|
||
|
|
EXPECT_FALSE(IsAlias);
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST_F(SelectionDAGAddressAnalysisTest, fixedSizeFrameObjectsWithinDiff) {
|
||
|
|
if (!TM)
|
||
|
|
return;
|
||
|
|
SDLoc Loc;
|
||
|
|
auto Int8VT = EVT::getIntegerVT(Context, 8);
|
||
|
|
// <vscale x 4 x i8>
|
||
|
|
auto VecVT = EVT::getVectorVT(Context, Int8VT, 4, true);
|
||
|
|
// <vscale x 2 x i8>
|
||
|
|
auto SubVecVT = EVT::getVectorVT(Context, Int8VT, 2, true);
|
||
|
|
// <2 x i8>
|
||
|
|
auto SubFixedVecVT2xi8 = EVT::getVectorVT(Context, Int8VT, 2);
|
||
|
|
SDValue FIPtr = DAG->CreateStackTemporary(VecVT);
|
||
|
|
int FI = cast<FrameIndexSDNode>(FIPtr.getNode())->getIndex();
|
||
|
|
MachinePointerInfo PtrInfo = MachinePointerInfo::getFixedStack(*MF, FI);
|
||
|
|
SDValue Value0 = DAG->getConstant(0, Loc, SubFixedVecVT2xi8);
|
||
|
|
SDValue Value1 = DAG->getConstant(0, Loc, SubVecVT);
|
||
|
|
TypeSize Offset0 = TypeSize::Fixed(0);
|
||
|
|
TypeSize Offset1 = SubFixedVecVT2xi8.getStoreSize();
|
||
|
|
SDValue Index0 = DAG->getMemBasePlusOffset(FIPtr, Offset0, Loc);
|
||
|
|
SDValue Index1 = DAG->getMemBasePlusOffset(FIPtr, Offset1, Loc);
|
||
|
|
SDValue Store0 = DAG->getStore(DAG->getEntryNode(), Loc, Value0, Index0,
|
||
|
|
PtrInfo.getWithOffset(Offset0));
|
||
|
|
SDValue Store1 = DAG->getStore(DAG->getEntryNode(), Loc, Value1, Index1,
|
||
|
|
PtrInfo.getWithOffset(Offset1));
|
||
|
|
Optional<int64_t> NumBytes0 = MemoryLocation::getSizeOrUnknown(
|
||
|
|
cast<StoreSDNode>(Store0)->getMemoryVT().getStoreSize());
|
||
|
|
Optional<int64_t> NumBytes1 = MemoryLocation::getSizeOrUnknown(
|
||
|
|
cast<StoreSDNode>(Store1)->getMemoryVT().getStoreSize());
|
||
|
|
|
||
|
|
bool IsAlias;
|
||
|
|
bool IsValid = BaseIndexOffset::computeAliasing(
|
||
|
|
Store0.getNode(), NumBytes0, Store1.getNode(), NumBytes1, *DAG, IsAlias);
|
||
|
|
EXPECT_TRUE(IsValid);
|
||
|
|
EXPECT_FALSE(IsAlias);
|
||
|
|
|
||
|
|
IsValid = BaseIndexOffset::computeAliasing(
|
||
|
|
Store1.getNode(), NumBytes1, Store0.getNode(), NumBytes0, *DAG, IsAlias);
|
||
|
|
EXPECT_TRUE(IsValid);
|
||
|
|
EXPECT_FALSE(IsAlias);
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST_F(SelectionDAGAddressAnalysisTest, fixedSizeFrameObjectsOutOfDiff) {
|
||
|
|
if (!TM)
|
||
|
|
return;
|
||
|
|
SDLoc Loc;
|
||
|
|
auto Int8VT = EVT::getIntegerVT(Context, 8);
|
||
|
|
// <vscale x 4 x i8>
|
||
|
|
auto VecVT = EVT::getVectorVT(Context, Int8VT, 4, true);
|
||
|
|
// <vscale x 2 x i8>
|
||
|
|
auto SubVecVT = EVT::getVectorVT(Context, Int8VT, 2, true);
|
||
|
|
// <2 x i8>
|
||
|
|
auto SubFixedVecVT2xi8 = EVT::getVectorVT(Context, Int8VT, 2);
|
||
|
|
// <4 x i8>
|
||
|
|
auto SubFixedVecVT4xi8 = EVT::getVectorVT(Context, Int8VT, 4);
|
||
|
|
SDValue FIPtr = DAG->CreateStackTemporary(VecVT);
|
||
|
|
int FI = cast<FrameIndexSDNode>(FIPtr.getNode())->getIndex();
|
||
|
|
MachinePointerInfo PtrInfo = MachinePointerInfo::getFixedStack(*MF, FI);
|
||
|
|
SDValue Value0 = DAG->getConstant(0, Loc, SubFixedVecVT4xi8);
|
||
|
|
SDValue Value1 = DAG->getConstant(0, Loc, SubVecVT);
|
||
|
|
TypeSize Offset0 = TypeSize::Fixed(0);
|
||
|
|
TypeSize Offset1 = SubFixedVecVT2xi8.getStoreSize();
|
||
|
|
SDValue Index0 = DAG->getMemBasePlusOffset(FIPtr, Offset0, Loc);
|
||
|
|
SDValue Index1 = DAG->getMemBasePlusOffset(FIPtr, Offset1, Loc);
|
||
|
|
SDValue Store0 = DAG->getStore(DAG->getEntryNode(), Loc, Value0, Index0,
|
||
|
|
PtrInfo.getWithOffset(Offset0));
|
||
|
|
SDValue Store1 = DAG->getStore(DAG->getEntryNode(), Loc, Value1, Index1,
|
||
|
|
PtrInfo.getWithOffset(Offset1));
|
||
|
|
Optional<int64_t> NumBytes0 = MemoryLocation::getSizeOrUnknown(
|
||
|
|
cast<StoreSDNode>(Store0)->getMemoryVT().getStoreSize());
|
||
|
|
Optional<int64_t> NumBytes1 = MemoryLocation::getSizeOrUnknown(
|
||
|
|
cast<StoreSDNode>(Store1)->getMemoryVT().getStoreSize());
|
||
|
|
|
||
|
|
bool IsAlias;
|
||
|
|
bool IsValid = BaseIndexOffset::computeAliasing(
|
||
|
|
Store0.getNode(), NumBytes0, Store1.getNode(), NumBytes1, *DAG, IsAlias);
|
||
|
|
EXPECT_TRUE(IsValid);
|
||
|
|
EXPECT_TRUE(IsAlias);
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST_F(SelectionDAGAddressAnalysisTest, twoFixedStackObjects) {
|
||
|
|
if (!TM)
|
||
|
|
return;
|
||
|
|
SDLoc Loc;
|
||
|
|
auto Int8VT = EVT::getIntegerVT(Context, 8);
|
||
|
|
// <vscale x 2 x i8>
|
||
|
|
auto VecVT = EVT::getVectorVT(Context, Int8VT, 2, true);
|
||
|
|
// <2 x i8>
|
||
|
|
auto FixedVecVT = EVT::getVectorVT(Context, Int8VT, 2);
|
||
|
|
SDValue FIPtr0 = DAG->CreateStackTemporary(FixedVecVT);
|
||
|
|
SDValue FIPtr1 = DAG->CreateStackTemporary(VecVT);
|
||
|
|
int FI0 = cast<FrameIndexSDNode>(FIPtr0.getNode())->getIndex();
|
||
|
|
int FI1 = cast<FrameIndexSDNode>(FIPtr1.getNode())->getIndex();
|
||
|
|
MachinePointerInfo PtrInfo0 = MachinePointerInfo::getFixedStack(*MF, FI0);
|
||
|
|
MachinePointerInfo PtrInfo1 = MachinePointerInfo::getFixedStack(*MF, FI1);
|
||
|
|
SDValue Value0 = DAG->getConstant(0, Loc, FixedVecVT);
|
||
|
|
SDValue Value1 = DAG->getConstant(0, Loc, VecVT);
|
||
|
|
TypeSize Offset0 = TypeSize::Fixed(0);
|
||
|
|
SDValue Index0 = DAG->getMemBasePlusOffset(FIPtr0, Offset0, Loc);
|
||
|
|
SDValue Index1 = DAG->getMemBasePlusOffset(FIPtr1, Offset0, Loc);
|
||
|
|
SDValue Store0 = DAG->getStore(DAG->getEntryNode(), Loc, Value0, Index0,
|
||
|
|
PtrInfo0.getWithOffset(Offset0));
|
||
|
|
SDValue Store1 = DAG->getStore(DAG->getEntryNode(), Loc, Value1, Index1,
|
||
|
|
PtrInfo1.getWithOffset(Offset0));
|
||
|
|
Optional<int64_t> NumBytes0 = MemoryLocation::getSizeOrUnknown(
|
||
|
|
cast<StoreSDNode>(Store0)->getMemoryVT().getStoreSize());
|
||
|
|
Optional<int64_t> NumBytes1 = MemoryLocation::getSizeOrUnknown(
|
||
|
|
cast<StoreSDNode>(Store1)->getMemoryVT().getStoreSize());
|
||
|
|
|
||
|
|
bool IsAlias;
|
||
|
|
bool IsValid = BaseIndexOffset::computeAliasing(
|
||
|
|
Store0.getNode(), NumBytes0, Store1.getNode(), NumBytes1, *DAG, IsAlias);
|
||
|
|
EXPECT_TRUE(IsValid);
|
||
|
|
EXPECT_FALSE(IsAlias);
|
||
|
|
}
|
||
|
|
|
||
|
|
} // end namespace llvm
|