From 1aca31cee242a4dcfb3300c474de8978c13905d3 Mon Sep 17 00:00:00 2001 From: Dominik Adamski Date: Tue, 23 Jan 2024 05:35:20 -0600 Subject: [PATCH] [OpenMPIRBuilderTest] Fix trip count test Update trip count check in LLVM unit test. --- llvm/unittests/Frontend/OpenMPIRBuilderTest.cpp | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/llvm/unittests/Frontend/OpenMPIRBuilderTest.cpp b/llvm/unittests/Frontend/OpenMPIRBuilderTest.cpp index 9f00d1bdaff0f..000c093ceaeed 100644 --- a/llvm/unittests/Frontend/OpenMPIRBuilderTest.cpp +++ b/llvm/unittests/Frontend/OpenMPIRBuilderTest.cpp @@ -2299,8 +2299,14 @@ TEST_F(OpenMPIRBuilderTest, StaticWorkshareLoopTarget) { // Check that no variables except for loop counter are used in loop body EXPECT_EQ(Constant::getNullValue(Builder.getPtrTy()), WorkshareLoopRuntimeCall->getArgOperand(2)); - // Check loop trip count argument - EXPECT_EQ(TripCount, WorkshareLoopRuntimeCall->getArgOperand(3)); + // Check loop trip count argument. + ConstantInt *TripCountConstInt = dyn_cast(TripCount); + EXPECT_NE(TripCountConstInt, nullptr); + ConstantInt *WorkshareLoopRuntimeCallTripCount = + dyn_cast(WorkshareLoopRuntimeCall->getArgOperand(3)); + EXPECT_NE(WorkshareLoopRuntimeCallTripCount, nullptr); + EXPECT_EQ(WorkshareLoopRuntimeCallTripCount->getSExtValue() + 1, + TripCountConstInt->getSExtValue()); } TEST_F(OpenMPIRBuilderTest, StaticWorkShareLoop) {