DeltaTests.cpp

Go to the documentation of this file.

/*
 * Copyright 2018 Nico Reißmann <nico.reissmann@gmail.com>
 * See COPYING for terms of redistribution.
 */
 
#include <gtest/gtest.h>
 
#include <jlm/llvm/ir/operators/delta.hpp>
#include <jlm/llvm/ir/RvsdgModule.hpp>
#include <jlm/rvsdg/TestOperations.hpp>
#include <jlm/rvsdg/TestType.hpp>
#include <jlm/rvsdg/view.hpp>
 
TEST(DeltaTests, TestDeltaCreation)
{
  using namespace jlm::llvm;
  using namespace jlm::rvsdg;
 
  // Arrange & Act
  auto valueType = TestType::createValueType();
  auto pointerType = PointerType::Create();
  jlm::llvm::LlvmRvsdgModule rvsdgModule(jlm::util::FilePath(""), "", "");
 
  auto imp = &jlm::rvsdg::GraphImport::Create(rvsdgModule.Rvsdg(), valueType, "");
 
  auto delta1 = jlm::rvsdg::DeltaNode::Create(
      &rvsdgModule.Rvsdg().GetRootRegion(),
      LlvmDeltaOperation::Create(valueType, "test-delta1", Linkage::externalLinkage, "", true, 4));
  auto dep = delta1->AddContextVar(*imp).inner;
  auto d1 = &delta1->finalize(
      TestOperation::createNode(delta1->subregion(), { dep }, { valueType })->output(0));
 
  auto delta2 = jlm::rvsdg::DeltaNode::Create(
      &rvsdgModule.Rvsdg().GetRootRegion(),
      LlvmDeltaOperation::Create(valueType, "test-delta2", Linkage::internalLinkage, "", false, 4));
  auto d2 = &delta2->finalize(
      TestOperation::createNode(delta2->subregion(), {}, { valueType })->output(0));
 
  GraphExport::Create(*d1, "");
  GraphExport::Create(*d2, "");
 
  jlm::rvsdg::view(rvsdgModule.Rvsdg(), stdout);
 
  // Assert
  EXPECT_EQ(rvsdgModule.Rvsdg().GetRootRegion().numNodes(), 2u);
 
  EXPECT_TRUE(delta1->constant());
  EXPECT_EQ(*delta1->Type(), *valueType);
 
  EXPECT_FALSE(delta2->constant());
  EXPECT_EQ(*delta2->Type(), *valueType);
}
 
TEST(DeltaTests, TestRemoveDeltaInputsWhere)
{
  using namespace jlm::llvm;
  using namespace jlm::rvsdg;
 
  // Arrange
  auto valueType = TestType::createValueType();
  jlm::llvm::LlvmRvsdgModule rvsdgModule(jlm::util::FilePath(""), "", "");
 
  auto x = &jlm::rvsdg::GraphImport::Create(rvsdgModule.Rvsdg(), valueType, "");
 
  auto deltaNode = jlm::rvsdg::DeltaNode::Create(
      &rvsdgModule.Rvsdg().GetRootRegion(),
      LlvmDeltaOperation::Create(valueType, "delta", Linkage::externalLinkage, "", true, 4));
  auto deltaInput0 = deltaNode->AddContextVar(*x).input;
  auto deltaInput1 = deltaNode->AddContextVar(*x).input;
  deltaNode->AddContextVar(*x);
 
  auto result = jlm::rvsdg::CreateOpNode<TestOperation>(
                    { deltaNode->MapInputContextVar(*deltaInput1).inner },
                    std::vector<std::shared_ptr<const Type>>{ valueType },
                    std::vector<std::shared_ptr<const Type>>{ valueType })
                    .output(0);
 
  deltaNode->finalize(result);
 
  // Act & Assert
  // Try to remove deltaInput1 even though it is used
  auto numRemovedInputs = deltaNode->RemoveDeltaInputsWhere(
      [&](const jlm::rvsdg::Input & input)
      {
        return input.index() == deltaInput1->index();
      });
  EXPECT_EQ(numRemovedInputs, 0u);
  EXPECT_EQ(deltaNode->ninputs(), 3u);
  EXPECT_EQ(deltaNode->GetContextVars().size(), 3u);
 
  // Remove deltaInput2
  numRemovedInputs = deltaNode->RemoveDeltaInputsWhere(
      [&](const jlm::rvsdg::Input & input)
      {
        return input.index() == 2;
      });
  EXPECT_EQ(numRemovedInputs, 1u);
  EXPECT_EQ(deltaNode->ninputs(), 2u);
  EXPECT_EQ(deltaNode->GetContextVars().size(), 2u);
  EXPECT_EQ(deltaNode->input(0), deltaInput0);
  EXPECT_EQ(deltaNode->input(1), deltaInput1);
 
  // Remove deltaInput0
  numRemovedInputs = deltaNode->RemoveDeltaInputsWhere(
      [&](const jlm::rvsdg::Input & input)
      {
        return input.index() == 0;
      });
  EXPECT_EQ(numRemovedInputs, 1u);
  EXPECT_EQ(deltaNode->ninputs(), 1u);
  EXPECT_EQ(deltaNode->GetContextVars().size(), 1u);
  EXPECT_EQ(deltaNode->input(0), deltaInput1);
  EXPECT_EQ(deltaInput1->index(), 0u);
  EXPECT_EQ(deltaNode->MapInputContextVar(*deltaInput1).inner->index(), 0u);
}
 
TEST(DeltaTests, TestPruneDeltaInputs)
{
  using namespace jlm::llvm;
  using namespace jlm::rvsdg;
 
  // Arrange
  auto valueType = TestType::createValueType();
  jlm::llvm::LlvmRvsdgModule rvsdgModule(jlm::util::FilePath(""), "", "");
 
  auto x = &jlm::rvsdg::GraphImport::Create(rvsdgModule.Rvsdg(), valueType, "");
 
  auto deltaNode = jlm::rvsdg::DeltaNode::Create(
      &rvsdgModule.Rvsdg().GetRootRegion(),
      LlvmDeltaOperation::Create(valueType, "delta", Linkage::externalLinkage, "", true, 4));
 
  deltaNode->AddContextVar(*x);
  auto deltaInput1 = deltaNode->AddContextVar(*x).input;
  deltaNode->AddContextVar(*x);
 
  auto result = jlm::rvsdg::CreateOpNode<TestOperation>(
                    { deltaNode->MapInputContextVar(*deltaInput1).inner },
                    std::vector<std::shared_ptr<const Type>>{ valueType },
                    std::vector<std::shared_ptr<const Type>>{ valueType })
                    .output(0);
 
  deltaNode->finalize(result);
 
  // Act
  auto numRemovedInputs = deltaNode->PruneDeltaInputs();
 
  // Assert
  EXPECT_EQ(numRemovedInputs, 2u);
  EXPECT_EQ(deltaNode->ninputs(), 1u);
  EXPECT_EQ(deltaNode->GetContextVars().size(), 1u);
  EXPECT_EQ(deltaNode->input(0), deltaInput1);
  EXPECT_EQ(deltaNode->subregion()->argument(0), deltaNode->MapInputContextVar(*deltaInput1).inner);
  EXPECT_EQ(deltaInput1->index(), 0u);
  EXPECT_EQ(deltaNode->MapInputContextVar(*deltaInput1).inner->index(), 0u);
}