265 lines
11 KiB
C++
265 lines
11 KiB
C++
//===- 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 identifiers[2] = {0, 1};
|
|
|
|
// Attach identifiers to domain ids.
|
|
space.getId(VarKind::Domain, 0) = Identifier(&identifiers[0]);
|
|
space.getId(VarKind::Domain, 1) = Identifier(&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(VarKind::Domain, 2), Identifier(&identifiers[0]));
|
|
EXPECT_EQ(space.getId(VarKind::Domain, 3), Identifier(&identifiers[1]));
|
|
}
|
|
|
|
TEST(PresburgerSpaceTest, removeVarRangeIdentifier) {
|
|
PresburgerSpace space = PresburgerSpace::getRelationSpace(2, 1, 3, 0);
|
|
space.resetIds();
|
|
|
|
int identifiers[6] = {0, 1, 2, 3, 4, 5};
|
|
|
|
// Attach identifiers to domain identifiers.
|
|
space.getId(VarKind::Domain, 0) = Identifier(&identifiers[0]);
|
|
space.getId(VarKind::Domain, 1) = Identifier(&identifiers[1]);
|
|
|
|
// Attach identifiers to range identifiers.
|
|
space.getId(VarKind::Range, 0) = Identifier(&identifiers[2]);
|
|
|
|
// Attach identifiers to symbol identifiers.
|
|
space.getId(VarKind::Symbol, 0) = Identifier(&identifiers[3]);
|
|
space.getId(VarKind::Symbol, 1) = Identifier(&identifiers[4]);
|
|
space.getId(VarKind::Symbol, 2) = Identifier(&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(VarKind::Domain, 0), Identifier(&identifiers[1]));
|
|
|
|
// Check if symbol identifiers are attached properly.
|
|
EXPECT_EQ(space.getId(VarKind::Range, 0), Identifier(&identifiers[4]));
|
|
EXPECT_EQ(space.getId(VarKind::Range, 1), Identifier(&identifiers[5]));
|
|
}
|
|
|
|
TEST(PresburgerSpaceTest, IdentifierIsEqual) {
|
|
PresburgerSpace space = PresburgerSpace::getRelationSpace(1, 2, 0, 0);
|
|
space.resetIds();
|
|
|
|
int identifiers[2] = {0, 1};
|
|
space.getId(VarKind::Domain, 0) = Identifier(&identifiers[0]);
|
|
space.getId(VarKind::Range, 0) = Identifier(&identifiers[0]);
|
|
space.getId(VarKind::Range, 1) = Identifier(&identifiers[1]);
|
|
|
|
EXPECT_EQ(space.getId(VarKind::Domain, 0), space.getId(VarKind::Range, 0));
|
|
EXPECT_FALSE(
|
|
space.getId(VarKind::Range, 0).isEqual(space.getId(VarKind::Range, 1)));
|
|
}
|
|
|
|
TEST(PresburgerSpaceTest, convertVarKind) {
|
|
PresburgerSpace space = PresburgerSpace::getRelationSpace(2, 2, 0, 0);
|
|
space.resetIds();
|
|
|
|
// Attach identifiers.
|
|
int identifiers[4] = {0, 1, 2, 3};
|
|
space.getId(VarKind::Domain, 0) = Identifier(&identifiers[0]);
|
|
space.getId(VarKind::Domain, 1) = Identifier(&identifiers[1]);
|
|
space.getId(VarKind::Range, 0) = Identifier(&identifiers[2]);
|
|
space.getId(VarKind::Range, 1) = Identifier(&identifiers[3]);
|
|
|
|
// Convert Range variables to symbols.
|
|
space.convertVarKind(VarKind::Range, 0, 2, VarKind::Symbol, 0);
|
|
|
|
// Check if the identifiers are moved to symbols.
|
|
EXPECT_EQ(space.getId(VarKind::Symbol, 0), Identifier(&identifiers[2]));
|
|
EXPECT_EQ(space.getId(VarKind::Symbol, 1), Identifier(&identifiers[3]));
|
|
|
|
// Convert 1 symbol to range identifier.
|
|
space.convertVarKind(VarKind::Symbol, 1, 1, VarKind::Range, 0);
|
|
|
|
// Check if the identifier is moved to range.
|
|
EXPECT_EQ(space.getId(VarKind::Range, 0), Identifier(&identifiers[3]));
|
|
}
|
|
|
|
TEST(PresburgerSpaceTest, convertVarKindLocals) {
|
|
PresburgerSpace space = PresburgerSpace::getRelationSpace(2, 2, 0, 0);
|
|
space.resetIds();
|
|
|
|
// Attach identifiers to range variables.
|
|
int identifiers[4] = {0, 1};
|
|
space.getId(VarKind::Range, 0) = Identifier(&identifiers[0]);
|
|
space.getId(VarKind::Range, 1) = Identifier(&identifiers[1]);
|
|
|
|
// Convert Range variables to locals i.e. project them out.
|
|
space.convertVarKind(VarKind::Range, 0, 2, VarKind::Local, 0);
|
|
|
|
// Check if the variables were moved.
|
|
EXPECT_EQ(space.getNumVarKind(VarKind::Range), 0u);
|
|
EXPECT_EQ(space.getNumVarKind(VarKind::Local), 2u);
|
|
|
|
// Convert the Local variables back to Range variables.
|
|
space.convertVarKind(VarKind::Local, 0, 2, VarKind::Range, 0);
|
|
|
|
// The identifier information should be lost.
|
|
EXPECT_FALSE(space.getId(VarKind::Range, 0).hasValue());
|
|
EXPECT_FALSE(space.getId(VarKind::Range, 1).hasValue());
|
|
}
|
|
|
|
TEST(PresburgerSpaceTest, convertVarKind2) {
|
|
PresburgerSpace space = PresburgerSpace::getRelationSpace(0, 2, 2, 0);
|
|
space.resetIds();
|
|
|
|
// Attach identifiers.
|
|
int identifiers[4] = {0, 1, 2, 3};
|
|
space.getId(VarKind::Range, 0) = Identifier(&identifiers[0]);
|
|
space.getId(VarKind::Range, 1) = Identifier(&identifiers[1]);
|
|
space.getId(VarKind::Symbol, 0) = Identifier(&identifiers[2]);
|
|
space.getId(VarKind::Symbol, 1) = Identifier(&identifiers[3]);
|
|
|
|
// Convert Range variables to symbols.
|
|
space.convertVarKind(VarKind::Range, 0, 2, VarKind::Symbol, 1);
|
|
|
|
// Check if the identifiers are moved to symbols.
|
|
EXPECT_EQ(space.getId(VarKind::Symbol, 0), Identifier(&identifiers[2]));
|
|
EXPECT_EQ(space.getId(VarKind::Symbol, 1), Identifier(&identifiers[0]));
|
|
EXPECT_EQ(space.getId(VarKind::Symbol, 2), Identifier(&identifiers[1]));
|
|
EXPECT_EQ(space.getId(VarKind::Symbol, 3), Identifier(&identifiers[3]));
|
|
}
|
|
|
|
TEST(PresburgerSpaceTest, mergeAndAlignSymbols) {
|
|
PresburgerSpace space = PresburgerSpace::getRelationSpace(3, 3, 2, 0);
|
|
space.resetIds();
|
|
|
|
PresburgerSpace otherSpace = PresburgerSpace::getRelationSpace(3, 2, 3, 0);
|
|
otherSpace.resetIds();
|
|
|
|
// Attach identifiers.
|
|
int identifiers[7] = {0, 1, 2, 3, 4, 5, 6};
|
|
int otherIdentifiers[8] = {10, 11, 12, 13, 14, 15, 16, 17};
|
|
|
|
space.getId(VarKind::Domain, 0) = Identifier(&identifiers[0]);
|
|
space.getId(VarKind::Domain, 1) = Identifier(&identifiers[1]);
|
|
// Note the common identifier.
|
|
space.getId(VarKind::Domain, 2) = Identifier(&otherIdentifiers[2]);
|
|
space.getId(VarKind::Range, 0) = Identifier(&identifiers[2]);
|
|
space.getId(VarKind::Range, 1) = Identifier(&identifiers[3]);
|
|
space.getId(VarKind::Range, 2) = Identifier(&identifiers[4]);
|
|
space.getId(VarKind::Symbol, 0) = Identifier(&identifiers[5]);
|
|
space.getId(VarKind::Symbol, 1) = Identifier(&identifiers[6]);
|
|
|
|
otherSpace.getId(VarKind::Domain, 0) = Identifier(&otherIdentifiers[0]);
|
|
otherSpace.getId(VarKind::Domain, 1) = Identifier(&otherIdentifiers[1]);
|
|
otherSpace.getId(VarKind::Domain, 2) = Identifier(&otherIdentifiers[2]);
|
|
otherSpace.getId(VarKind::Range, 0) = Identifier(&otherIdentifiers[3]);
|
|
otherSpace.getId(VarKind::Range, 1) = Identifier(&otherIdentifiers[4]);
|
|
// Note the common identifier.
|
|
otherSpace.getId(VarKind::Symbol, 0) = Identifier(&identifiers[6]);
|
|
otherSpace.getId(VarKind::Symbol, 1) = Identifier(&otherIdentifiers[5]);
|
|
otherSpace.getId(VarKind::Symbol, 2) = Identifier(&otherIdentifiers[7]);
|
|
|
|
space.mergeAndAlignSymbols(otherSpace);
|
|
|
|
// Check if merge & align is successful.
|
|
// Check symbol var identifiers.
|
|
EXPECT_EQ(4u, space.getNumSymbolVars());
|
|
EXPECT_EQ(4u, otherSpace.getNumSymbolVars());
|
|
EXPECT_EQ(space.getId(VarKind::Symbol, 0), Identifier(&identifiers[5]));
|
|
EXPECT_EQ(space.getId(VarKind::Symbol, 1), Identifier(&identifiers[6]));
|
|
EXPECT_EQ(space.getId(VarKind::Symbol, 2), Identifier(&otherIdentifiers[5]));
|
|
EXPECT_EQ(space.getId(VarKind::Symbol, 3), Identifier(&otherIdentifiers[7]));
|
|
EXPECT_EQ(otherSpace.getId(VarKind::Symbol, 0), Identifier(&identifiers[5]));
|
|
EXPECT_EQ(otherSpace.getId(VarKind::Symbol, 1), Identifier(&identifiers[6]));
|
|
EXPECT_EQ(otherSpace.getId(VarKind::Symbol, 2),
|
|
Identifier(&otherIdentifiers[5]));
|
|
EXPECT_EQ(otherSpace.getId(VarKind::Symbol, 3),
|
|
Identifier(&otherIdentifiers[7]));
|
|
// Check that domain and range var identifiers are not affected.
|
|
EXPECT_EQ(3u, space.getNumDomainVars());
|
|
EXPECT_EQ(3u, space.getNumRangeVars());
|
|
EXPECT_EQ(space.getId(VarKind::Domain, 0), Identifier(&identifiers[0]));
|
|
EXPECT_EQ(space.getId(VarKind::Domain, 1), Identifier(&identifiers[1]));
|
|
EXPECT_EQ(space.getId(VarKind::Domain, 2), Identifier(&otherIdentifiers[2]));
|
|
EXPECT_EQ(space.getId(VarKind::Range, 0), Identifier(&identifiers[2]));
|
|
EXPECT_EQ(space.getId(VarKind::Range, 1), Identifier(&identifiers[3]));
|
|
EXPECT_EQ(space.getId(VarKind::Range, 2), Identifier(&identifiers[4]));
|
|
EXPECT_EQ(3u, otherSpace.getNumDomainVars());
|
|
EXPECT_EQ(2u, otherSpace.getNumRangeVars());
|
|
EXPECT_EQ(otherSpace.getId(VarKind::Domain, 0),
|
|
Identifier(&otherIdentifiers[0]));
|
|
EXPECT_EQ(otherSpace.getId(VarKind::Domain, 1),
|
|
Identifier(&otherIdentifiers[1]));
|
|
EXPECT_EQ(otherSpace.getId(VarKind::Domain, 2),
|
|
Identifier(&otherIdentifiers[2]));
|
|
EXPECT_EQ(otherSpace.getId(VarKind::Range, 0),
|
|
Identifier(&otherIdentifiers[3]));
|
|
EXPECT_EQ(otherSpace.getId(VarKind::Range, 1),
|
|
Identifier(&otherIdentifiers[4]));
|
|
}
|