diff options
Diffstat (limited to 'llvm/unittests/Frontend/OpenMPIRBuilderTest.cpp')
| -rw-r--r-- | llvm/unittests/Frontend/OpenMPIRBuilderTest.cpp | 30 |
1 files changed, 9 insertions, 21 deletions
diff --git a/llvm/unittests/Frontend/OpenMPIRBuilderTest.cpp b/llvm/unittests/Frontend/OpenMPIRBuilderTest.cpp index c2da3f3..962dcc2 100644 --- a/llvm/unittests/Frontend/OpenMPIRBuilderTest.cpp +++ b/llvm/unittests/Frontend/OpenMPIRBuilderTest.cpp @@ -1721,7 +1721,7 @@ TEST_P(OpenMPIRBuilderTestWithParams, DynamicWorkShareLoop) { omp::OMPScheduleType SchedType = GetParam(); uint32_t ChunkSize = 1; - switch (SchedType & ~omp::OMPScheduleType::ModifierMask) { + switch (SchedType) { case omp::OMPScheduleType::DynamicChunked: case omp::OMPScheduleType::GuidedChunked: ChunkSize = 7; @@ -1794,9 +1794,8 @@ TEST_P(OpenMPIRBuilderTestWithParams, DynamicWorkShareLoop) { EXPECT_EQ(InitCall->getCalledFunction()->getName(), "__kmpc_dispatch_init_4u"); EXPECT_EQ(InitCall->getNumArgOperands(), 7U); - EXPECT_EQ(InitCall->getArgOperand(6), ConstantInt::get(LCTy, ChunkSize)); - ConstantInt *SchedVal = cast<ConstantInt>(InitCall->getArgOperand(2)); - EXPECT_EQ(SchedVal->getValue(), static_cast<uint64_t>(SchedType)); + EXPECT_EQ(InitCall->getArgOperand(6), + ConstantInt::get(Type::getInt32Ty(Ctx), ChunkSize)); ConstantInt *OrigLowerBound = dyn_cast<ConstantInt>(LowerBoundStore->getValueOperand()); @@ -1828,23 +1827,12 @@ TEST_P(OpenMPIRBuilderTestWithParams, DynamicWorkShareLoop) { EXPECT_FALSE(verifyModule(*M, &errs())); } -INSTANTIATE_TEST_CASE_P( - OpenMPWSLoopSchedulingTypes, OpenMPIRBuilderTestWithParams, - ::testing::Values(omp::OMPScheduleType::DynamicChunked, - omp::OMPScheduleType::GuidedChunked, - omp::OMPScheduleType::Auto, omp::OMPScheduleType::Runtime, - omp::OMPScheduleType::DynamicChunked | - omp::OMPScheduleType::ModifierMonotonic, - omp::OMPScheduleType::DynamicChunked | - omp::OMPScheduleType::ModifierNonmonotonic, - omp::OMPScheduleType::GuidedChunked | - omp::OMPScheduleType::ModifierMonotonic, - omp::OMPScheduleType::GuidedChunked | - omp::OMPScheduleType::ModifierNonmonotonic, - omp::OMPScheduleType::Auto | - omp::OMPScheduleType::ModifierMonotonic, - omp::OMPScheduleType::Runtime | - omp::OMPScheduleType::ModifierMonotonic)); +INSTANTIATE_TEST_SUITE_P(OpenMPWSLoopSchedulingTypes, + OpenMPIRBuilderTestWithParams, + ::testing::Values(omp::OMPScheduleType::DynamicChunked, + omp::OMPScheduleType::GuidedChunked, + omp::OMPScheduleType::Auto, + omp::OMPScheduleType::Runtime)); TEST_F(OpenMPIRBuilderTest, MasterDirective) { using InsertPointTy = OpenMPIRBuilder::InsertPointTy; |