1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
|
//===- PresburgerSpaceTest.cpp - Tests for PresburgerSpace ----------------===//
//
// 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/Presburger/PresburgerSpace.h"
#include <gmock/gmock.h>
#include <gtest/gtest.h>
using namespace mlir;
using namespace presburger;
TEST(PresburgerSpaceTest, insertId) {
PresburgerSpace space = PresburgerSpace::getRelationSpace(2, 2, 1);
// Try inserting 2 domain ids.
space.insertVar(VarKind::Domain, 0, 2);
EXPECT_EQ(space.getNumDomainVars(), 4u);
// Try inserting 1 range ids.
space.insertVar(VarKind::Range, 0, 1);
EXPECT_EQ(space.getNumRangeVars(), 3u);
}
TEST(PresburgerSpaceTest, insertIdSet) {
PresburgerSpace space = PresburgerSpace::getSetSpace(2, 1);
// Try inserting 2 dimension ids. The space should have 4 range ids since
// spaces which do not distinguish between domain, range are implemented like
// this.
space.insertVar(VarKind::SetDim, 0, 2);
EXPECT_EQ(space.getNumRangeVars(), 4u);
}
TEST(PresburgerSpaceTest, removeIdRange) {
PresburgerSpace space = PresburgerSpace::getRelationSpace(2, 1, 3);
// Remove 1 domain identifier.
space.removeVarRange(VarKind::Domain, 0, 1);
EXPECT_EQ(space.getNumDomainVars(), 1u);
// Remove 1 symbol and 1 range identifier.
space.removeVarRange(VarKind::Symbol, 0, 1);
space.removeVarRange(VarKind::Range, 0, 1);
EXPECT_EQ(space.getNumDomainVars(), 1u);
EXPECT_EQ(space.getNumRangeVars(), 0u);
EXPECT_EQ(space.getNumSymbolVars(), 2u);
}
TEST(PresburgerSpaceTest, insertVarIdentifier) {
PresburgerSpace space = PresburgerSpace::getRelationSpace(2, 2, 1, 0);
space.resetIds<int *>();
// Attach identifiers to domain ids.
int identifiers[2] = {0, 1};
space.setId<int *>(VarKind::Domain, 0, &identifiers[0]);
space.setId<int *>(VarKind::Domain, 1, &identifiers[1]);
// Try inserting 2 domain ids.
space.insertVar(VarKind::Domain, 0, 2);
EXPECT_EQ(space.getNumDomainVars(), 4u);
// Try inserting 1 range ids.
space.insertVar(VarKind::Range, 0, 1);
EXPECT_EQ(space.getNumRangeVars(), 3u);
// Check if the identifiers for the old ids are still attached properly.
EXPECT_EQ(*space.getId<int *>(VarKind::Domain, 2), identifiers[0]);
EXPECT_EQ(*space.getId<int *>(VarKind::Domain, 3), identifiers[1]);
}
TEST(PresburgerSpaceTest, removeVarRangeIdentifier) {
PresburgerSpace space = PresburgerSpace::getRelationSpace(2, 1, 3, 0);
space.resetIds<int *>();
int identifiers[6] = {0, 1, 2, 3, 4, 5};
// Attach identifiers to domain identifiers.
space.setId<int *>(VarKind::Domain, 0, &identifiers[0]);
space.setId<int *>(VarKind::Domain, 1, &identifiers[1]);
// Attach identifiers to range identifiers.
space.setId<int *>(VarKind::Range, 0, &identifiers[2]);
// Attach identifiers to symbol identifiers.
space.setId<int *>(VarKind::Symbol, 0, &identifiers[3]);
space.setId<int *>(VarKind::Symbol, 1, &identifiers[4]);
space.setId<int *>(VarKind::Symbol, 2, &identifiers[5]);
// Remove 1 domain identifier.
space.removeVarRange(VarKind::Domain, 0, 1);
EXPECT_EQ(space.getNumDomainVars(), 1u);
// Remove 1 symbol and 1 range identifier.
space.removeVarRange(VarKind::Symbol, 0, 1);
space.removeVarRange(VarKind::Range, 0, 1);
EXPECT_EQ(space.getNumDomainVars(), 1u);
EXPECT_EQ(space.getNumRangeVars(), 0u);
EXPECT_EQ(space.getNumSymbolVars(), 2u);
// Check if domain identifiers are attached properly.
EXPECT_EQ(*space.getId<int *>(VarKind::Domain, 0), identifiers[1]);
// Check if symbol identifiers are attached properly.
EXPECT_EQ(*space.getId<int *>(VarKind::Range, 0), identifiers[4]);
EXPECT_EQ(*space.getId<int *>(VarKind::Range, 1), identifiers[5]);
}
|
