[RISCV][VLOPT] Add vector narrowing integer right shift instructions to isSupportedIn...
[llvm-project.git] / llvm / unittests / Analysis / AliasSetTrackerTest.cpp
blob68bd41a1e8589b7222f86feedf1f704700b45fcf
1 //=======- AliasSetTrackerTest.cpp - Unit test for the Alias Set Tracker -===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
9 #include "llvm/Analysis/AliasSetTracker.h"
10 #include "llvm/Analysis/AliasAnalysis.h"
11 #include "llvm/Analysis/TargetLibraryInfo.h"
12 #include "llvm/Analysis/TypeBasedAliasAnalysis.h"
13 #include "llvm/AsmParser/Parser.h"
14 #include "llvm/IR/LLVMContext.h"
15 #include "llvm/IR/Module.h"
16 #include "llvm/Support/SourceMgr.h"
17 #include "llvm/TargetParser/Triple.h"
18 #include "gtest/gtest.h"
20 using namespace llvm;
22 TEST(AliasSetTracker, AliasUnknownInst) {
23 StringRef Assembly = R"(
24 @a = common global i32 0, align 4
25 @b = common global float 0.000000e+00, align 4
27 ; Function Attrs: nounwind ssp uwtable
28 define i32 @read_a() #0 {
29 %1 = load i32, i32* @a, align 4, !tbaa !3
30 ret i32 %1
33 ; Function Attrs: nounwind ssp uwtable
34 define void @write_b() #0 {
35 store float 1.000000e+01, float* @b, align 4, !tbaa !7
36 ret void
39 ; Function Attrs: nounwind ssp uwtable
40 define void @test() #0 {
41 %1 = call i32 @read_a(), !tbaa !3
42 call void @write_b(), !tbaa !7
43 ret void
46 !3 = !{!4, !4, i64 0}
47 !4 = !{!"int", !5, i64 0}
48 !5 = !{!"omnipotent char", !6, i64 0}
49 !6 = !{!"Simple C/C++ TBAA"}
50 !7 = !{!8, !8, i64 0}
51 !8 = !{!"float", !5, i64 0}
52 )";
54 // Parse the IR. The two calls in @test can not access aliasing elements.
55 LLVMContext Context;
56 SMDiagnostic Error;
57 auto M = parseAssemblyString(Assembly, Error, Context);
58 ASSERT_TRUE(M) << "Bad assembly?";
60 // Initialize the alias result.
61 Triple Trip(M->getTargetTriple());
62 TargetLibraryInfoImpl TLII(Trip);
63 TargetLibraryInfo TLI(TLII);
64 AAResults AA(TLI);
65 TypeBasedAAResult TBAAR;
66 AA.addAAResult(TBAAR);
68 // Initialize the alias set tracker for the @test function.
69 Function *Test = M->getFunction("test");
70 ASSERT_NE(Test, nullptr);
71 BatchAAResults BAA(AA);
72 AliasSetTracker AST(BAA);
73 for (auto &BB : *Test)
74 AST.add(BB);
75 // There should be 2 disjoint alias sets. 1 from each call.
76 ASSERT_EQ((int)AST.getAliasSets().size(), 2);
78 // Directly test aliasesUnknownInst.
79 // Now every call instruction should only alias one alias set.
80 BatchAAResults BatchAA(AA);
81 for (auto &Inst : *Test->begin()) {
82 bool FoundAS = false;
83 for (AliasSet &AS : AST) {
84 if (!Inst.mayReadOrWriteMemory())
85 continue;
86 if (!isModOrRefSet(AS.aliasesUnknownInst(&Inst, BatchAA)))
87 continue;
88 ASSERT_NE(FoundAS, true);
89 FoundAS = true;