[llvm] [SandboxIR] Add the ExtractElementInst class (PR #102706)
via llvm-commits
llvm-commits at lists.llvm.org
Fri Aug 9 19:13:01 PDT 2024
================
@@ -631,6 +631,50 @@ define void @foo(i1 %c0, i8 %v0, i8 %v1, i1 %c1) {
}
}
+TEST_F(SandboxIRTest, ExtractElementInst) {
+ parseIR(C, R"IR(
+define void @foo(<2 x i8> %vec, i32 %idx) {
+ %ins0 = extractelement <2 x i8> %vec, i32 %idx
+ ret void
+}
+)IR");
+ Function &LLVMF = *M->getFunction("foo");
+ sandboxir::Context Ctx(C);
+ auto &F = *Ctx.createFunction(&LLVMF);
+ auto *ArgVec = F.getArg(0);
+ auto *ArgIdx = F.getArg(1);
+ auto *BB = &*F.begin();
+ auto It = BB->begin();
+ auto *EI = cast<sandboxir::ExtractElementInst>(&*It++);
+ auto *Ret = &*It++;
+
+ EXPECT_EQ(EI->getOpcode(), sandboxir::Instruction::Opcode::ExtractElement);
+ EXPECT_EQ(EI->getOperand(0), ArgVec);
+ EXPECT_EQ(EI->getOperand(1), ArgIdx);
+ EXPECT_EQ(EI->getVectorOperand(), ArgVec);
+ EXPECT_EQ(EI->getIndexOperand(), ArgIdx);
+ EXPECT_EQ(EI->getVectorOperandType(), ArgVec->getType());
+
+ auto *NewI1 =
+ cast<sandboxir::ExtractElementInst>(sandboxir::ExtractElementInst::create(
+ ArgVec, ArgIdx, Ret, Ctx, "NewInsBeforeRet"));
----------------
vporpo wrote:
nit: "NewInsBeforeRet" -> "NewExtrBeforeRet"
https://github.com/llvm/llvm-project/pull/102706
More information about the llvm-commits
mailing list