[llvm] [SandboxIR][NFC] Factor out common test for CastInst subclasses (PR #101410)
via llvm-commits
llvm-commits at lists.llvm.org
Wed Jul 31 13:52:46 PDT 2024
https://github.com/vporpo created https://github.com/llvm/llvm-project/pull/101410
The tests for most CastInst sub-classes, except AddrSpaceCastInst, are very similar.
This patch creates a common template function for all of them.
>From 264441aa35ff1592db4cd0a8136c9e55f3e12cce Mon Sep 17 00:00:00 2001
From: Vasileios Porpodas <vporpodas at google.com>
Date: Wed, 31 Jul 2024 12:59:36 -0700
Subject: [PATCH] [SandboxIR][NFC] Factor out common test for CastInst
subclasses
The tests for most CastInst sub-classes, except AddrSpaceCastInst, are very
similar. This patch creates a common template function for all of them.
---
llvm/unittests/SandboxIR/SandboxIRTest.cpp | 465 ++++-----------------
1 file changed, 71 insertions(+), 394 deletions(-)
diff --git a/llvm/unittests/SandboxIR/SandboxIRTest.cpp b/llvm/unittests/SandboxIR/SandboxIRTest.cpp
index 9d4fba404a43c..813f8138b9d8e 100644
--- a/llvm/unittests/SandboxIR/SandboxIRTest.cpp
+++ b/llvm/unittests/SandboxIR/SandboxIRTest.cpp
@@ -34,6 +34,8 @@ struct SandboxIRTest : public testing::Test {
return &BB;
llvm_unreachable("Expected to find basic block!");
}
+ template <typename SubclassT, sandboxir::Instruction::Opcode OpcodeT>
+ void testCastInst(const char *IRToParse, Type *SrcTy, Type *DstTy);
};
TEST_F(SandboxIRTest, ClassID) {
@@ -1620,13 +1622,12 @@ define void @foo(i32 %arg, float %farg, double %darg, ptr %ptr) {
}
}
-TEST_F(SandboxIRTest, SIToFPInst) {
- parseIR(C, R"IR(
-define void @foo(i32 %arg) {
- %sitofp = sitofp i32 %arg to float
- ret void
-}
-)IR");
+/// CastInst's subclasses are very similar so we can use a common test function
+/// for them.
+template <typename SubclassT, sandboxir::Instruction::Opcode OpcodeT>
+void SandboxIRTest::testCastInst(const char *IRToParse, Type *SrcTy,
+ Type *DstTy) {
+ parseIR(C, IRToParse);
Function &LLVMF = *M->getFunction("foo");
sandboxir::Context Ctx(C);
sandboxir::Function *F = Ctx.createFunction(&LLVMF);
@@ -1634,413 +1635,142 @@ define void @foo(i32 %arg) {
auto *Arg = F->getArg(ArgIdx++);
auto *BB = &*F->begin();
auto It = BB->begin();
- Type *Ti32 = Type::getInt32Ty(C);
- Type *Tfloat = Type::getFloatTy(C);
- auto *SIToFP = cast<sandboxir::SIToFPInst>(&*It++);
- EXPECT_EQ(SIToFP->getOpcode(), sandboxir::Instruction::Opcode::SIToFP);
- EXPECT_EQ(SIToFP->getSrcTy(), Ti32);
- EXPECT_EQ(SIToFP->getDestTy(), Tfloat);
+ auto *CI = cast<SubclassT>(&*It++);
+ EXPECT_EQ(CI->getOpcode(), OpcodeT);
+ EXPECT_EQ(CI->getSrcTy(), SrcTy);
+ EXPECT_EQ(CI->getDestTy(), DstTy);
auto *Ret = cast<sandboxir::ReturnInst>(&*It++);
{
// Check create() WhereIt, WhereBB
- auto *NewI = cast<sandboxir::SIToFPInst>(
- sandboxir::SIToFPInst::create(Arg, Tfloat, /*WhereIt=*/BB->end(),
- /*WhereBB=*/BB, Ctx, "SIToFP"));
+ auto *NewI =
+ cast<SubclassT>(SubclassT::create(Arg, DstTy, /*WhereIt=*/BB->end(),
+ /*WhereBB=*/BB, Ctx, "NewCI"));
// Check getOpcode().
- EXPECT_EQ(NewI->getOpcode(), sandboxir::Instruction::Opcode::SIToFP);
+ EXPECT_EQ(NewI->getOpcode(), OpcodeT);
// Check getSrcTy().
EXPECT_EQ(NewI->getSrcTy(), Arg->getType());
// Check getDestTy().
- EXPECT_EQ(NewI->getDestTy(), Tfloat);
+ EXPECT_EQ(NewI->getDestTy(), DstTy);
// Check instr position.
EXPECT_EQ(NewI->getNextNode(), nullptr);
EXPECT_EQ(NewI->getPrevNode(), Ret);
+ // Check instr name.
+ EXPECT_EQ(NewI->getName(), "NewCI");
}
{
// Check create() InsertBefore.
- auto *NewI = cast<sandboxir::SIToFPInst>(
- sandboxir::SIToFPInst::create(Arg, Tfloat,
- /*InsertBefore=*/Ret, Ctx, "SIToFP"));
+ auto *NewI =
+ cast<SubclassT>(SubclassT::create(Arg, DstTy,
+ /*InsertBefore=*/Ret, Ctx, "NewCI"));
// Check getOpcode().
- EXPECT_EQ(NewI->getOpcode(), sandboxir::Instruction::Opcode::SIToFP);
+ EXPECT_EQ(NewI->getOpcode(), OpcodeT);
// Check getSrcTy().
EXPECT_EQ(NewI->getSrcTy(), Arg->getType());
// Check getDestTy().
- EXPECT_EQ(NewI->getDestTy(), Tfloat);
+ EXPECT_EQ(NewI->getDestTy(), DstTy);
// Check instr position.
EXPECT_EQ(NewI->getNextNode(), Ret);
}
{
// Check create() InsertAtEnd.
- auto *NewI = cast<sandboxir::SIToFPInst>(
- sandboxir::SIToFPInst::create(Arg, Tfloat,
- /*InsertAtEnd=*/BB, Ctx, "SIToFP"));
+ auto *NewI =
+ cast<SubclassT>(SubclassT::create(Arg, DstTy,
+ /*InsertAtEnd=*/BB, Ctx, "NewCI"));
// Check getOpcode().
- EXPECT_EQ(NewI->getOpcode(), sandboxir::Instruction::Opcode::SIToFP);
+ EXPECT_EQ(NewI->getOpcode(), OpcodeT);
// Check getSrcTy().
EXPECT_EQ(NewI->getSrcTy(), Arg->getType());
// Check getDestTy().
- EXPECT_EQ(NewI->getDestTy(), Tfloat);
+ EXPECT_EQ(NewI->getDestTy(), DstTy);
// Check instr position.
EXPECT_EQ(NewI->getNextNode(), nullptr);
EXPECT_EQ(NewI->getParent(), BB);
}
}
+TEST_F(SandboxIRTest, SIToFPInst) {
+ testCastInst<sandboxir::SIToFPInst, sandboxir::Instruction::Opcode::SIToFP>(
+ R"IR(
+define void @foo(i32 %arg) {
+ %sitofp = sitofp i32 %arg to float
+ ret void
+}
+)IR",
+ /*SrcTy=*/Type::getInt32Ty(C),
+ /*DstTy=*/Type::getFloatTy(C));
+}
+
TEST_F(SandboxIRTest, FPToUIInst) {
- parseIR(C, R"IR(
+ testCastInst<sandboxir::FPToUIInst, sandboxir::Instruction::Opcode::FPToUI>(
+ R"IR(
define void @foo(float %arg) {
%fptoui = fptoui float %arg to i32
ret void
}
-)IR");
- Function &LLVMF = *M->getFunction("foo");
- sandboxir::Context Ctx(C);
- sandboxir::Function *F = Ctx.createFunction(&LLVMF);
- unsigned ArgIdx = 0;
- auto *Arg = F->getArg(ArgIdx++);
- auto *BB = &*F->begin();
- auto It = BB->begin();
- Type *Ti32 = Type::getInt32Ty(C);
- Type *Tfloat = Type::getFloatTy(C);
-
- auto *FPToUI = cast<sandboxir::FPToUIInst>(&*It++);
- EXPECT_EQ(FPToUI->getOpcode(), sandboxir::Instruction::Opcode::FPToUI);
- EXPECT_EQ(FPToUI->getSrcTy(), Tfloat);
- EXPECT_EQ(FPToUI->getDestTy(), Ti32);
- auto *Ret = cast<sandboxir::ReturnInst>(&*It++);
-
- {
- // Check create() WhereIt, WhereBB
- auto *NewI = cast<sandboxir::FPToUIInst>(
- sandboxir::FPToUIInst::create(Arg, Ti32, /*WhereIt=*/BB->end(),
- /*WhereBB=*/BB, Ctx, "FPToUI"));
- // Check getOpcode().
- EXPECT_EQ(NewI->getOpcode(), sandboxir::Instruction::Opcode::FPToUI);
- // Check getSrcTy().
- EXPECT_EQ(NewI->getSrcTy(), Arg->getType());
- // Check getDestTy().
- EXPECT_EQ(NewI->getDestTy(), Ti32);
- // Check instr position.
- EXPECT_EQ(NewI->getNextNode(), nullptr);
- EXPECT_EQ(NewI->getPrevNode(), Ret);
- }
- {
- // Check create() InsertBefore.
- auto *NewI = cast<sandboxir::FPToUIInst>(
- sandboxir::FPToUIInst::create(Arg, Ti32,
- /*InsertBefore=*/Ret, Ctx, "FPToUI"));
- // Check getOpcode().
- EXPECT_EQ(NewI->getOpcode(), sandboxir::Instruction::Opcode::FPToUI);
- // Check getSrcTy().
- EXPECT_EQ(NewI->getSrcTy(), Arg->getType());
- // Check getDestTy().
- EXPECT_EQ(NewI->getDestTy(), Ti32);
- // Check instr position.
- EXPECT_EQ(NewI->getNextNode(), Ret);
- }
- {
- // Check create() InsertAtEnd.
- auto *NewI = cast<sandboxir::FPToUIInst>(
- sandboxir::FPToUIInst::create(Arg, Ti32,
- /*InsertAtEnd=*/BB, Ctx, "FPToUI"));
- // Check getOpcode().
- EXPECT_EQ(NewI->getOpcode(), sandboxir::Instruction::Opcode::FPToUI);
- // Check getSrcTy().
- EXPECT_EQ(NewI->getSrcTy(), Arg->getType());
- // Check getDestTy().
- EXPECT_EQ(NewI->getDestTy(), Ti32);
- // Check instr position.
- EXPECT_EQ(NewI->getNextNode(), nullptr);
- EXPECT_EQ(NewI->getParent(), BB);
- }
+)IR",
+ /*SrcTy=*/Type::getFloatTy(C), /*DstTy=*/Type::getInt32Ty(C));
}
TEST_F(SandboxIRTest, FPToSIInst) {
- parseIR(C, R"IR(
+ testCastInst<sandboxir::FPToSIInst, sandboxir::Instruction::Opcode::FPToSI>(
+ R"IR(
define void @foo(float %arg) {
%fptosi = fptosi float %arg to i32
ret void
}
-)IR");
- Function &LLVMF = *M->getFunction("foo");
- sandboxir::Context Ctx(C);
- sandboxir::Function *F = Ctx.createFunction(&LLVMF);
- unsigned ArgIdx = 0;
- auto *Arg = F->getArg(ArgIdx++);
- auto *BB = &*F->begin();
- auto It = BB->begin();
- Type *Ti32 = Type::getInt32Ty(C);
- Type *Tfloat = Type::getFloatTy(C);
-
- auto *FPToSI = cast<sandboxir::FPToSIInst>(&*It++);
- EXPECT_EQ(FPToSI->getOpcode(), sandboxir::Instruction::Opcode::FPToSI);
- EXPECT_EQ(FPToSI->getSrcTy(), Tfloat);
- EXPECT_EQ(FPToSI->getDestTy(), Ti32);
- auto *Ret = cast<sandboxir::ReturnInst>(&*It++);
-
- {
- // Check create() WhereIt, WhereBB
- auto *NewI = cast<sandboxir::FPToSIInst>(
- sandboxir::FPToSIInst::create(Arg, Ti32, /*WhereIt=*/BB->end(),
- /*WhereBB=*/BB, Ctx, "FPToSI"));
- // Check getOpcode().
- EXPECT_EQ(NewI->getOpcode(), sandboxir::Instruction::Opcode::FPToSI);
- // Check getSrcTy().
- EXPECT_EQ(NewI->getSrcTy(), Arg->getType());
- // Check getDestTy().
- EXPECT_EQ(NewI->getDestTy(), Ti32);
- // Check instr position.
- EXPECT_EQ(NewI->getNextNode(), nullptr);
- EXPECT_EQ(NewI->getPrevNode(), Ret);
- }
- {
- // Check create() InsertBefore.
- auto *NewI = cast<sandboxir::FPToSIInst>(
- sandboxir::FPToSIInst::create(Arg, Ti32,
- /*InsertBefore=*/Ret, Ctx, "FPToSI"));
- // Check getOpcode().
- EXPECT_EQ(NewI->getOpcode(), sandboxir::Instruction::Opcode::FPToSI);
- // Check getSrcTy().
- EXPECT_EQ(NewI->getSrcTy(), Arg->getType());
- // Check getDestTy().
- EXPECT_EQ(NewI->getDestTy(), Ti32);
- // Check instr position.
- EXPECT_EQ(NewI->getNextNode(), Ret);
- }
- {
- // Check create() InsertAtEnd.
- auto *NewI = cast<sandboxir::FPToSIInst>(
- sandboxir::FPToSIInst::create(Arg, Ti32,
- /*InsertAtEnd=*/BB, Ctx, "FPToSI"));
- // Check getOpcode().
- EXPECT_EQ(NewI->getOpcode(), sandboxir::Instruction::Opcode::FPToSI);
- // Check getSrcTy().
- EXPECT_EQ(NewI->getSrcTy(), Arg->getType());
- // Check getDestTy().
- EXPECT_EQ(NewI->getDestTy(), Ti32);
- // Check instr position.
- EXPECT_EQ(NewI->getNextNode(), nullptr);
- EXPECT_EQ(NewI->getParent(), BB);
- }
+)IR",
+ /*SrcTy=*/Type::getFloatTy(C), /*DstTy=*/Type::getInt32Ty(C));
}
TEST_F(SandboxIRTest, IntToPtrInst) {
- parseIR(C, R"IR(
+ testCastInst<sandboxir::IntToPtrInst,
+ sandboxir::Instruction::Opcode::IntToPtr>(
+ R"IR(
define void @foo(i32 %arg) {
%inttoptr = inttoptr i32 %arg to ptr
ret void
}
-)IR");
- Function &LLVMF = *M->getFunction("foo");
- sandboxir::Context Ctx(C);
- sandboxir::Function *F = Ctx.createFunction(&LLVMF);
- unsigned ArgIdx = 0;
- auto *Arg = F->getArg(ArgIdx++);
- auto *BB = &*F->begin();
- auto It = BB->begin();
- Type *Ti32 = Type::getInt32Ty(C);
- Type *Tptr = Ti32->getPointerTo();
-
- auto *IntToPtr = cast<sandboxir::IntToPtrInst>(&*It++);
- EXPECT_EQ(IntToPtr->getOpcode(), sandboxir::Instruction::Opcode::IntToPtr);
- EXPECT_EQ(IntToPtr->getSrcTy(), Ti32);
- EXPECT_EQ(IntToPtr->getDestTy(), Tptr);
- auto *Ret = cast<sandboxir::ReturnInst>(&*It++);
-
- {
- // Check create() WhereIt, WhereBB
- auto *NewI = cast<sandboxir::IntToPtrInst>(
- sandboxir::IntToPtrInst::create(Arg, Tptr, /*WhereIt=*/BB->end(),
- /*WhereBB=*/BB, Ctx, "IntToPtr"));
- // Check getOpcode().
- EXPECT_EQ(NewI->getOpcode(), sandboxir::Instruction::Opcode::IntToPtr);
- // Check getSrcTy().
- EXPECT_EQ(NewI->getSrcTy(), Arg->getType());
- // Check getDestTy().
- EXPECT_EQ(NewI->getDestTy(), Tptr);
- // Check instr position.
- EXPECT_EQ(NewI->getNextNode(), nullptr);
- EXPECT_EQ(NewI->getPrevNode(), Ret);
- }
- {
- // Check create() InsertBefore.
- auto *NewI = cast<sandboxir::IntToPtrInst>(
- sandboxir::IntToPtrInst::create(Arg, Tptr,
- /*InsertBefore=*/Ret, Ctx, "IntToPtr"));
- // Check getOpcode().
- EXPECT_EQ(NewI->getOpcode(), sandboxir::Instruction::Opcode::IntToPtr);
- // Check getSrcTy().
- EXPECT_EQ(NewI->getSrcTy(), Arg->getType());
- // Check getDestTy().
- EXPECT_EQ(NewI->getDestTy(), Tptr);
- // Check instr position.
- EXPECT_EQ(NewI->getNextNode(), Ret);
- }
- {
- // Check create() InsertAtEnd.
- auto *NewI = cast<sandboxir::IntToPtrInst>(
- sandboxir::IntToPtrInst::create(Arg, Tptr,
- /*InsertAtEnd=*/BB, Ctx, "IntToPtr"));
- // Check getOpcode().
- EXPECT_EQ(NewI->getOpcode(), sandboxir::Instruction::Opcode::IntToPtr);
- // Check getSrcTy().
- EXPECT_EQ(NewI->getSrcTy(), Arg->getType());
- // Check getDestTy().
- EXPECT_EQ(NewI->getDestTy(), Tptr);
- // Check instr position.
- EXPECT_EQ(NewI->getNextNode(), nullptr);
- EXPECT_EQ(NewI->getParent(), BB);
- }
+)IR",
+ /*SrcTy=*/Type::getInt32Ty(C), /*DstTy=*/PointerType::get(C, 0));
}
TEST_F(SandboxIRTest, PtrToIntInst) {
- parseIR(C, R"IR(
+ testCastInst<sandboxir::PtrToIntInst,
+ sandboxir::Instruction::Opcode::PtrToInt>(
+ R"IR(
define void @foo(ptr %ptr) {
%ptrtoint = ptrtoint ptr %ptr to i32
ret void
}
-)IR");
- Function &LLVMF = *M->getFunction("foo");
- sandboxir::Context Ctx(C);
- sandboxir::Function *F = Ctx.createFunction(&LLVMF);
- unsigned ArgIdx = 0;
- auto *Arg = F->getArg(ArgIdx++);
- auto *BB = &*F->begin();
- auto It = BB->begin();
- Type *Ti32 = Type::getInt32Ty(C);
- Type *Tptr = Ti32->getPointerTo();
-
- auto *PtrToInt = cast<sandboxir::PtrToIntInst>(&*It++);
- EXPECT_EQ(PtrToInt->getOpcode(), sandboxir::Instruction::Opcode::PtrToInt);
- EXPECT_EQ(PtrToInt->getSrcTy(), Tptr);
- EXPECT_EQ(PtrToInt->getDestTy(), Ti32);
- auto *Ret = cast<sandboxir::ReturnInst>(&*It++);
-
- {
- // Check create() WhereIt, WhereBB
- auto *NewI = cast<sandboxir::PtrToIntInst>(
- sandboxir::PtrToIntInst::create(Arg, Ti32, /*WhereIt=*/BB->end(),
- /*WhereBB=*/BB, Ctx, "PtrToInt"));
- // Check getOpcode().
- EXPECT_EQ(NewI->getOpcode(), sandboxir::Instruction::Opcode::PtrToInt);
- // Check getSrcTy().
- EXPECT_EQ(NewI->getSrcTy(), Arg->getType());
- // Check getDestTy().
- EXPECT_EQ(NewI->getDestTy(), Ti32);
- // Check instr position.
- EXPECT_EQ(NewI->getNextNode(), nullptr);
- EXPECT_EQ(NewI->getPrevNode(), Ret);
- }
- {
- // Check create() InsertBefore.
- auto *NewI = cast<sandboxir::PtrToIntInst>(
- sandboxir::PtrToIntInst::create(Arg, Ti32,
- /*InsertBefore=*/Ret, Ctx, "PtrToInt"));
- // Check getOpcode().
- EXPECT_EQ(NewI->getOpcode(), sandboxir::Instruction::Opcode::PtrToInt);
- // Check getSrcTy().
- EXPECT_EQ(NewI->getSrcTy(), Arg->getType());
- // Check getDestTy().
- EXPECT_EQ(NewI->getDestTy(), Ti32);
- // Check instr position.
- EXPECT_EQ(NewI->getNextNode(), Ret);
- }
- {
- // Check create() InsertAtEnd.
- auto *NewI = cast<sandboxir::PtrToIntInst>(
- sandboxir::PtrToIntInst::create(Arg, Ti32,
- /*InsertAtEnd=*/BB, Ctx, "PtrToInt"));
- // Check getOpcode().
- EXPECT_EQ(NewI->getOpcode(), sandboxir::Instruction::Opcode::PtrToInt);
- // Check getSrcTy().
- EXPECT_EQ(NewI->getSrcTy(), Arg->getType());
- // Check getDestTy().
- EXPECT_EQ(NewI->getDestTy(), Ti32);
- // Check instr position.
- EXPECT_EQ(NewI->getNextNode(), nullptr);
- EXPECT_EQ(NewI->getParent(), BB);
- }
+)IR",
+ /*SrcTy=*/PointerType::get(C, 0), /*DstTy=*/Type::getInt32Ty(C));
}
TEST_F(SandboxIRTest, BitCastInst) {
- parseIR(C, R"IR(
+ testCastInst<sandboxir::BitCastInst, sandboxir::Instruction::Opcode::BitCast>(
+ R"IR(
define void @foo(i32 %arg) {
%bitcast = bitcast i32 %arg to float
ret void
}
-)IR");
- Function &LLVMF = *M->getFunction("foo");
- sandboxir::Context Ctx(C);
- sandboxir::Function *F = Ctx.createFunction(&LLVMF);
- unsigned ArgIdx = 0;
- auto *Arg = F->getArg(ArgIdx++);
- auto *BB = &*F->begin();
- auto It = BB->begin();
- Type *Ti32 = Type::getInt32Ty(C);
- Type *Tfloat = Type::getFloatTy(C);
-
- auto *BitCast = cast<sandboxir::BitCastInst>(&*It++);
- EXPECT_EQ(BitCast->getOpcode(), sandboxir::Instruction::Opcode::BitCast);
- EXPECT_EQ(BitCast->getSrcTy(), Ti32);
- EXPECT_EQ(BitCast->getDestTy(), Tfloat);
- auto *Ret = cast<sandboxir::ReturnInst>(&*It++);
-
- {
- // Check create() WhereIt, WhereBB
- auto *NewI = cast<sandboxir::BitCastInst>(
- sandboxir::BitCastInst::create(Arg, Tfloat, /*WhereIt=*/BB->end(),
- /*WhereBB=*/BB, Ctx, "BitCast"));
- // Check getOpcode().
- EXPECT_EQ(NewI->getOpcode(), sandboxir::Instruction::Opcode::BitCast);
- // Check getSrcTy().
- EXPECT_EQ(NewI->getSrcTy(), Arg->getType());
- // Check getDestTy().
- EXPECT_EQ(NewI->getDestTy(), Tfloat);
- // Check instr position.
- EXPECT_EQ(NewI->getNextNode(), nullptr);
- EXPECT_EQ(NewI->getPrevNode(), Ret);
- }
- {
- // Check create() InsertBefore.
- auto *NewI = cast<sandboxir::BitCastInst>(
- sandboxir::BitCastInst::create(Arg, Tfloat,
- /*InsertBefore=*/Ret, Ctx, "BitCast"));
- // Check getOpcode().
- EXPECT_EQ(NewI->getOpcode(), sandboxir::Instruction::Opcode::BitCast);
- // Check getSrcTy().
- EXPECT_EQ(NewI->getSrcTy(), Arg->getType());
- // Check getDestTy().
- EXPECT_EQ(NewI->getDestTy(), Tfloat);
- // Check instr position.
- EXPECT_EQ(NewI->getNextNode(), Ret);
- }
- {
- // Check create() InsertAtEnd.
- auto *NewI = cast<sandboxir::BitCastInst>(
- sandboxir::BitCastInst::create(Arg, Tfloat,
- /*InsertAtEnd=*/BB, Ctx, "BitCast"));
- // Check getOpcode().
- EXPECT_EQ(NewI->getOpcode(), sandboxir::Instruction::Opcode::BitCast);
- // Check getSrcTy().
- EXPECT_EQ(NewI->getSrcTy(), Arg->getType());
- // Check getDestTy().
- EXPECT_EQ(NewI->getDestTy(), Tfloat);
- // Check instr position.
- EXPECT_EQ(NewI->getNextNode(), nullptr);
- EXPECT_EQ(NewI->getParent(), BB);
- }
+)IR",
+ /*SrcTy=*/Type::getInt32Ty(C), /*DstTy=*/Type::getFloatTy(C));
}
TEST_F(SandboxIRTest, AddrSpaceCastInst) {
- parseIR(C, R"IR(
+ Type *Tptr0 = PointerType::get(C, 0);
+ Type *Tptr1 = PointerType::get(C, 1);
+ testCastInst<sandboxir::AddrSpaceCastInst,
+ sandboxir::Instruction::Opcode::AddrSpaceCast>(R"IR(
define void @foo(ptr %ptr) {
%addrspacecast = addrspacecast ptr %ptr to ptr addrspace(1)
ret void
}
-)IR");
+)IR",
+ /*SrcTy=*/Tptr0,
+ /*DstTy=*/Tptr1);
Function &LLVMF = *M->getFunction("foo");
sandboxir::Context Ctx(C);
sandboxir::Function *F = Ctx.createFunction(&LLVMF);
@@ -2048,67 +1778,14 @@ define void @foo(ptr %ptr) {
auto *Arg = F->getArg(ArgIdx++);
auto *BB = &*F->begin();
auto It = BB->begin();
- Type *Tptr = Type::getInt32Ty(C)->getPointerTo();
- Type *Tptr1 = Tptr->getPointerTo(1);
auto *AddrSpaceCast = cast<sandboxir::AddrSpaceCastInst>(&*It++);
EXPECT_EQ(AddrSpaceCast->getOpcode(),
sandboxir::Instruction::Opcode::AddrSpaceCast);
- EXPECT_EQ(AddrSpaceCast->getSrcTy(), Tptr);
- EXPECT_EQ(AddrSpaceCast->getDestTy(), Tptr1);
EXPECT_EQ(AddrSpaceCast->getPointerOperand(), Arg);
EXPECT_EQ(sandboxir::AddrSpaceCastInst::getPointerOperandIndex(), 0u);
EXPECT_EQ(AddrSpaceCast->getSrcAddressSpace(),
- cast<PointerType>(Tptr)->getPointerAddressSpace());
+ cast<PointerType>(Tptr0)->getPointerAddressSpace());
EXPECT_EQ(AddrSpaceCast->getDestAddressSpace(),
cast<PointerType>(Tptr1)->getPointerAddressSpace());
- auto *Ret = cast<sandboxir::ReturnInst>(&*It++);
-
- {
- // Check create() WhereIt, WhereBB
- auto *NewI =
- cast<sandboxir::AddrSpaceCastInst>(sandboxir::AddrSpaceCastInst::create(
- Arg, Tptr1, /*WhereIt=*/BB->end(),
- /*WhereBB=*/BB, Ctx, "AddrSpaceCast"));
- // Check getOpcode().
- EXPECT_EQ(NewI->getOpcode(), sandboxir::Instruction::Opcode::AddrSpaceCast);
- // Check getSrcTy().
- EXPECT_EQ(NewI->getSrcTy(), Arg->getType());
- // Check getDestTy().
- EXPECT_EQ(NewI->getDestTy(), Tptr1);
- // Check instr position.
- EXPECT_EQ(NewI->getNextNode(), nullptr);
- EXPECT_EQ(NewI->getPrevNode(), Ret);
- }
- {
- // Check create() InsertBefore.
- auto *NewI =
- cast<sandboxir::AddrSpaceCastInst>(sandboxir::AddrSpaceCastInst::create(
- Arg, Tptr1,
- /*InsertBefore=*/Ret, Ctx, "AddrSpaceCast"));
- // Check getOpcode().
- EXPECT_EQ(NewI->getOpcode(), sandboxir::Instruction::Opcode::AddrSpaceCast);
- // Check getSrcTy().
- EXPECT_EQ(NewI->getSrcTy(), Arg->getType());
- // Check getDestTy().
- EXPECT_EQ(NewI->getDestTy(), Tptr1);
- // Check instr position.
- EXPECT_EQ(NewI->getNextNode(), Ret);
- }
- {
- // Check create() InsertAtEnd.
- auto *NewI =
- cast<sandboxir::AddrSpaceCastInst>(sandboxir::AddrSpaceCastInst::create(
- Arg, Tptr1,
- /*InsertAtEnd=*/BB, Ctx, "AddrSpaceCast"));
- // Check getOpcode().
- EXPECT_EQ(NewI->getOpcode(), sandboxir::Instruction::Opcode::AddrSpaceCast);
- // Check getSrcTy().
- EXPECT_EQ(NewI->getSrcTy(), Arg->getType());
- // Check getDestTy().
- EXPECT_EQ(NewI->getDestTy(), Tptr1);
- // Check instr position.
- EXPECT_EQ(NewI->getNextNode(), nullptr);
- EXPECT_EQ(NewI->getParent(), BB);
- }
}
More information about the llvm-commits
mailing list