[clang] [llvm] [OpenMP][OMPIRBuilder] Handle non-failing calls properly (PR #115863)
Sergio Afonso via cfe-commits
cfe-commits at lists.llvm.org
Fri Nov 15 04:55:34 PST 2024
https://github.com/skatrak updated https://github.com/llvm/llvm-project/pull/115863
>From 4753a00ac8450af74805c42e6580a9ebf8ee6671 Mon Sep 17 00:00:00 2001
From: Sergio Afonso <safonsof at amd.com>
Date: Tue, 12 Nov 2024 12:41:15 +0000
Subject: [PATCH 1/3] [OpenMP][OMPIRBuilder] Handle non-failing calls properly
The macro used to enable asserts and the one that `llvm::Error` and
`llvm::Expected` use to ensure all created instances are checked are not the
same. By making these checks inside of an `assert` in cases where errors are
not expected, certain build configurations would trigger runtime failures.
The `llvm::cantFail()` function, which was intended for this use case, is used
by this patch in place of `assert` to prevent these runtime failures.
---
clang/lib/CodeGen/CGOpenMPRuntime.cpp | 32 +-
clang/lib/CodeGen/CGOpenMPRuntimeGPU.cpp | 9 +-
clang/lib/CodeGen/CGStmtOpenMP.cpp | 88 ++-
llvm/lib/Frontend/OpenMP/OMPIRBuilder.cpp | 32 +-
llvm/lib/Transforms/IPO/OpenMPOpt.cpp | 32 +-
.../Frontend/OpenMPIRBuilderTest.cpp | 635 ++++++++----------
6 files changed, 353 insertions(+), 475 deletions(-)
diff --git a/clang/lib/CodeGen/CGOpenMPRuntime.cpp b/clang/lib/CodeGen/CGOpenMPRuntime.cpp
index d714af035d21a2..9c22103a1a75c6 100644
--- a/clang/lib/CodeGen/CGOpenMPRuntime.cpp
+++ b/clang/lib/CodeGen/CGOpenMPRuntime.cpp
@@ -2332,11 +2332,10 @@ void CGOpenMPRuntime::emitBarrierCall(CodeGenFunction &CGF, SourceLocation Loc,
auto *OMPRegionInfo =
dyn_cast_or_null<CGOpenMPRegionInfo>(CGF.CapturedStmtInfo);
if (CGF.CGM.getLangOpts().OpenMPIRBuilder) {
- llvm::OpenMPIRBuilder::InsertPointOrErrorTy AfterIP =
- OMPBuilder.createBarrier(CGF.Builder, Kind, ForceSimpleCall,
- EmitChecks);
- assert(AfterIP && "unexpected error creating barrier");
- CGF.Builder.restoreIP(*AfterIP);
+ llvm::OpenMPIRBuilder::InsertPointTy AfterIP =
+ cantFail(OMPBuilder.createBarrier(CGF.Builder, Kind, ForceSimpleCall,
+ EmitChecks));
+ CGF.Builder.restoreIP(AfterIP);
return;
}
@@ -5932,10 +5931,9 @@ void CGOpenMPRuntime::emitTargetOutlinedFunctionHelper(
return CGF.GenerateOpenMPCapturedStmtFunction(CS, D.getBeginLoc());
};
- llvm::Error Err = OMPBuilder.emitTargetRegionFunction(
+ cantFail(OMPBuilder.emitTargetRegionFunction(
EntryInfo, GenerateOutlinedFunction, IsOffloadEntry, OutlinedFn,
- OutlinedFnID);
- assert(!Err && "unexpected error creating target region");
+ OutlinedFnID));
if (!OutlinedFn)
return;
@@ -9676,12 +9674,11 @@ static void emitTargetCallKernelLaunch(
NumTargetItems, RTArgs, NumIterations, NumTeams, NumThreads,
DynCGGroupMem, HasNoWait);
- llvm::OpenMPIRBuilder::InsertPointOrErrorTy AfterIP =
- OMPRuntime->getOMPBuilder().emitKernelLaunch(
+ llvm::OpenMPIRBuilder::InsertPointTy AfterIP =
+ cantFail(OMPRuntime->getOMPBuilder().emitKernelLaunch(
CGF.Builder, OutlinedFnID, EmitTargetCallFallbackCB, Args, DeviceID,
- RTLoc, AllocaIP);
- assert(AfterIP && "unexpected error creating kernel launch");
- CGF.Builder.restoreIP(*AfterIP);
+ RTLoc, AllocaIP));
+ CGF.Builder.restoreIP(AfterIP);
};
if (RequiresOuterTask)
@@ -10358,12 +10355,11 @@ void CGOpenMPRuntime::emitTargetDataCalls(
InsertPointTy CodeGenIP(CGF.Builder.GetInsertBlock(),
CGF.Builder.GetInsertPoint());
llvm::OpenMPIRBuilder::LocationDescription OmpLoc(CodeGenIP);
- llvm::OpenMPIRBuilder::InsertPointOrErrorTy AfterIP =
- OMPBuilder.createTargetData(
+ llvm::OpenMPIRBuilder::InsertPointTy AfterIP =
+ cantFail(OMPBuilder.createTargetData(
OmpLoc, AllocaIP, CodeGenIP, DeviceID, IfCondVal, Info, GenMapInfoCB,
- /*MapperFunc=*/nullptr, BodyCB, DeviceAddrCB, CustomMapperCB, RTLoc);
- assert(AfterIP && "unexpected error creating target data");
- CGF.Builder.restoreIP(*AfterIP);
+ /*MapperFunc=*/nullptr, BodyCB, DeviceAddrCB, CustomMapperCB, RTLoc));
+ CGF.Builder.restoreIP(AfterIP);
}
void CGOpenMPRuntime::emitTargetDataStandAloneCall(
diff --git a/clang/lib/CodeGen/CGOpenMPRuntimeGPU.cpp b/clang/lib/CodeGen/CGOpenMPRuntimeGPU.cpp
index 43dc0e62284602..6cca3c5ce2fa01 100644
--- a/clang/lib/CodeGen/CGOpenMPRuntimeGPU.cpp
+++ b/clang/lib/CodeGen/CGOpenMPRuntimeGPU.cpp
@@ -1753,14 +1753,13 @@ void CGOpenMPRuntimeGPU::emitReduction(
Idx++;
}
- llvm::OpenMPIRBuilder::InsertPointOrErrorTy AfterIP =
- OMPBuilder.createReductionsGPU(
+ llvm::OpenMPIRBuilder::InsertPointTy AfterIP =
+ cantFail(OMPBuilder.createReductionsGPU(
OmpLoc, AllocaIP, CodeGenIP, ReductionInfos, false, TeamsReduction,
DistributeReduction, llvm::OpenMPIRBuilder::ReductionGenCBKind::Clang,
CGF.getTarget().getGridValue(),
- C.getLangOpts().OpenMPCUDAReductionBufNum, RTLoc);
- assert(AfterIP && "unexpected error creating GPU reductions");
- CGF.Builder.restoreIP(*AfterIP);
+ C.getLangOpts().OpenMPCUDAReductionBufNum, RTLoc));
+ CGF.Builder.restoreIP(AfterIP);
return;
}
diff --git a/clang/lib/CodeGen/CGStmtOpenMP.cpp b/clang/lib/CodeGen/CGStmtOpenMP.cpp
index 390516fea38498..ad1453ddbaaec2 100644
--- a/clang/lib/CodeGen/CGStmtOpenMP.cpp
+++ b/clang/lib/CodeGen/CGStmtOpenMP.cpp
@@ -1839,11 +1839,10 @@ void CodeGenFunction::EmitOMPParallelDirective(const OMPParallelDirective &S) {
CodeGenFunction::CGCapturedStmtRAII CapInfoRAII(*this, &CGSI);
llvm::OpenMPIRBuilder::InsertPointTy AllocaIP(
AllocaInsertPt->getParent(), AllocaInsertPt->getIterator());
- llvm::OpenMPIRBuilder::InsertPointOrErrorTy AfterIP =
+ llvm::OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(
OMPBuilder.createParallel(Builder, AllocaIP, BodyGenCB, PrivCB, FiniCB,
- IfCond, NumThreads, ProcBind, S.hasCancel());
- assert(AfterIP && "unexpected error creating parallel");
- Builder.restoreIP(*AfterIP);
+ IfCond, NumThreads, ProcBind, S.hasCancel()));
+ Builder.restoreIP(AfterIP);
return;
}
@@ -2135,10 +2134,8 @@ void CodeGenFunction::EmitOMPCanonicalLoop(const OMPCanonicalLoop *S) {
return llvm::Error::success();
};
- llvm::Expected<llvm::CanonicalLoopInfo *> Result =
- OMPBuilder.createCanonicalLoop(Builder, BodyGen, DistVal);
- assert(Result && "unexpected error creating canonical loop");
- llvm::CanonicalLoopInfo *CL = *Result;
+ llvm::CanonicalLoopInfo *CL =
+ cantFail(OMPBuilder.createCanonicalLoop(Builder, BodyGen, DistVal));
// Finish up the loop.
Builder.restoreIP(CL->getAfterIP());
@@ -4024,13 +4021,11 @@ static void emitOMPForDirective(const OMPLoopDirective &S, CodeGenFunction &CGF,
CGM.getOpenMPRuntime().getOMPBuilder();
llvm::OpenMPIRBuilder::InsertPointTy AllocaIP(
CGF.AllocaInsertPt->getParent(), CGF.AllocaInsertPt->getIterator());
- llvm::OpenMPIRBuilder::InsertPointOrErrorTy AfterIP =
- OMPBuilder.applyWorkshareLoop(
- CGF.Builder.getCurrentDebugLocation(), CLI, AllocaIP,
- NeedsBarrier, SchedKind, ChunkSize, /*HasSimdModifier=*/false,
- /*HasMonotonicModifier=*/false, /*HasNonmonotonicModifier=*/false,
- /*HasOrderedClause=*/false);
- assert(AfterIP && "unexpected error creating workshare loop");
+ cantFail(OMPBuilder.applyWorkshareLoop(
+ CGF.Builder.getCurrentDebugLocation(), CLI, AllocaIP, NeedsBarrier,
+ SchedKind, ChunkSize, /*HasSimdModifier=*/false,
+ /*HasMonotonicModifier=*/false, /*HasNonmonotonicModifier=*/false,
+ /*HasOrderedClause=*/false));
return;
}
@@ -4311,12 +4306,11 @@ void CodeGenFunction::EmitOMPSectionsDirective(const OMPSectionsDirective &S) {
CodeGenFunction::CGCapturedStmtRAII CapInfoRAII(*this, &CGSI);
llvm::OpenMPIRBuilder::InsertPointTy AllocaIP(
AllocaInsertPt->getParent(), AllocaInsertPt->getIterator());
- llvm::OpenMPIRBuilder::InsertPointOrErrorTy AfterIP =
- OMPBuilder.createSections(Builder, AllocaIP, SectionCBVector, PrivCB,
- FiniCB, S.hasCancel(),
- S.getSingleClause<OMPNowaitClause>());
- assert(AfterIP && "unexpected error creating sections");
- Builder.restoreIP(*AfterIP);
+ llvm::OpenMPIRBuilder::InsertPointTy AfterIP =
+ cantFail(OMPBuilder.createSections(
+ Builder, AllocaIP, SectionCBVector, PrivCB, FiniCB, S.hasCancel(),
+ S.getSingleClause<OMPNowaitClause>()));
+ Builder.restoreIP(AfterIP);
return;
}
{
@@ -4354,10 +4348,9 @@ void CodeGenFunction::EmitOMPSectionDirective(const OMPSectionDirective &S) {
LexicalScope Scope(*this, S.getSourceRange());
EmitStopPoint(&S);
- llvm::OpenMPIRBuilder::InsertPointOrErrorTy AfterIP =
- OMPBuilder.createSection(Builder, BodyGenCB, FiniCB);
- assert(AfterIP && "unexpected error creating section");
- Builder.restoreIP(*AfterIP);
+ llvm::OpenMPIRBuilder::InsertPointTy AfterIP =
+ cantFail(OMPBuilder.createSection(Builder, BodyGenCB, FiniCB));
+ Builder.restoreIP(AfterIP);
return;
}
@@ -4440,10 +4433,9 @@ void CodeGenFunction::EmitOMPMasterDirective(const OMPMasterDirective &S) {
LexicalScope Scope(*this, S.getSourceRange());
EmitStopPoint(&S);
- llvm::OpenMPIRBuilder::InsertPointOrErrorTy AfterIP =
- OMPBuilder.createMaster(Builder, BodyGenCB, FiniCB);
- assert(AfterIP && "unexpected error creating master");
- Builder.restoreIP(*AfterIP);
+ llvm::OpenMPIRBuilder::InsertPointTy AfterIP =
+ cantFail(OMPBuilder.createMaster(Builder, BodyGenCB, FiniCB));
+ Builder.restoreIP(AfterIP);
return;
}
@@ -4491,10 +4483,9 @@ void CodeGenFunction::EmitOMPMaskedDirective(const OMPMaskedDirective &S) {
LexicalScope Scope(*this, S.getSourceRange());
EmitStopPoint(&S);
- llvm::OpenMPIRBuilder::InsertPointOrErrorTy AfterIP =
- OMPBuilder.createMasked(Builder, BodyGenCB, FiniCB, FilterVal);
- assert(AfterIP && "unexpected error creating masked");
- Builder.restoreIP(*AfterIP);
+ llvm::OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(
+ OMPBuilder.createMasked(Builder, BodyGenCB, FiniCB, FilterVal));
+ Builder.restoreIP(AfterIP);
return;
}
@@ -4535,11 +4526,11 @@ void CodeGenFunction::EmitOMPCriticalDirective(const OMPCriticalDirective &S) {
LexicalScope Scope(*this, S.getSourceRange());
EmitStopPoint(&S);
- llvm::OpenMPIRBuilder::InsertPointOrErrorTy AfterIP =
- OMPBuilder.createCritical(Builder, BodyGenCB, FiniCB,
- S.getDirectiveName().getAsString(), HintInst);
- assert(AfterIP && "unexpected error creating critical");
- Builder.restoreIP(*AfterIP);
+ llvm::OpenMPIRBuilder::InsertPointTy AfterIP =
+ cantFail(OMPBuilder.createCritical(Builder, BodyGenCB, FiniCB,
+ S.getDirectiveName().getAsString(),
+ HintInst));
+ Builder.restoreIP(AfterIP);
return;
}
@@ -5503,10 +5494,9 @@ void CodeGenFunction::EmitOMPTaskgroupDirective(
CodeGenFunction::CGCapturedStmtInfo CapStmtInfo;
if (!CapturedStmtInfo)
CapturedStmtInfo = &CapStmtInfo;
- llvm::OpenMPIRBuilder::InsertPointOrErrorTy AfterIP =
- OMPBuilder.createTaskgroup(Builder, AllocaIP, BodyGenCB);
- assert(AfterIP && "unexpected error creating taskgroup");
- Builder.restoreIP(*AfterIP);
+ llvm::OpenMPIRBuilder::InsertPointTy AfterIP =
+ cantFail(OMPBuilder.createTaskgroup(Builder, AllocaIP, BodyGenCB));
+ Builder.restoreIP(AfterIP);
return;
}
auto &&CodeGen = [&S](CodeGenFunction &CGF, PrePostActionTy &Action) {
@@ -6109,10 +6099,9 @@ void CodeGenFunction::EmitOMPOrderedDirective(const OMPOrderedDirective &S) {
};
OMPLexicalScope Scope(*this, S, OMPD_unknown);
- llvm::OpenMPIRBuilder::InsertPointOrErrorTy AfterIP =
- OMPBuilder.createOrderedThreadsSimd(Builder, BodyGenCB, FiniCB, !C);
- assert(AfterIP && "unexpected error creating ordered");
- Builder.restoreIP(*AfterIP);
+ llvm::OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(
+ OMPBuilder.createOrderedThreadsSimd(Builder, BodyGenCB, FiniCB, !C));
+ Builder.restoreIP(AfterIP);
}
return;
}
@@ -7388,10 +7377,9 @@ void CodeGenFunction::EmitOMPCancelDirective(const OMPCancelDirective &S) {
if (IfCond)
IfCondition = EmitScalarExpr(IfCond,
/*IgnoreResultAssign=*/true);
- llvm::OpenMPIRBuilder::InsertPointOrErrorTy AfterIP =
- OMPBuilder.createCancel(Builder, IfCondition, S.getCancelRegion());
- assert(AfterIP && "unexpected error creating cancel");
- return Builder.restoreIP(*AfterIP);
+ llvm::OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(
+ OMPBuilder.createCancel(Builder, IfCondition, S.getCancelRegion()));
+ return Builder.restoreIP(AfterIP);
}
}
diff --git a/llvm/lib/Frontend/OpenMP/OMPIRBuilder.cpp b/llvm/lib/Frontend/OpenMP/OMPIRBuilder.cpp
index d2e4dc1c85dfd2..5d9a7784d9e68c 100644
--- a/llvm/lib/Frontend/OpenMP/OMPIRBuilder.cpp
+++ b/llvm/lib/Frontend/OpenMP/OMPIRBuilder.cpp
@@ -4225,7 +4225,11 @@ OpenMPIRBuilder::applyStaticChunkedWorkshareLoop(DebugLoc DL,
// Create outer "dispatch" loop for enumerating the chunks.
BasicBlock *DispatchEnter = splitBB(Builder, true);
Value *DispatchCounter;
- Expected<CanonicalLoopInfo *> LoopResult = createCanonicalLoop(
+
+ // It is safe to assume this didn't return an error because the callback
+ // passed into createCanonicalLoop is the only possible error source, and it
+ // always returns success.
+ CanonicalLoopInfo *DispatchCLI = cantFail(createCanonicalLoop(
{Builder.saveIP(), DL},
[&](InsertPointTy BodyIP, Value *Counter) {
DispatchCounter = Counter;
@@ -4233,15 +4237,7 @@ OpenMPIRBuilder::applyStaticChunkedWorkshareLoop(DebugLoc DL,
},
FirstChunkStart, CastedTripCount, NextChunkStride,
/*IsSigned=*/false, /*InclusiveStop=*/false, /*ComputeIP=*/{},
- "dispatch");
- if (!LoopResult) {
- // It is safe to assume this didn't return an error because the callback
- // passed into createCanonicalLoop is the only possible error source, and it
- // always returns success. Need to still cast the result into bool to avoid
- // runtime errors.
- llvm_unreachable("unexpected error creating canonical loop");
- }
- CanonicalLoopInfo *DispatchCLI = *LoopResult;
+ "dispatch"));
// Remember the BasicBlocks of the dispatch loop we need, then invalidate to
// not have to preserve the canonical invariant.
@@ -6542,16 +6538,12 @@ OpenMPIRBuilder::InsertPointOrErrorTy OpenMPIRBuilder::createTargetData(
};
bool RequiresOuterTargetTask = Info.HasNoWait;
- if (!RequiresOuterTargetTask) {
- Error Err = TaskBodyCB(/*DeviceID=*/nullptr, /*RTLoc=*/nullptr,
- /*TargetTaskAllocaIP=*/{});
- assert(!Err && "TaskBodyCB expected to succeed");
- } else {
- InsertPointOrErrorTy AfterIP =
- emitTargetTask(TaskBodyCB, DeviceID, SrcLocInfo, AllocaIP,
- /*Dependencies=*/{}, Info.HasNoWait);
- assert(AfterIP && "TaskBodyCB expected to succeed");
- }
+ if (!RequiresOuterTargetTask)
+ cantFail(TaskBodyCB(/*DeviceID=*/nullptr, /*RTLoc=*/nullptr,
+ /*TargetTaskAllocaIP=*/{}));
+ else
+ cantFail(emitTargetTask(TaskBodyCB, DeviceID, SrcLocInfo, AllocaIP,
+ /*Dependencies=*/{}, Info.HasNoWait));
} else {
Function *BeginMapperFunc = getOrCreateRuntimeFunctionPtr(
omp::OMPRTL___tgt_target_data_begin_mapper);
diff --git a/llvm/lib/Transforms/IPO/OpenMPOpt.cpp b/llvm/lib/Transforms/IPO/OpenMPOpt.cpp
index 9e25620710fc84..153f5c7510efae 100644
--- a/llvm/lib/Transforms/IPO/OpenMPOpt.cpp
+++ b/llvm/lib/Transforms/IPO/OpenMPOpt.cpp
@@ -1178,15 +1178,12 @@ struct OpenMPOpt {
OpenMPIRBuilder::LocationDescription Loc(
InsertPointTy(ParentBB, ParentBB->end()), DL);
- OpenMPIRBuilder::InsertPointOrErrorTy SeqAfterIP =
- OMPInfoCache.OMPBuilder.createMaster(Loc, BodyGenCB, FiniCB);
- assert(SeqAfterIP && "Unexpected error creating master");
+ OpenMPIRBuilder::InsertPointTy SeqAfterIP = cantFail(
+ OMPInfoCache.OMPBuilder.createMaster(Loc, BodyGenCB, FiniCB));
+ cantFail(
+ OMPInfoCache.OMPBuilder.createBarrier(SeqAfterIP, OMPD_parallel));
- OpenMPIRBuilder::InsertPointOrErrorTy BarrierAfterIP =
- OMPInfoCache.OMPBuilder.createBarrier(*SeqAfterIP, OMPD_parallel);
- assert(BarrierAfterIP && "Unexpected error creating barrier");
-
- BranchInst::Create(SeqAfterBB, SeqAfterIP->getBlock());
+ BranchInst::Create(SeqAfterBB, SeqAfterIP.getBlock());
LLVM_DEBUG(dbgs() << TAG << "After sequential inlining " << *OuterFn
<< "\n");
@@ -1256,12 +1253,11 @@ struct OpenMPOpt {
OriginalFn->getEntryBlock().getFirstInsertionPt());
// Create the merged parallel region with default proc binding, to
// avoid overriding binding settings, and without explicit cancellation.
- OpenMPIRBuilder::InsertPointOrErrorTy AfterIP =
- OMPInfoCache.OMPBuilder.createParallel(
+ OpenMPIRBuilder::InsertPointTy AfterIP =
+ cantFail(OMPInfoCache.OMPBuilder.createParallel(
Loc, AllocaIP, BodyGenCB, PrivCB, FiniCB, nullptr, nullptr,
- OMP_PROC_BIND_default, /* IsCancellable */ false);
- assert(AfterIP && "Unexpected error creating parallel");
- BranchInst::Create(AfterBB, AfterIP->getBlock());
+ OMP_PROC_BIND_default, /* IsCancellable */ false));
+ BranchInst::Create(AfterBB, AfterIP.getBlock());
// Perform the actual outlining.
OMPInfoCache.OMPBuilder.finalize(OriginalFn);
@@ -1297,12 +1293,10 @@ struct OpenMPOpt {
if (CI != MergableCIs.back()) {
// TODO: Remove barrier if the merged parallel region includes the
// 'nowait' clause.
- OpenMPIRBuilder::InsertPointOrErrorTy AfterIP =
- OMPInfoCache.OMPBuilder.createBarrier(
- InsertPointTy(NewCI->getParent(),
- NewCI->getNextNode()->getIterator()),
- OMPD_parallel);
- assert(AfterIP && "Unexpected error creating barrier");
+ cantFail(OMPInfoCache.OMPBuilder.createBarrier(
+ InsertPointTy(NewCI->getParent(),
+ NewCI->getNextNode()->getIterator()),
+ OMPD_parallel));
}
CI->eraseFromParent();
diff --git a/llvm/unittests/Frontend/OpenMPIRBuilderTest.cpp b/llvm/unittests/Frontend/OpenMPIRBuilderTest.cpp
index 630cd03c688012..dfdb5d69501ed4 100644
--- a/llvm/unittests/Frontend/OpenMPIRBuilderTest.cpp
+++ b/llvm/unittests/Frontend/OpenMPIRBuilderTest.cpp
@@ -235,10 +235,9 @@ class OpenMPIRBuilderTest : public testing::Test {
return Error::success();
};
- Expected<CanonicalLoopInfo *> LoopResult =
- OMPBuilder.createCanonicalLoop(Loc, LoopBodyGenCB, CastedTripCount);
- assert(LoopResult && "unexpected error");
- CanonicalLoopInfo *Loop = *LoopResult;
+
+ CanonicalLoopInfo *Loop = cantFail(
+ OMPBuilder.createCanonicalLoop(Loc, LoopBodyGenCB, CastedTripCount));
// Finalize the function.
Builder.restoreIP(Loop->getAfterIP());
@@ -345,18 +344,14 @@ TEST_F(OpenMPIRBuilderTest, CreateBarrier) {
IRBuilder<> Builder(BB);
- OpenMPIRBuilder::InsertPointOrErrorTy BarrierIP1 =
- OMPBuilder.createBarrier({IRBuilder<>::InsertPoint()}, OMPD_for);
- assert(BarrierIP1 && "unexpected error");
+ cantFail(OMPBuilder.createBarrier({IRBuilder<>::InsertPoint()}, OMPD_for));
EXPECT_TRUE(M->global_empty());
EXPECT_EQ(M->size(), 1U);
EXPECT_EQ(F->size(), 1U);
EXPECT_EQ(BB->size(), 0U);
OpenMPIRBuilder::LocationDescription Loc({Builder.saveIP()});
- OpenMPIRBuilder::InsertPointOrErrorTy BarrierIP2 =
- OMPBuilder.createBarrier(Loc, OMPD_for);
- assert(BarrierIP2 && "unexpected error");
+ cantFail(OMPBuilder.createBarrier(Loc, OMPD_for));
EXPECT_FALSE(M->global_empty());
EXPECT_EQ(M->size(), 3U);
EXPECT_EQ(F->size(), 1U);
@@ -399,10 +394,9 @@ TEST_F(OpenMPIRBuilderTest, CreateCancel) {
IRBuilder<> Builder(BB);
OpenMPIRBuilder::LocationDescription Loc({Builder.saveIP()});
- OpenMPIRBuilder::InsertPointOrErrorTy NewIP =
- OMPBuilder.createCancel(Loc, nullptr, OMPD_parallel);
- assert(NewIP && "unexpected error");
- Builder.restoreIP(*NewIP);
+ OpenMPIRBuilder::InsertPointTy NewIP =
+ cantFail(OMPBuilder.createCancel(Loc, nullptr, OMPD_parallel));
+ Builder.restoreIP(NewIP);
EXPECT_FALSE(M->global_empty());
EXPECT_EQ(M->size(), 4U);
EXPECT_EQ(F->size(), 4U);
@@ -424,7 +418,7 @@ TEST_F(OpenMPIRBuilderTest, CreateCancel) {
EXPECT_EQ(Cancel->getNumUses(), 1U);
Instruction *CancelBBTI = Cancel->getParent()->getTerminator();
EXPECT_EQ(CancelBBTI->getNumSuccessors(), 2U);
- EXPECT_EQ(CancelBBTI->getSuccessor(0), NewIP->getBlock());
+ EXPECT_EQ(CancelBBTI->getSuccessor(0), NewIP.getBlock());
EXPECT_EQ(CancelBBTI->getSuccessor(1)->size(), 3U);
CallInst *GTID1 = dyn_cast<CallInst>(&CancelBBTI->getSuccessor(1)->front());
EXPECT_NE(GTID1, nullptr);
@@ -468,10 +462,9 @@ TEST_F(OpenMPIRBuilderTest, CreateCancelIfCond) {
IRBuilder<> Builder(BB);
OpenMPIRBuilder::LocationDescription Loc({Builder.saveIP()});
- OpenMPIRBuilder::InsertPointOrErrorTy NewIP =
- OMPBuilder.createCancel(Loc, Builder.getTrue(), OMPD_parallel);
- assert(NewIP && "unexpected error");
- Builder.restoreIP(*NewIP);
+ OpenMPIRBuilder::InsertPointTy NewIP =
+ cantFail(OMPBuilder.createCancel(Loc, Builder.getTrue(), OMPD_parallel));
+ Builder.restoreIP(NewIP);
EXPECT_FALSE(M->global_empty());
EXPECT_EQ(M->size(), 4U);
EXPECT_EQ(F->size(), 7U);
@@ -499,7 +492,7 @@ TEST_F(OpenMPIRBuilderTest, CreateCancelIfCond) {
EXPECT_EQ(CancelBBTI->getNumSuccessors(), 2U);
EXPECT_EQ(CancelBBTI->getSuccessor(0)->size(), 1U);
EXPECT_EQ(CancelBBTI->getSuccessor(0)->getUniqueSuccessor(),
- NewIP->getBlock());
+ NewIP.getBlock());
EXPECT_EQ(CancelBBTI->getSuccessor(1)->size(), 3U);
CallInst *GTID1 = dyn_cast<CallInst>(&CancelBBTI->getSuccessor(1)->front());
EXPECT_NE(GTID1, nullptr);
@@ -543,10 +536,9 @@ TEST_F(OpenMPIRBuilderTest, CreateCancelBarrier) {
IRBuilder<> Builder(BB);
OpenMPIRBuilder::LocationDescription Loc({Builder.saveIP()});
- OpenMPIRBuilder::InsertPointOrErrorTy NewIP =
- OMPBuilder.createBarrier(Loc, OMPD_for);
- assert(NewIP && "unexpected error");
- Builder.restoreIP(*NewIP);
+ OpenMPIRBuilder::InsertPointTy NewIP =
+ cantFail(OMPBuilder.createBarrier(Loc, OMPD_for));
+ Builder.restoreIP(NewIP);
EXPECT_FALSE(M->global_empty());
EXPECT_EQ(M->size(), 3U);
EXPECT_EQ(F->size(), 4U);
@@ -568,7 +560,7 @@ TEST_F(OpenMPIRBuilderTest, CreateCancelBarrier) {
EXPECT_EQ(Barrier->getNumUses(), 1U);
Instruction *BarrierBBTI = Barrier->getParent()->getTerminator();
EXPECT_EQ(BarrierBBTI->getNumSuccessors(), 2U);
- EXPECT_EQ(BarrierBBTI->getSuccessor(0), NewIP->getBlock());
+ EXPECT_EQ(BarrierBBTI->getSuccessor(0), NewIP.getBlock());
EXPECT_EQ(BarrierBBTI->getSuccessor(1)->size(), 1U);
EXPECT_EQ(BarrierBBTI->getSuccessor(1)->getTerminator()->getNumSuccessors(),
1U);
@@ -591,9 +583,7 @@ TEST_F(OpenMPIRBuilderTest, DbgLoc) {
IRBuilder<> Builder(BB);
OpenMPIRBuilder::LocationDescription Loc({Builder.saveIP(), DL});
- OpenMPIRBuilder::InsertPointOrErrorTy AfterIP =
- OMPBuilder.createBarrier(Loc, OMPD_for);
- assert(AfterIP && "unexpected error");
+ cantFail(OMPBuilder.createBarrier(Loc, OMPD_for));
CallInst *GTID = dyn_cast<CallInst>(&BB->front());
CallInst *Barrier = dyn_cast<CallInst>(GTID->getNextNode());
EXPECT_EQ(GTID->getDebugLoc(), DL);
@@ -692,16 +682,15 @@ TEST_F(OpenMPIRBuilderTest, ParallelSimpleGPU) {
IRBuilder<>::InsertPoint AllocaIP(&F->getEntryBlock(),
F->getEntryBlock().getFirstInsertionPt());
- OpenMPIRBuilder::InsertPointOrErrorTy AfterIP =
- OMPBuilder.createParallel(Loc, AllocaIP, BodyGenCB, PrivCB, FiniCB,
- nullptr, nullptr, OMP_PROC_BIND_default, false);
- assert(AfterIP && "unexpected error");
+ OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(OMPBuilder.createParallel(
+ Loc, AllocaIP, BodyGenCB, PrivCB, FiniCB, nullptr, nullptr,
+ OMP_PROC_BIND_default, false));
EXPECT_EQ(NumBodiesGenerated, 1U);
EXPECT_EQ(NumPrivatizedVars, 1U);
EXPECT_EQ(NumFinalizationPoints, 1U);
- Builder.restoreIP(*AfterIP);
+ Builder.restoreIP(AfterIP);
Builder.CreateRetVoid();
OMPBuilder.finalize();
@@ -805,15 +794,14 @@ TEST_F(OpenMPIRBuilderTest, ParallelSimple) {
IRBuilder<>::InsertPoint AllocaIP(&F->getEntryBlock(),
F->getEntryBlock().getFirstInsertionPt());
- OpenMPIRBuilder::InsertPointOrErrorTy AfterIP =
- OMPBuilder.createParallel(Loc, AllocaIP, BodyGenCB, PrivCB, FiniCB,
- nullptr, nullptr, OMP_PROC_BIND_default, false);
- assert(AfterIP && "unexpected error");
+ OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(OMPBuilder.createParallel(
+ Loc, AllocaIP, BodyGenCB, PrivCB, FiniCB, nullptr, nullptr,
+ OMP_PROC_BIND_default, false));
EXPECT_EQ(NumBodiesGenerated, 1U);
EXPECT_EQ(NumPrivatizedVars, 1U);
EXPECT_EQ(NumFinalizationPoints, 1U);
- Builder.restoreIP(*AfterIP);
+ Builder.restoreIP(AfterIP);
Builder.CreateRetVoid();
OMPBuilder.finalize();
@@ -894,28 +882,26 @@ TEST_F(OpenMPIRBuilderTest, ParallelNested) {
BasicBlock *NewBB = SplitBlock(CGBB, &*CodeGenIP.getPoint());
CGBB->getTerminator()->eraseFromParent();
- OpenMPIRBuilder::InsertPointOrErrorTy AfterIP = OMPBuilder.createParallel(
+ OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(OMPBuilder.createParallel(
InsertPointTy(CGBB, CGBB->end()), AllocaIP, InnerBodyGenCB, PrivCB,
- FiniCB, nullptr, nullptr, OMP_PROC_BIND_default, false);
- assert(AfterIP && "unexpected error");
+ FiniCB, nullptr, nullptr, OMP_PROC_BIND_default, false));
- Builder.restoreIP(*AfterIP);
+ Builder.restoreIP(AfterIP);
Builder.CreateBr(NewBB);
return Error::success();
};
IRBuilder<>::InsertPoint AllocaIP(&F->getEntryBlock(),
F->getEntryBlock().getFirstInsertionPt());
- OpenMPIRBuilder::InsertPointOrErrorTy AfterIP =
- OMPBuilder.createParallel(Loc, AllocaIP, OuterBodyGenCB, PrivCB, FiniCB,
- nullptr, nullptr, OMP_PROC_BIND_default, false);
- assert(AfterIP && "unexpected error");
+ OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(OMPBuilder.createParallel(
+ Loc, AllocaIP, OuterBodyGenCB, PrivCB, FiniCB, nullptr, nullptr,
+ OMP_PROC_BIND_default, false));
EXPECT_EQ(NumInnerBodiesGenerated, 1U);
EXPECT_EQ(NumOuterBodiesGenerated, 1U);
EXPECT_EQ(NumFinalizationPoints, 2U);
- Builder.restoreIP(*AfterIP);
+ Builder.restoreIP(AfterIP);
Builder.CreateRetVoid();
OMPBuilder.finalize();
@@ -998,36 +984,35 @@ TEST_F(OpenMPIRBuilderTest, ParallelNested2Inner) {
NewBB1->getTerminator()->eraseFromParent();
;
- OpenMPIRBuilder::InsertPointOrErrorTy AfterIP1 = OMPBuilder.createParallel(
- InsertPointTy(CGBB, CGBB->end()), AllocaIP, InnerBodyGenCB, PrivCB,
- FiniCB, nullptr, nullptr, OMP_PROC_BIND_default, false);
- assert(AfterIP1 && "unexpected error");
+ OpenMPIRBuilder::InsertPointTy AfterIP1 =
+ cantFail(OMPBuilder.createParallel(
+ InsertPointTy(CGBB, CGBB->end()), AllocaIP, InnerBodyGenCB, PrivCB,
+ FiniCB, nullptr, nullptr, OMP_PROC_BIND_default, false));
- Builder.restoreIP(*AfterIP1);
+ Builder.restoreIP(AfterIP1);
Builder.CreateBr(NewBB1);
- OpenMPIRBuilder::InsertPointOrErrorTy AfterIP2 = OMPBuilder.createParallel(
- InsertPointTy(NewBB1, NewBB1->end()), AllocaIP, InnerBodyGenCB, PrivCB,
- FiniCB, nullptr, nullptr, OMP_PROC_BIND_default, false);
- assert(AfterIP2 && "unexpected error");
+ OpenMPIRBuilder::InsertPointTy AfterIP2 =
+ cantFail(OMPBuilder.createParallel(
+ InsertPointTy(NewBB1, NewBB1->end()), AllocaIP, InnerBodyGenCB,
+ PrivCB, FiniCB, nullptr, nullptr, OMP_PROC_BIND_default, false));
- Builder.restoreIP(*AfterIP2);
+ Builder.restoreIP(AfterIP2);
Builder.CreateBr(NewBB2);
return Error::success();
};
IRBuilder<>::InsertPoint AllocaIP(&F->getEntryBlock(),
F->getEntryBlock().getFirstInsertionPt());
- OpenMPIRBuilder::InsertPointOrErrorTy AfterIP =
- OMPBuilder.createParallel(Loc, AllocaIP, OuterBodyGenCB, PrivCB, FiniCB,
- nullptr, nullptr, OMP_PROC_BIND_default, false);
- assert(AfterIP && "unexpected error");
+ OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(OMPBuilder.createParallel(
+ Loc, AllocaIP, OuterBodyGenCB, PrivCB, FiniCB, nullptr, nullptr,
+ OMP_PROC_BIND_default, false));
EXPECT_EQ(NumInnerBodiesGenerated, 2U);
EXPECT_EQ(NumOuterBodiesGenerated, 1U);
EXPECT_EQ(NumFinalizationPoints, 3U);
- Builder.restoreIP(*AfterIP);
+ Builder.restoreIP(AfterIP);
Builder.CreateRetVoid();
OMPBuilder.finalize();
@@ -1133,17 +1118,16 @@ TEST_F(OpenMPIRBuilderTest, ParallelIfCond) {
IRBuilder<>::InsertPoint AllocaIP(&F->getEntryBlock(),
F->getEntryBlock().getFirstInsertionPt());
- OpenMPIRBuilder::InsertPointOrErrorTy AfterIP =
+ OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(
OMPBuilder.createParallel(Loc, AllocaIP, BodyGenCB, PrivCB, FiniCB,
Builder.CreateIsNotNull(F->arg_begin()),
- nullptr, OMP_PROC_BIND_default, false);
- assert(AfterIP && "unexpected error");
+ nullptr, OMP_PROC_BIND_default, false));
EXPECT_EQ(NumBodiesGenerated, 1U);
EXPECT_EQ(NumPrivatizedVars, 1U);
EXPECT_EQ(NumFinalizationPoints, 1U);
- Builder.restoreIP(*AfterIP);
+ Builder.restoreIP(AfterIP);
Builder.CreateRetVoid();
OMPBuilder.finalize();
@@ -1198,10 +1182,9 @@ TEST_F(OpenMPIRBuilderTest, ParallelCancelBarrier) {
// Create three barriers, two cancel barriers but only one checked.
Function *CBFn, *BFn;
- OpenMPIRBuilder::InsertPointOrErrorTy BarrierIP1 =
- OMPBuilder.createBarrier(Builder.saveIP(), OMPD_parallel);
- assert(BarrierIP1 && "unexpected error");
- Builder.restoreIP(*BarrierIP1);
+ OpenMPIRBuilder::InsertPointTy BarrierIP1 =
+ cantFail(OMPBuilder.createBarrier(Builder.saveIP(), OMPD_parallel));
+ Builder.restoreIP(BarrierIP1);
CBFn = M->getFunction("__kmpc_cancel_barrier");
BFn = M->getFunction("__kmpc_barrier");
@@ -1212,10 +1195,9 @@ TEST_F(OpenMPIRBuilderTest, ParallelCancelBarrier) {
ASSERT_EQ(CBFn->user_back()->getNumUses(), 1U);
CheckedBarrier = cast<CallInst>(CBFn->user_back());
- OpenMPIRBuilder::InsertPointOrErrorTy BarrierIP2 =
- OMPBuilder.createBarrier(Builder.saveIP(), OMPD_parallel, true);
- assert(BarrierIP2 && "unexpected error");
- Builder.restoreIP(*BarrierIP2);
+ OpenMPIRBuilder::InsertPointTy BarrierIP2 = cantFail(
+ OMPBuilder.createBarrier(Builder.saveIP(), OMPD_parallel, true));
+ Builder.restoreIP(BarrierIP2);
CBFn = M->getFunction("__kmpc_cancel_barrier");
BFn = M->getFunction("__kmpc_barrier");
ASSERT_NE(CBFn, nullptr);
@@ -1225,10 +1207,10 @@ TEST_F(OpenMPIRBuilderTest, ParallelCancelBarrier) {
ASSERT_TRUE(isa<CallInst>(BFn->user_back()));
ASSERT_EQ(BFn->user_back()->getNumUses(), 0U);
- OpenMPIRBuilder::InsertPointOrErrorTy BarrierIP3 =
- OMPBuilder.createBarrier(Builder.saveIP(), OMPD_parallel, false, false);
- assert(BarrierIP3 && "unexpected error");
- Builder.restoreIP(*BarrierIP3);
+ OpenMPIRBuilder::InsertPointTy BarrierIP3 =
+ cantFail(OMPBuilder.createBarrier(Builder.saveIP(), OMPD_parallel,
+ false, false));
+ Builder.restoreIP(BarrierIP3);
ASSERT_EQ(CBFn->getNumUses(), 2U);
ASSERT_EQ(BFn->getNumUses(), 1U);
ASSERT_TRUE(CBFn->user_back() != CheckedBarrier);
@@ -1258,18 +1240,17 @@ TEST_F(OpenMPIRBuilderTest, ParallelCancelBarrier) {
IRBuilder<>::InsertPoint AllocaIP(&F->getEntryBlock(),
F->getEntryBlock().getFirstInsertionPt());
- OpenMPIRBuilder::InsertPointOrErrorTy AfterIP = OMPBuilder.createParallel(
+ OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(OMPBuilder.createParallel(
Loc, AllocaIP, BODYGENCB_WRAPPER(BodyGenCB), PrivCB, FiniCB,
Builder.CreateIsNotNull(F->arg_begin()), nullptr, OMP_PROC_BIND_default,
- true);
- assert(AfterIP && "unexpected error");
+ true));
EXPECT_EQ(NumBodiesGenerated, 1U);
EXPECT_EQ(NumPrivatizedVars, 0U);
EXPECT_EQ(NumFinalizationPoints, 2U);
EXPECT_EQ(FakeDestructor->getNumUses(), 2U);
- Builder.restoreIP(*AfterIP);
+ Builder.restoreIP(AfterIP);
Builder.CreateRetVoid();
OMPBuilder.finalize();
@@ -1345,11 +1326,10 @@ TEST_F(OpenMPIRBuilderTest, ParallelForwardAsPointers) {
IRBuilder<>::InsertPoint AllocaIP(&F->getEntryBlock(),
F->getEntryBlock().getFirstInsertionPt());
- OpenMPIRBuilder::InsertPointOrErrorTy AfterIP =
- OMPBuilder.createParallel(Loc, AllocaIP, BodyGenCB, PrivCB, FiniCB,
- nullptr, nullptr, OMP_PROC_BIND_default, false);
- assert(AfterIP && "unexpected error");
- Builder.restoreIP(*AfterIP);
+ OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(OMPBuilder.createParallel(
+ Loc, AllocaIP, BodyGenCB, PrivCB, FiniCB, nullptr, nullptr,
+ OMP_PROC_BIND_default, false));
+ Builder.restoreIP(AfterIP);
Builder.CreateRetVoid();
OMPBuilder.finalize();
@@ -1382,10 +1362,8 @@ TEST_F(OpenMPIRBuilderTest, CanonicalLoopSimple) {
return Error::success();
};
- Expected<CanonicalLoopInfo *> LoopResult =
- OMPBuilder.createCanonicalLoop(Loc, LoopBodyGenCB, TripCount);
- assert(LoopResult && "unexpected error");
- CanonicalLoopInfo *Loop = *LoopResult;
+ CanonicalLoopInfo *Loop =
+ cantFail(OMPBuilder.createCanonicalLoop(Loc, LoopBodyGenCB, TripCount));
Builder.restoreIP(Loop->getAfterIP());
ReturnInst *RetInst = Builder.CreateRetVoid();
@@ -1440,11 +1418,9 @@ TEST_F(OpenMPIRBuilderTest, CanonicalLoopBounds) {
auto LoopBodyGenCB = [&](InsertPointTy CodeGenIP, llvm::Value *LC) {
return Error::success();
};
- Expected<CanonicalLoopInfo *> LoopResult =
+ CanonicalLoopInfo *Loop = cantFail(
OMPBuilder.createCanonicalLoop(Loc, LoopBodyGenCB, StartVal, StopVal,
- StepVal, IsSigned, InclusiveStop);
- assert(LoopResult && "unexpected error");
- CanonicalLoopInfo *Loop = *LoopResult;
+ StepVal, IsSigned, InclusiveStop));
Loop->assertOK();
Builder.restoreIP(Loop->getAfterIP());
Value *TripCount = Loop->getTripCount();
@@ -1539,20 +1515,16 @@ TEST_F(OpenMPIRBuilderTest, CollapseNestedLoops) {
Call = createPrintfCall(Builder, "body i=%d j=%d\\n", {OuterLC, InnerLC});
return Error::success();
};
- Expected<CanonicalLoopInfo *> LoopResult = OMPBuilder.createCanonicalLoop(
- Builder.saveIP(), InnerLoopBodyGenCB, InnerTripCount, "inner");
- assert(LoopResult && "unexpected error");
- InnerLoop = *LoopResult;
+ InnerLoop = cantFail(OMPBuilder.createCanonicalLoop(
+ Builder.saveIP(), InnerLoopBodyGenCB, InnerTripCount, "inner"));
Builder.restoreIP(InnerLoop->getAfterIP());
InbetweenTrail =
createPrintfCall(Builder, "In-between trail i=%d\\n", {OuterLC});
return Error::success();
};
- Expected<CanonicalLoopInfo *> LoopResult = OMPBuilder.createCanonicalLoop(
- OuterLoc, OuterLoopBodyGenCB, OuterTripCount, "outer");
- assert(LoopResult && "unexpected error");
- CanonicalLoopInfo *OuterLoop = *LoopResult;
+ CanonicalLoopInfo *OuterLoop = cantFail(OMPBuilder.createCanonicalLoop(
+ OuterLoc, OuterLoopBodyGenCB, OuterTripCount, "outer"));
// Finish the function.
Builder.restoreIP(OuterLoop->getAfterIP());
@@ -1664,16 +1636,12 @@ TEST_F(OpenMPIRBuilderTest, TileNestedLoops) {
createPrintfCall(Builder, "i=%d j=%d\\n", {OuterLC, InnerLC});
return Error::success();
};
- Expected<CanonicalLoopInfo *> LoopResult = OMPBuilder.createCanonicalLoop(
- OuterCodeGenIP, InnerLoopBodyGenCB, TripCount, "inner");
- assert(LoopResult && "unexpected error");
- InnerLoop = *LoopResult;
+ InnerLoop = cantFail(OMPBuilder.createCanonicalLoop(
+ OuterCodeGenIP, InnerLoopBodyGenCB, TripCount, "inner"));
return Error::success();
};
- Expected<CanonicalLoopInfo *> LoopResult = OMPBuilder.createCanonicalLoop(
- Loc, OuterLoopBodyGenCB, TripCount, "outer");
- assert(LoopResult && "unexpected error");
- CanonicalLoopInfo *OuterLoop = *LoopResult;
+ CanonicalLoopInfo *OuterLoop = cantFail(OMPBuilder.createCanonicalLoop(
+ Loc, OuterLoopBodyGenCB, TripCount, "outer"));
// Finalize the function.
Builder.restoreIP(OuterLoop->getAfterIP());
@@ -1770,18 +1738,14 @@ TEST_F(OpenMPIRBuilderTest, TileNestedLoopsWithBounds) {
Call = createPrintfCall(Builder, "i=%d j=%d\\n", {OuterLC, InnerLC});
return Error::success();
};
- Expected<CanonicalLoopInfo *> LoopResult = OMPBuilder.createCanonicalLoop(
+ InnerLoop = cantFail(OMPBuilder.createCanonicalLoop(
OuterCodeGenIP, InnerLoopBodyGenCB, InnerStartVal, InnerStopVal,
- InnerStep, false, false, ComputeIP, "inner");
- assert(LoopResult && "unexpected error");
- InnerLoop = *LoopResult;
+ InnerStep, false, false, ComputeIP, "inner"));
return Error::success();
};
- Expected<CanonicalLoopInfo *> LoopResult = OMPBuilder.createCanonicalLoop(
+ CanonicalLoopInfo *OuterLoop = cantFail(OMPBuilder.createCanonicalLoop(
Loc, OuterLoopBodyGenCB, OuterStartVal, OuterStopVal, OuterStep, false,
- false, ComputeIP, "outer");
- assert(LoopResult && "unexpected error");
- CanonicalLoopInfo *OuterLoop = *LoopResult;
+ false, ComputeIP, "outer"));
// Finalize the function
Builder.restoreIP(OuterLoop->getAfterIP());
@@ -1888,11 +1852,9 @@ TEST_F(OpenMPIRBuilderTest, TileSingleLoopCounts) {
auto LoopBodyGenCB = [&](InsertPointTy CodeGenIP, llvm::Value *LC) {
return Error::success();
};
- Expected<CanonicalLoopInfo *> LoopResult =
+ CanonicalLoopInfo *Loop = cantFail(
OMPBuilder.createCanonicalLoop(Loc, LoopBodyGenCB, StartVal, StopVal,
- StepVal, IsSigned, InclusiveStop);
- assert(LoopResult && "unexpected error");
- CanonicalLoopInfo *Loop = *LoopResult;
+ StepVal, IsSigned, InclusiveStop));
InsertPointTy AfterIP = Loop->getAfterIP();
// Tile the loop.
@@ -2343,20 +2305,18 @@ TEST_F(OpenMPIRBuilderTest, StaticWorkshareLoopTarget) {
Value *StepVal = ConstantInt::get(LCTy, 2);
auto LoopBodyGen = [&](InsertPointTy, Value *) { return Error::success(); };
- Expected<CanonicalLoopInfo *> LoopResult = OMPBuilder.createCanonicalLoop(
- Loc, LoopBodyGen, StartVal, StopVal, StepVal, false, false);
- assert(LoopResult && "unexpected error");
- CanonicalLoopInfo *CLI = *LoopResult;
+ CanonicalLoopInfo *CLI = cantFail(OMPBuilder.createCanonicalLoop(
+ Loc, LoopBodyGen, StartVal, StopVal, StepVal, false, false));
BasicBlock *Preheader = CLI->getPreheader();
Value *TripCount = CLI->getTripCount();
Builder.SetInsertPoint(BB, BB->getFirstInsertionPt());
- OpenMPIRBuilder::InsertPointOrErrorTy AfterIP = OMPBuilder.applyWorkshareLoop(
- DL, CLI, AllocaIP, true, OMP_SCHEDULE_Static, nullptr, false, false,
- false, false, WorksharingLoopType::ForStaticLoop);
- assert(AfterIP && "unexpected error");
- Builder.restoreIP(*AfterIP);
+ OpenMPIRBuilder::InsertPointTy AfterIP =
+ cantFail(OMPBuilder.applyWorkshareLoop(
+ DL, CLI, AllocaIP, true, OMP_SCHEDULE_Static, nullptr, false, false,
+ false, false, WorksharingLoopType::ForStaticLoop));
+ Builder.restoreIP(AfterIP);
Builder.CreateRetVoid();
OMPBuilder.finalize();
@@ -2409,11 +2369,9 @@ TEST_F(OpenMPIRBuilderTest, StaticWorkShareLoop) {
return Error::success();
};
- Expected<CanonicalLoopInfo *> LoopResult = OMPBuilder.createCanonicalLoop(
+ CanonicalLoopInfo *CLI = cantFail(OMPBuilder.createCanonicalLoop(
Loc, LoopBodyGen, StartVal, StopVal, StepVal,
- /*IsSigned=*/false, /*InclusiveStop=*/false);
- assert(LoopResult && "unexpected error");
- CanonicalLoopInfo *CLI = *LoopResult;
+ /*IsSigned=*/false, /*InclusiveStop=*/false));
BasicBlock *Preheader = CLI->getPreheader();
BasicBlock *Body = CLI->getBody();
Value *IV = CLI->getIndVar();
@@ -2422,9 +2380,8 @@ TEST_F(OpenMPIRBuilderTest, StaticWorkShareLoop) {
Builder.SetInsertPoint(BB, BB->getFirstInsertionPt());
InsertPointTy AllocaIP = Builder.saveIP();
- OpenMPIRBuilder::InsertPointOrErrorTy AfterIP = OMPBuilder.applyWorkshareLoop(
- DL, CLI, AllocaIP, /*NeedsBarrier=*/true, OMP_SCHEDULE_Static);
- assert(AfterIP && "unexpected error");
+ cantFail(OMPBuilder.applyWorkshareLoop(
+ DL, CLI, AllocaIP, /*NeedsBarrier=*/true, OMP_SCHEDULE_Static));
BasicBlock *Cond = Body->getSinglePredecessor();
Instruction *Cmp = &*Cond->begin();
@@ -2516,9 +2473,9 @@ TEST_P(OpenMPIRBuilderTestWithIVBits, StaticChunkedWorkshareLoop) {
Value *ChunkSize = ConstantInt::get(LCTy, 5);
InsertPointTy AllocaIP{&F->getEntryBlock(),
F->getEntryBlock().getFirstInsertionPt()};
- OpenMPIRBuilder::InsertPointOrErrorTy AfterIP = OMPBuilder.applyWorkshareLoop(
- DL, CLI, AllocaIP, /*NeedsBarrier=*/true, OMP_SCHEDULE_Static, ChunkSize);
- assert(AfterIP && "unexpected error");
+ cantFail(OMPBuilder.applyWorkshareLoop(DL, CLI, AllocaIP,
+ /*NeedsBarrier=*/true,
+ OMP_SCHEDULE_Static, ChunkSize));
OMPBuilder.finalize();
EXPECT_FALSE(verifyModule(*M, &errs()));
@@ -2609,11 +2566,9 @@ TEST_P(OpenMPIRBuilderTestWithParams, DynamicWorkShareLoop) {
return Error::success();
};
- Expected<CanonicalLoopInfo *> LoopResult = OMPBuilder.createCanonicalLoop(
+ CanonicalLoopInfo *CLI = cantFail(OMPBuilder.createCanonicalLoop(
Loc, LoopBodyGen, StartVal, StopVal, StepVal,
- /*IsSigned=*/false, /*InclusiveStop=*/false);
- assert(LoopResult && "unexpected error");
- CanonicalLoopInfo *CLI = *LoopResult;
+ /*IsSigned=*/false, /*InclusiveStop=*/false));
Builder.SetInsertPoint(BB, BB->getFirstInsertionPt());
InsertPointTy AllocaIP = Builder.saveIP();
@@ -2626,19 +2581,18 @@ TEST_P(OpenMPIRBuilderTestWithParams, DynamicWorkShareLoop) {
BasicBlock *LatchBlock = CLI->getLatch();
Value *IV = CLI->getIndVar();
- OpenMPIRBuilder::InsertPointOrErrorTy EndIP = OMPBuilder.applyWorkshareLoop(
+ OpenMPIRBuilder::InsertPointTy EndIP = cantFail(OMPBuilder.applyWorkshareLoop(
DL, CLI, AllocaIP, /*NeedsBarrier=*/true, getSchedKind(SchedType),
ChunkVal, /*Simd=*/false,
(SchedType & omp::OMPScheduleType::ModifierMonotonic) ==
omp::OMPScheduleType::ModifierMonotonic,
(SchedType & omp::OMPScheduleType::ModifierNonmonotonic) ==
omp::OMPScheduleType::ModifierNonmonotonic,
- /*Ordered=*/false);
- assert(EndIP && "unexpected error");
+ /*Ordered=*/false));
// The returned value should be the "after" point.
- ASSERT_EQ(EndIP->getBlock(), AfterIP.getBlock());
- ASSERT_EQ(EndIP->getPoint(), AfterIP.getPoint());
+ ASSERT_EQ(EndIP.getBlock(), AfterIP.getBlock());
+ ASSERT_EQ(EndIP.getPoint(), AfterIP.getPoint());
auto AllocaIter = BB->begin();
ASSERT_GE(std::distance(BB->begin(), BB->end()), 4);
@@ -2713,7 +2667,7 @@ TEST_P(OpenMPIRBuilderTestWithParams, DynamicWorkShareLoop) {
EXPECT_EQ(NumCallsInExitBlock, 2u);
// Add a termination to our block and check that it is internally consistent.
- Builder.restoreIP(*EndIP);
+ Builder.restoreIP(EndIP);
Builder.CreateRetVoid();
OMPBuilder.finalize();
EXPECT_FALSE(verifyModule(*M, &errs()));
@@ -2756,11 +2710,9 @@ TEST_F(OpenMPIRBuilderTest, DynamicWorkShareLoopOrdered) {
return llvm::Error::success();
};
- Expected<CanonicalLoopInfo *> LoopResult = OMPBuilder.createCanonicalLoop(
+ CanonicalLoopInfo *CLI = cantFail(OMPBuilder.createCanonicalLoop(
Loc, LoopBodyGen, StartVal, StopVal, StepVal,
- /*IsSigned=*/false, /*InclusiveStop=*/false);
- assert(LoopResult && "unexpected error");
- CanonicalLoopInfo *CLI = *LoopResult;
+ /*IsSigned=*/false, /*InclusiveStop=*/false));
Builder.SetInsertPoint(BB, BB->getFirstInsertionPt());
InsertPointTy AllocaIP = Builder.saveIP();
@@ -2772,15 +2724,14 @@ TEST_F(OpenMPIRBuilderTest, DynamicWorkShareLoopOrdered) {
BasicBlock *LatchBlock = CLI->getLatch();
Value *IV = CLI->getIndVar();
- OpenMPIRBuilder::InsertPointOrErrorTy EndIP = OMPBuilder.applyWorkshareLoop(
+ OpenMPIRBuilder::InsertPointTy EndIP = cantFail(OMPBuilder.applyWorkshareLoop(
DL, CLI, AllocaIP, /*NeedsBarrier=*/true, OMP_SCHEDULE_Static, ChunkVal,
/*HasSimdModifier=*/false, /*HasMonotonicModifier=*/false,
/*HasNonmonotonicModifier=*/false,
- /*HasOrderedClause=*/true);
- assert(EndIP && "unexpected error");
+ /*HasOrderedClause=*/true));
// Add a termination to our block and check that it is internally consistent.
- Builder.restoreIP(*EndIP);
+ Builder.restoreIP(EndIP);
Builder.CreateRetVoid();
OMPBuilder.finalize();
EXPECT_FALSE(verifyModule(*M, &errs()));
@@ -2864,10 +2815,9 @@ TEST_F(OpenMPIRBuilderTest, MasterDirective) {
EXPECT_NE(IPBB->end(), IP.getPoint());
};
- OpenMPIRBuilder::InsertPointOrErrorTy AfterIP = OMPBuilder.createMaster(
- Builder, BODYGENCB_WRAPPER(BodyGenCB), FINICB_WRAPPER(FiniCB));
- assert(AfterIP && "unexpected error");
- Builder.restoreIP(*AfterIP);
+ OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(OMPBuilder.createMaster(
+ Builder, BODYGENCB_WRAPPER(BodyGenCB), FINICB_WRAPPER(FiniCB)));
+ Builder.restoreIP(AfterIP);
Value *EntryBBTI = EntryBB->getTerminator();
EXPECT_NE(EntryBBTI, nullptr);
EXPECT_TRUE(isa<BranchInst>(EntryBBTI));
@@ -2945,10 +2895,9 @@ TEST_F(OpenMPIRBuilderTest, MaskedDirective) {
};
Constant *Filter = ConstantInt::get(Type::getInt32Ty(M->getContext()), 0);
- OpenMPIRBuilder::InsertPointOrErrorTy AfterIP = OMPBuilder.createMasked(
- Builder, BODYGENCB_WRAPPER(BodyGenCB), FINICB_WRAPPER(FiniCB), Filter);
- assert(AfterIP && "unexpected error");
- Builder.restoreIP(*AfterIP);
+ OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(OMPBuilder.createMasked(
+ Builder, BODYGENCB_WRAPPER(BodyGenCB), FINICB_WRAPPER(FiniCB), Filter));
+ Builder.restoreIP(AfterIP);
Value *EntryBBTI = EntryBB->getTerminator();
EXPECT_NE(EntryBBTI, nullptr);
EXPECT_TRUE(isa<BranchInst>(EntryBBTI));
@@ -3013,11 +2962,10 @@ TEST_F(OpenMPIRBuilderTest, CriticalDirective) {
};
BasicBlock *EntryBB = Builder.GetInsertBlock();
- OpenMPIRBuilder::InsertPointOrErrorTy AfterIP =
+ OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(
OMPBuilder.createCritical(Builder, BODYGENCB_WRAPPER(BodyGenCB),
- FINICB_WRAPPER(FiniCB), "testCRT", nullptr);
- assert(AfterIP && "unexpected error");
- Builder.restoreIP(*AfterIP);
+ FINICB_WRAPPER(FiniCB), "testCRT", nullptr));
+ Builder.restoreIP(AfterIP);
CallInst *CriticalEntryCI = nullptr;
for (auto &EI : *EntryBB) {
@@ -3264,11 +3212,10 @@ TEST_F(OpenMPIRBuilderTest, OrderedDirectiveThreads) {
// Test for "#omp ordered [threads]"
BasicBlock *EntryBB = Builder.GetInsertBlock();
- OpenMPIRBuilder::InsertPointOrErrorTy AfterIP =
- OMPBuilder.createOrderedThreadsSimd(Builder, BODYGENCB_WRAPPER(BodyGenCB),
- FINICB_WRAPPER(FiniCB), true);
- assert(AfterIP && "unexpected error");
- Builder.restoreIP(*AfterIP);
+ OpenMPIRBuilder::InsertPointTy AfterIP =
+ cantFail(OMPBuilder.createOrderedThreadsSimd(
+ Builder, BODYGENCB_WRAPPER(BodyGenCB), FINICB_WRAPPER(FiniCB), true));
+ Builder.restoreIP(AfterIP);
Builder.CreateRetVoid();
OMPBuilder.finalize();
@@ -3338,11 +3285,10 @@ TEST_F(OpenMPIRBuilderTest, OrderedDirectiveSimd) {
// Test for "#omp ordered simd"
BasicBlock *EntryBB = Builder.GetInsertBlock();
- OpenMPIRBuilder::InsertPointOrErrorTy AfterIP =
+ OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(
OMPBuilder.createOrderedThreadsSimd(Builder, BODYGENCB_WRAPPER(BodyGenCB),
- FINICB_WRAPPER(FiniCB), false);
- assert(AfterIP && "unexpected error");
- Builder.restoreIP(*AfterIP);
+ FINICB_WRAPPER(FiniCB), false));
+ Builder.restoreIP(AfterIP);
Builder.CreateRetVoid();
OMPBuilder.finalize();
@@ -3455,11 +3401,10 @@ TEST_F(OpenMPIRBuilderTest, SingleDirective) {
EXPECT_NE(IPBB->end(), IP.getPoint());
};
- OpenMPIRBuilder::InsertPointOrErrorTy AfterIP =
+ OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(
OMPBuilder.createSingle(Builder, BODYGENCB_WRAPPER(BodyGenCB),
- FINICB_WRAPPER(FiniCB), /*IsNowait*/ false);
- assert(AfterIP && "unexpected error");
- Builder.restoreIP(*AfterIP);
+ FINICB_WRAPPER(FiniCB), /*IsNowait*/ false));
+ Builder.restoreIP(AfterIP);
Value *EntryBBTI = EntryBB->getTerminator();
EXPECT_NE(EntryBBTI, nullptr);
EXPECT_TRUE(isa<BranchInst>(EntryBBTI));
@@ -3548,11 +3493,10 @@ TEST_F(OpenMPIRBuilderTest, SingleDirectiveNowait) {
EXPECT_NE(IPBB->end(), IP.getPoint());
};
- OpenMPIRBuilder::InsertPointOrErrorTy AfterIP =
+ OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(
OMPBuilder.createSingle(Builder, BODYGENCB_WRAPPER(BodyGenCB),
- FINICB_WRAPPER(FiniCB), /*IsNowait*/ true);
- assert(AfterIP && "unexpected error");
- Builder.restoreIP(*AfterIP);
+ FINICB_WRAPPER(FiniCB), /*IsNowait*/ true));
+ Builder.restoreIP(AfterIP);
Value *EntryBBTI = EntryBB->getTerminator();
EXPECT_NE(EntryBBTI, nullptr);
EXPECT_TRUE(isa<BranchInst>(EntryBBTI));
@@ -3670,11 +3614,10 @@ TEST_F(OpenMPIRBuilderTest, SingleDirectiveCopyPrivate) {
EXPECT_NE(IPBB->end(), IP.getPoint());
};
- OpenMPIRBuilder::InsertPointOrErrorTy AfterIP = OMPBuilder.createSingle(
+ OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(OMPBuilder.createSingle(
Builder, BODYGENCB_WRAPPER(BodyGenCB), FINICB_WRAPPER(FiniCB),
- /*IsNowait*/ false, {CPVar}, {CopyFunc});
- assert(AfterIP && "unexpected error");
- Builder.restoreIP(*AfterIP);
+ /*IsNowait*/ false, {CPVar}, {CopyFunc}));
+ Builder.restoreIP(AfterIP);
Value *EntryBBTI = EntryBB->getTerminator();
EXPECT_NE(EntryBBTI, nullptr);
EXPECT_TRUE(isa<BranchInst>(EntryBBTI));
@@ -3935,10 +3878,10 @@ TEST_F(OpenMPIRBuilderTest, OMPAtomicUpdate) {
Sub = IRB.CreateSub(ConstVal, Atomic);
return Sub;
};
- OpenMPIRBuilder::InsertPointOrErrorTy AfterIP = OMPBuilder.createAtomicUpdate(
- Builder, AllocaIP, X, Expr, AO, RMWOp, UpdateOp, IsXLHSInRHSPart);
- assert(AfterIP && "unexpected error");
- Builder.restoreIP(*AfterIP);
+ OpenMPIRBuilder::InsertPointTy AfterIP =
+ cantFail(OMPBuilder.createAtomicUpdate(Builder, AllocaIP, X, Expr, AO,
+ RMWOp, UpdateOp, IsXLHSInRHSPart));
+ Builder.restoreIP(AfterIP);
BasicBlock *ContBB = EntryBB->getSingleSuccessor();
BranchInst *ContTI = dyn_cast<BranchInst>(ContBB->getTerminator());
EXPECT_NE(ContTI, nullptr);
@@ -4004,10 +3947,10 @@ TEST_F(OpenMPIRBuilderTest, OMPAtomicUpdateFloat) {
Sub = IRB.CreateFSub(ConstVal, Atomic);
return Sub;
};
- OpenMPIRBuilder::InsertPointOrErrorTy AfterIP = OMPBuilder.createAtomicUpdate(
- Builder, AllocaIP, X, Expr, AO, RMWOp, UpdateOp, IsXLHSInRHSPart);
- assert(AfterIP && "unexpected error");
- Builder.restoreIP(*AfterIP);
+ OpenMPIRBuilder::InsertPointTy AfterIP =
+ cantFail(OMPBuilder.createAtomicUpdate(Builder, AllocaIP, X, Expr, AO,
+ RMWOp, UpdateOp, IsXLHSInRHSPart));
+ Builder.restoreIP(AfterIP);
BasicBlock *ContBB = EntryBB->getSingleSuccessor();
BranchInst *ContTI = dyn_cast<BranchInst>(ContBB->getTerminator());
EXPECT_NE(ContTI, nullptr);
@@ -4072,10 +4015,10 @@ TEST_F(OpenMPIRBuilderTest, OMPAtomicUpdateIntr) {
Sub = IRB.CreateSub(ConstVal, Atomic);
return Sub;
};
- OpenMPIRBuilder::InsertPointOrErrorTy AfterIP = OMPBuilder.createAtomicUpdate(
- Builder, AllocaIP, X, Expr, AO, RMWOp, UpdateOp, IsXLHSInRHSPart);
- assert(AfterIP && "unexpected error");
- Builder.restoreIP(*AfterIP);
+ OpenMPIRBuilder::InsertPointTy AfterIP =
+ cantFail(OMPBuilder.createAtomicUpdate(Builder, AllocaIP, X, Expr, AO,
+ RMWOp, UpdateOp, IsXLHSInRHSPart));
+ Builder.restoreIP(AfterIP);
BasicBlock *ContBB = EntryBB->getSingleSuccessor();
BranchInst *ContTI = dyn_cast<BranchInst>(ContBB->getTerminator());
EXPECT_NE(ContTI, nullptr);
@@ -4146,12 +4089,11 @@ TEST_F(OpenMPIRBuilderTest, OMPAtomicCapture) {
// integer update - not used
auto UpdateOp = [&](Value *Atomic, IRBuilder<> &IRB) { return nullptr; };
- OpenMPIRBuilder::InsertPointOrErrorTy AfterIP =
- OMPBuilder.createAtomicCapture(Builder, AllocaIP, X, V, Expr, AO, RMWOp,
- UpdateOp, UpdateExpr, IsPostfixUpdate,
- IsXLHSInRHSPart);
- assert(AfterIP && "unexpected error");
- Builder.restoreIP(*AfterIP);
+ OpenMPIRBuilder::InsertPointTy AfterIP =
+ cantFail(OMPBuilder.createAtomicCapture(
+ Builder, AllocaIP, X, V, Expr, AO, RMWOp, UpdateOp, UpdateExpr,
+ IsPostfixUpdate, IsXLHSInRHSPart));
+ Builder.restoreIP(AfterIP);
EXPECT_EQ(EntryBB->getParent()->size(), 1U);
AtomicRMWInst *ARWM = dyn_cast<AtomicRMWInst>(Init->getNextNode());
EXPECT_NE(ARWM, nullptr);
@@ -4511,11 +4453,10 @@ TEST_F(OpenMPIRBuilderTest, CreateTeams) {
};
OpenMPIRBuilder::LocationDescription Loc({Builder.saveIP(), DL});
- OpenMPIRBuilder::InsertPointOrErrorTy AfterIP = OMPBuilder.createTeams(
+ OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(OMPBuilder.createTeams(
Builder, BodyGenCB, /*NumTeamsLower=*/nullptr, /*NumTeamsUpper=*/nullptr,
- /*ThreadLimit=*/nullptr, /*IfExpr=*/nullptr);
- assert(AfterIP && "unexpected error");
- Builder.restoreIP(*AfterIP);
+ /*ThreadLimit=*/nullptr, /*IfExpr=*/nullptr));
+ Builder.restoreIP(AfterIP);
OMPBuilder.finalize();
Builder.CreateRetVoid();
@@ -4576,12 +4517,11 @@ TEST_F(OpenMPIRBuilderTest, CreateTeamsWithThreadLimit) {
};
// `F` has an argument - an integer, so we use that as the thread limit.
- OpenMPIRBuilder::InsertPointOrErrorTy AfterIP = OMPBuilder.createTeams(
+ OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(OMPBuilder.createTeams(
/*=*/Builder, BodyGenCB, /*NumTeamsLower=*/nullptr,
/*NumTeamsUpper=*/nullptr, /*ThreadLimit=*/F->arg_begin(),
- /*IfExpr=*/nullptr);
- assert(AfterIP && "unexpected error");
- Builder.restoreIP(*AfterIP);
+ /*IfExpr=*/nullptr));
+ Builder.restoreIP(AfterIP);
Builder.CreateRetVoid();
OMPBuilder.finalize();
@@ -4630,14 +4570,13 @@ TEST_F(OpenMPIRBuilderTest, CreateTeamsWithNumTeamsUpper) {
// `F` already has an integer argument, so we use that as upper bound to
// `num_teams`
- OpenMPIRBuilder::InsertPointOrErrorTy AfterIP =
- OMPBuilder.createTeams(Builder, BodyGenCB,
- /*NumTeamsLower=*/nullptr,
- /*NumTeamsUpper=*/F->arg_begin(),
- /*ThreadLimit=*/nullptr,
- /*IfExpr=*/nullptr);
- assert(AfterIP && "unexpected error");
- Builder.restoreIP(*AfterIP);
+ OpenMPIRBuilder::InsertPointTy AfterIP =
+ cantFail(OMPBuilder.createTeams(Builder, BodyGenCB,
+ /*NumTeamsLower=*/nullptr,
+ /*NumTeamsUpper=*/F->arg_begin(),
+ /*ThreadLimit=*/nullptr,
+ /*IfExpr=*/nullptr));
+ Builder.restoreIP(AfterIP);
Builder.CreateRetVoid();
OMPBuilder.finalize();
@@ -4691,11 +4630,10 @@ TEST_F(OpenMPIRBuilderTest, CreateTeamsWithNumTeamsBoth) {
// `F` already has an integer argument, so we use that as upper bound to
// `num_teams`
- OpenMPIRBuilder::InsertPointOrErrorTy AfterIP =
+ OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(
OMPBuilder.createTeams(Builder, BodyGenCB, NumTeamsLower, NumTeamsUpper,
- /*ThreadLimit=*/nullptr, /*IfExpr=*/nullptr);
- assert(AfterIP && "unexpected error");
- Builder.restoreIP(*AfterIP);
+ /*ThreadLimit=*/nullptr, /*IfExpr=*/nullptr));
+ Builder.restoreIP(AfterIP);
Builder.CreateRetVoid();
OMPBuilder.finalize();
@@ -4755,10 +4693,9 @@ TEST_F(OpenMPIRBuilderTest, CreateTeamsWithNumTeamsAndThreadLimit) {
};
OpenMPIRBuilder::LocationDescription Loc({Builder.saveIP(), DL});
- OpenMPIRBuilder::InsertPointOrErrorTy AfterIP = OMPBuilder.createTeams(
- Builder, BodyGenCB, NumTeamsLower, NumTeamsUpper, ThreadLimit, nullptr);
- assert(AfterIP && "unexpected error");
- Builder.restoreIP(*AfterIP);
+ OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(OMPBuilder.createTeams(
+ Builder, BodyGenCB, NumTeamsLower, NumTeamsUpper, ThreadLimit, nullptr));
+ Builder.restoreIP(AfterIP);
Builder.CreateRetVoid();
OMPBuilder.finalize();
@@ -4810,11 +4747,10 @@ TEST_F(OpenMPIRBuilderTest, CreateTeamsWithIfCondition) {
// `F` already has an integer argument, so we use that as upper bound to
// `num_teams`
- OpenMPIRBuilder::InsertPointOrErrorTy AfterIP = OMPBuilder.createTeams(
+ OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(OMPBuilder.createTeams(
Builder, BodyGenCB, /*NumTeamsLower=*/nullptr, /*NumTeamsUpper=*/nullptr,
- /*ThreadLimit=*/nullptr, IfExpr);
- assert(AfterIP && "unexpected error");
- Builder.restoreIP(*AfterIP);
+ /*ThreadLimit=*/nullptr, IfExpr));
+ Builder.restoreIP(AfterIP);
Builder.CreateRetVoid();
OMPBuilder.finalize();
@@ -4876,10 +4812,9 @@ TEST_F(OpenMPIRBuilderTest, CreateTeamsWithIfConditionAndNumTeams) {
// `F` already has an integer argument, so we use that as upper bound to
// `num_teams`
- OpenMPIRBuilder::InsertPointOrErrorTy AfterIP = OMPBuilder.createTeams(
- Builder, BodyGenCB, NumTeamsLower, NumTeamsUpper, ThreadLimit, IfExpr);
- assert(AfterIP && "unexpected error");
- Builder.restoreIP(*AfterIP);
+ OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(OMPBuilder.createTeams(
+ Builder, BodyGenCB, NumTeamsLower, NumTeamsUpper, ThreadLimit, IfExpr));
+ Builder.restoreIP(AfterIP);
Builder.CreateRetVoid();
OMPBuilder.finalize();
@@ -5139,13 +5074,12 @@ TEST_F(OpenMPIRBuilderTest, CreateReductions) {
// Do nothing in finalization.
auto FiniCB = [&](InsertPointTy CodeGenIP) { return Error::success(); };
- OpenMPIRBuilder::InsertPointOrErrorTy AfterIP =
+ OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(
OMPBuilder.createParallel(Loc, OuterAllocaIP, BodyGenCB, PrivCB, FiniCB,
/* IfCondition */ nullptr,
/* NumThreads */ nullptr, OMP_PROC_BIND_default,
- /* IsCancellable */ false);
- assert(AfterIP && "unexpected error");
- Builder.restoreIP(*AfterIP);
+ /* IsCancellable */ false));
+ Builder.restoreIP(AfterIP);
OpenMPIRBuilder::ReductionInfo ReductionInfos[] = {
{SumType, SumReduced, SumPrivatized,
@@ -5157,13 +5091,10 @@ TEST_F(OpenMPIRBuilderTest, CreateReductions) {
OMPBuilder.Config.setIsGPU(false);
bool ReduceVariableByRef[] = {false, false};
+ cantFail(OMPBuilder.createReductions(BodyIP, BodyAllocaIP, ReductionInfos,
+ ReduceVariableByRef));
- OpenMPIRBuilder::InsertPointOrErrorTy ReductionsIP =
- OMPBuilder.createReductions(BodyIP, BodyAllocaIP, ReductionInfos,
- ReduceVariableByRef);
- assert(ReductionsIP && "unexpected error");
-
- Builder.restoreIP(*AfterIP);
+ Builder.restoreIP(AfterIP);
Builder.CreateRetVoid();
OMPBuilder.finalize(F);
@@ -5399,42 +5330,36 @@ TEST_F(OpenMPIRBuilderTest, CreateTwoReductions) {
// Do nothing in finalization.
auto FiniCB = [&](InsertPointTy CodeGenIP) { return Error::success(); };
- OpenMPIRBuilder::InsertPointOrErrorTy AfterIP1 =
+ OpenMPIRBuilder::InsertPointTy AfterIP1 = cantFail(
OMPBuilder.createParallel(Loc, OuterAllocaIP, FirstBodyGenCB, PrivCB,
FiniCB, /* IfCondition */ nullptr,
/* NumThreads */ nullptr, OMP_PROC_BIND_default,
- /* IsCancellable */ false);
- assert(AfterIP1 && "unexpected error");
- Builder.restoreIP(*AfterIP1);
- OpenMPIRBuilder::InsertPointOrErrorTy AfterIP2 = OMPBuilder.createParallel(
+ /* IsCancellable */ false));
+ Builder.restoreIP(AfterIP1);
+ OpenMPIRBuilder::InsertPointTy AfterIP2 = cantFail(OMPBuilder.createParallel(
{Builder.saveIP(), DL}, OuterAllocaIP, SecondBodyGenCB, PrivCB, FiniCB,
/* IfCondition */ nullptr,
/* NumThreads */ nullptr, OMP_PROC_BIND_default,
- /* IsCancellable */ false);
- assert(AfterIP2 && "unexpected error");
- Builder.restoreIP(*AfterIP2);
+ /* IsCancellable */ false));
+ Builder.restoreIP(AfterIP2);
OMPBuilder.Config.setIsGPU(false);
bool ReduceVariableByRef[] = {false};
- OpenMPIRBuilder::InsertPointOrErrorTy ReductionsIP1 =
- OMPBuilder.createReductions(
- FirstBodyIP, FirstBodyAllocaIP,
- {{SumType, SumReduced, SumPrivatized,
- /*EvaluationKind=*/OpenMPIRBuilder::EvalKind::Scalar, sumReduction,
- /*ReductionGenClang=*/nullptr, sumAtomicReduction}},
- ReduceVariableByRef);
- assert(ReductionsIP1 && "unexpected error");
- OpenMPIRBuilder::InsertPointOrErrorTy ReductionsIP2 =
- OMPBuilder.createReductions(
- SecondBodyIP, SecondBodyAllocaIP,
- {{XorType, XorReduced, XorPrivatized,
- /*EvaluationKind=*/OpenMPIRBuilder::EvalKind::Scalar, xorReduction,
- /*ReductionGenClang=*/nullptr, xorAtomicReduction}},
- ReduceVariableByRef);
- assert(ReductionsIP2 && "unexpected error");
-
- Builder.restoreIP(*AfterIP2);
+ cantFail(OMPBuilder.createReductions(
+ FirstBodyIP, FirstBodyAllocaIP,
+ {{SumType, SumReduced, SumPrivatized,
+ /*EvaluationKind=*/OpenMPIRBuilder::EvalKind::Scalar, sumReduction,
+ /*ReductionGenClang=*/nullptr, sumAtomicReduction}},
+ ReduceVariableByRef));
+ cantFail(OMPBuilder.createReductions(
+ SecondBodyIP, SecondBodyAllocaIP,
+ {{XorType, XorReduced, XorPrivatized,
+ /*EvaluationKind=*/OpenMPIRBuilder::EvalKind::Scalar, xorReduction,
+ /*ReductionGenClang=*/nullptr, xorAtomicReduction}},
+ ReduceVariableByRef));
+
+ Builder.restoreIP(AfterIP2);
Builder.CreateRetVoid();
OMPBuilder.finalize(F);
@@ -5512,10 +5437,9 @@ TEST_F(OpenMPIRBuilderTest, CreateSectionsSimple) {
llvm::Value *&ReplVal) { return CodeGenIP; };
IRBuilder<>::InsertPoint AllocaIP(&F->getEntryBlock(),
F->getEntryBlock().getFirstInsertionPt());
- OpenMPIRBuilder::InsertPointOrErrorTy AfterIP = OMPBuilder.createSections(
- Loc, AllocaIP, SectionCBVector, PrivCB, FiniCB, false, false);
- assert(AfterIP && "unexpected error");
- Builder.restoreIP(*AfterIP);
+ OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(OMPBuilder.createSections(
+ Loc, AllocaIP, SectionCBVector, PrivCB, FiniCB, false, false));
+ Builder.restoreIP(AfterIP);
Builder.CreateRetVoid(); // Required at the end of the function
EXPECT_NE(F->getEntryBlock().getTerminator(), nullptr);
EXPECT_FALSE(verifyModule(*M, &errs()));
@@ -5569,11 +5493,10 @@ TEST_F(OpenMPIRBuilderTest, CreateSections) {
IRBuilder<>::InsertPoint AllocaIP(&F->getEntryBlock(),
F->getEntryBlock().getFirstInsertionPt());
- OpenMPIRBuilder::InsertPointOrErrorTy AfterIP =
- OMPBuilder.createSections(Loc, AllocaIP, SectionCBVector, PrivCB,
- FINICB_WRAPPER(FiniCB), false, false);
- assert(AfterIP && "unexpected error");
- Builder.restoreIP(*AfterIP);
+ OpenMPIRBuilder::InsertPointTy AfterIP =
+ cantFail(OMPBuilder.createSections(Loc, AllocaIP, SectionCBVector, PrivCB,
+ FINICB_WRAPPER(FiniCB), false, false));
+ Builder.restoreIP(AfterIP);
Builder.CreateRetVoid(); // Required at the end of the function
// Switch BB's predecessor is loop condition BB, whose successor at index 1 is
@@ -5659,10 +5582,9 @@ TEST_F(OpenMPIRBuilderTest, CreateSectionsNoWait) {
llvm::Value *&ReplVal) { return CodeGenIP; };
auto FiniCB = [&](InsertPointTy IP) { return Error::success(); };
- OpenMPIRBuilder::InsertPointOrErrorTy AfterIP = OMPBuilder.createSections(
- Loc, AllocaIP, SectionCBVector, PrivCB, FiniCB, false, true);
- assert(AfterIP && "unexpected error");
- Builder.restoreIP(*AfterIP);
+ OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(OMPBuilder.createSections(
+ Loc, AllocaIP, SectionCBVector, PrivCB, FiniCB, false, true));
+ Builder.restoreIP(AfterIP);
Builder.CreateRetVoid(); // Required at the end of the function
for (auto &Inst : instructions(*F)) {
EXPECT_FALSE(isa<CallInst>(Inst) &&
@@ -5883,11 +5805,10 @@ TEST_F(OpenMPIRBuilderTest, TargetEnterData) {
OMPBuilder.Config.setIsGPU(true);
llvm::omp::RuntimeFunction RTLFunc = OMPRTL___tgt_target_data_begin_mapper;
- OpenMPIRBuilder::InsertPointOrErrorTy AfterIP = OMPBuilder.createTargetData(
+ OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(OMPBuilder.createTargetData(
Loc, AllocaIP, Builder.saveIP(), Builder.getInt64(DeviceID),
- /* IfCond= */ nullptr, Info, GenMapInfoCB, &RTLFunc);
- assert(AfterIP && "unexpected error");
- Builder.restoreIP(*AfterIP);
+ /* IfCond= */ nullptr, Info, GenMapInfoCB, &RTLFunc));
+ Builder.restoreIP(AfterIP);
CallInst *TargetDataCall = dyn_cast<CallInst>(&BB->back());
EXPECT_NE(TargetDataCall, nullptr);
@@ -5944,11 +5865,10 @@ TEST_F(OpenMPIRBuilderTest, TargetExitData) {
OMPBuilder.Config.setIsGPU(true);
llvm::omp::RuntimeFunction RTLFunc = OMPRTL___tgt_target_data_end_mapper;
- OpenMPIRBuilder::InsertPointOrErrorTy AfterIP = OMPBuilder.createTargetData(
+ OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(OMPBuilder.createTargetData(
Loc, AllocaIP, Builder.saveIP(), Builder.getInt64(DeviceID),
- /* IfCond= */ nullptr, Info, GenMapInfoCB, &RTLFunc);
- assert(AfterIP && "unexpected error");
- Builder.restoreIP(*AfterIP);
+ /* IfCond= */ nullptr, Info, GenMapInfoCB, &RTLFunc));
+ Builder.restoreIP(AfterIP);
CallInst *TargetDataCall = dyn_cast<CallInst>(&BB->back());
EXPECT_NE(TargetDataCall, nullptr);
@@ -6054,12 +5974,11 @@ TEST_F(OpenMPIRBuilderTest, TargetDataRegion) {
return Builder.saveIP();
};
- OpenMPIRBuilder::InsertPointOrErrorTy TargetDataIP1 =
- OMPBuilder.createTargetData(
+ OpenMPIRBuilder::InsertPointTy TargetDataIP1 =
+ cantFail(OMPBuilder.createTargetData(
Loc, AllocaIP, Builder.saveIP(), Builder.getInt64(DeviceID),
- /* IfCond= */ nullptr, Info, GenMapInfoCB, nullptr, BodyCB);
- assert(TargetDataIP1 && "unexpected error");
- Builder.restoreIP(*TargetDataIP1);
+ /* IfCond= */ nullptr, Info, GenMapInfoCB, nullptr, BodyCB));
+ Builder.restoreIP(TargetDataIP1);
CallInst *TargetDataCall = dyn_cast<CallInst>(&BB->back());
EXPECT_NE(TargetDataCall, nullptr);
@@ -6082,12 +6001,11 @@ TEST_F(OpenMPIRBuilderTest, TargetDataRegion) {
EXPECT_EQ(TargetDataCall, nullptr);
return Builder.saveIP();
};
- OpenMPIRBuilder::InsertPointOrErrorTy TargetDataIP2 =
- OMPBuilder.createTargetData(
+ OpenMPIRBuilder::InsertPointTy TargetDataIP2 =
+ cantFail(OMPBuilder.createTargetData(
Loc, AllocaIP, Builder.saveIP(), Builder.getInt64(DeviceID),
- /* IfCond= */ nullptr, Info, GenMapInfoCB, nullptr, BodyTargetCB);
- assert(TargetDataIP2 && "unexpected error");
- Builder.restoreIP(*TargetDataIP2);
+ /* IfCond= */ nullptr, Info, GenMapInfoCB, nullptr, BodyTargetCB));
+ Builder.restoreIP(TargetDataIP2);
EXPECT_TRUE(CheckDevicePassBodyGen);
Builder.CreateRetVoid();
@@ -6182,11 +6100,10 @@ TEST_F(OpenMPIRBuilderTest, TargetRegion) {
TargetRegionEntryInfo EntryInfo("func", 42, 4711, 17);
OpenMPIRBuilder::LocationDescription OmpLoc({Builder.saveIP(), DL});
- OpenMPIRBuilder::InsertPointOrErrorTy AfterIP = OMPBuilder.createTarget(
+ OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(OMPBuilder.createTarget(
OmpLoc, /*IsOffloadEntry=*/true, Builder.saveIP(), Builder.saveIP(),
- EntryInfo, -1, 0, Inputs, GenMapInfoCB, BodyGenCB, SimpleArgAccessorCB);
- assert(AfterIP && "unexpected error");
- Builder.restoreIP(*AfterIP);
+ EntryInfo, -1, 0, Inputs, GenMapInfoCB, BodyGenCB, SimpleArgAccessorCB));
+ Builder.restoreIP(AfterIP);
OMPBuilder.finalize();
Builder.CreateRetVoid();
@@ -6292,13 +6209,12 @@ TEST_F(OpenMPIRBuilderTest, TargetRegionDevice) {
TargetRegionEntryInfo EntryInfo("parent", /*DeviceID=*/1, /*FileID=*/2,
/*Line=*/3, /*Count=*/0);
- OpenMPIRBuilder::InsertPointOrErrorTy AfterIP =
+ OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(
OMPBuilder.createTarget(Loc, /*IsOffloadEntry=*/true, EntryIP, EntryIP,
EntryInfo, /*NumTeams=*/-1,
/*NumThreads=*/0, CapturedArgs, GenMapInfoCB,
- BodyGenCB, SimpleArgAccessorCB);
- assert(AfterIP && "unexpected error");
- Builder.restoreIP(*AfterIP);
+ BodyGenCB, SimpleArgAccessorCB));
+ Builder.restoreIP(AfterIP);
Builder.CreateRetVoid();
OMPBuilder.finalize();
@@ -6443,13 +6359,12 @@ TEST_F(OpenMPIRBuilderTest, ConstantAllocaRaise) {
TargetRegionEntryInfo EntryInfo("parent", /*DeviceID=*/1, /*FileID=*/2,
/*Line=*/3, /*Count=*/0);
- OpenMPIRBuilder::InsertPointOrErrorTy AfterIP =
+ OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(
OMPBuilder.createTarget(Loc, /*IsOffloadEntry=*/true, EntryIP, EntryIP,
EntryInfo, /*NumTeams=*/-1,
/*NumThreads=*/0, CapturedArgs, GenMapInfoCB,
- BodyGenCB, SimpleArgAccessorCB);
- assert(AfterIP && "unexpected error");
- Builder.restoreIP(*AfterIP);
+ BodyGenCB, SimpleArgAccessorCB));
+ Builder.restoreIP(AfterIP);
Builder.CreateRetVoid();
OMPBuilder.finalize();
@@ -6568,10 +6483,10 @@ TEST_F(OpenMPIRBuilderTest, CreateTask) {
BasicBlock *BodyBB = splitBB(Builder, /*CreateBranch=*/true, "alloca.split");
OpenMPIRBuilder::LocationDescription Loc(
InsertPointTy(BodyBB, BodyBB->getFirstInsertionPt()), DL);
- OpenMPIRBuilder::InsertPointOrErrorTy AfterIP = OMPBuilder.createTask(
- Loc, InsertPointTy(AllocaBB, AllocaBB->getFirstInsertionPt()), BodyGenCB);
- assert(AfterIP && "unexpected error");
- Builder.restoreIP(*AfterIP);
+ OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(OMPBuilder.createTask(
+ Loc, InsertPointTy(AllocaBB, AllocaBB->getFirstInsertionPt()),
+ BodyGenCB));
+ Builder.restoreIP(AfterIP);
OMPBuilder.finalize();
Builder.CreateRetVoid();
@@ -6677,10 +6592,10 @@ TEST_F(OpenMPIRBuilderTest, CreateTaskNoArgs) {
BasicBlock *BodyBB = splitBB(Builder, /*CreateBranch=*/true, "alloca.split");
OpenMPIRBuilder::LocationDescription Loc(
InsertPointTy(BodyBB, BodyBB->getFirstInsertionPt()), DL);
- OpenMPIRBuilder::InsertPointOrErrorTy AfterIP = OMPBuilder.createTask(
- Loc, InsertPointTy(AllocaBB, AllocaBB->getFirstInsertionPt()), BodyGenCB);
- assert(AfterIP && "unexpected error");
- Builder.restoreIP(*AfterIP);
+ OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(OMPBuilder.createTask(
+ Loc, InsertPointTy(AllocaBB, AllocaBB->getFirstInsertionPt()),
+ BodyGenCB));
+ Builder.restoreIP(AfterIP);
OMPBuilder.finalize();
Builder.CreateRetVoid();
@@ -6709,11 +6624,10 @@ TEST_F(OpenMPIRBuilderTest, CreateTaskUntied) {
BasicBlock *BodyBB = splitBB(Builder, /*CreateBranch=*/true, "alloca.split");
OpenMPIRBuilder::LocationDescription Loc(
InsertPointTy(BodyBB, BodyBB->getFirstInsertionPt()), DL);
- OpenMPIRBuilder::InsertPointOrErrorTy AfterIP = OMPBuilder.createTask(
+ OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(OMPBuilder.createTask(
Loc, InsertPointTy(AllocaBB, AllocaBB->getFirstInsertionPt()), BodyGenCB,
- /*Tied=*/false);
- assert(AfterIP && "unexpected error");
- Builder.restoreIP(*AfterIP);
+ /*Tied=*/false));
+ Builder.restoreIP(AfterIP);
OMPBuilder.finalize();
Builder.CreateRetVoid();
@@ -6750,11 +6664,10 @@ TEST_F(OpenMPIRBuilderTest, CreateTaskDepend) {
Type::getInt32Ty(M->getContext()), InDep);
DDS.push_back(DDIn);
}
- OpenMPIRBuilder::InsertPointOrErrorTy AfterIP = OMPBuilder.createTask(
+ OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(OMPBuilder.createTask(
Loc, InsertPointTy(AllocaBB, AllocaBB->getFirstInsertionPt()), BodyGenCB,
- /*Tied=*/false, /*Final*/ nullptr, /*IfCondition*/ nullptr, DDS);
- assert(AfterIP && "unexpected error");
- Builder.restoreIP(*AfterIP);
+ /*Tied=*/false, /*Final*/ nullptr, /*IfCondition*/ nullptr, DDS));
+ Builder.restoreIP(AfterIP);
OMPBuilder.finalize();
Builder.CreateRetVoid();
@@ -6824,11 +6737,10 @@ TEST_F(OpenMPIRBuilderTest, CreateTaskFinal) {
CmpInst::Predicate::ICMP_EQ, F->getArg(0),
ConstantInt::get(Type::getInt32Ty(M->getContext()), 0U));
OpenMPIRBuilder::LocationDescription Loc(Builder.saveIP(), DL);
- OpenMPIRBuilder::InsertPointOrErrorTy AfterIP =
- OMPBuilder.createTask(Loc, AllocaIP, BodyGenCB,
- /*Tied=*/false, Final);
- assert(AfterIP && "unexpected error");
- Builder.restoreIP(*AfterIP);
+ OpenMPIRBuilder::InsertPointTy AfterIP =
+ cantFail(OMPBuilder.createTask(Loc, AllocaIP, BodyGenCB,
+ /*Tied=*/false, Final));
+ Builder.restoreIP(AfterIP);
OMPBuilder.finalize();
Builder.CreateRetVoid();
@@ -6883,11 +6795,10 @@ TEST_F(OpenMPIRBuilderTest, CreateTaskIfCondition) {
CmpInst::Predicate::ICMP_EQ, F->getArg(0),
ConstantInt::get(Type::getInt32Ty(M->getContext()), 0U));
OpenMPIRBuilder::LocationDescription Loc(Builder.saveIP(), DL);
- OpenMPIRBuilder::InsertPointOrErrorTy AfterIP =
+ OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(
OMPBuilder.createTask(Loc, AllocaIP, BodyGenCB,
- /*Tied=*/false, /*Final=*/nullptr, IfCondition);
- assert(AfterIP && "unexpected error");
- Builder.restoreIP(*AfterIP);
+ /*Tied=*/false, /*Final=*/nullptr, IfCondition));
+ Builder.restoreIP(AfterIP);
OMPBuilder.finalize();
Builder.CreateRetVoid();
@@ -6978,10 +6889,10 @@ TEST_F(OpenMPIRBuilderTest, CreateTaskgroup) {
BasicBlock *BodyBB = splitBB(Builder, /*CreateBranch=*/true, "alloca.split");
OpenMPIRBuilder::LocationDescription Loc(
InsertPointTy(BodyBB, BodyBB->getFirstInsertionPt()), DL);
- OpenMPIRBuilder::InsertPointOrErrorTy AfterIP = OMPBuilder.createTaskgroup(
- Loc, InsertPointTy(AllocaBB, AllocaBB->getFirstInsertionPt()), BodyGenCB);
- assert(AfterIP && "unexpected error");
- Builder.restoreIP(*AfterIP);
+ OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(OMPBuilder.createTaskgroup(
+ Loc, InsertPointTy(AllocaBB, AllocaBB->getFirstInsertionPt()),
+ BodyGenCB));
+ Builder.restoreIP(AfterIP);
OMPBuilder.finalize();
Builder.CreateRetVoid();
@@ -7057,10 +6968,9 @@ TEST_F(OpenMPIRBuilderTest, CreateTaskgroupWithTasks) {
return Error::success();
};
OpenMPIRBuilder::LocationDescription Loc(Builder.saveIP(), DL);
- OpenMPIRBuilder::InsertPointOrErrorTy TaskIP1 =
- OMPBuilder.createTask(Loc, AllocaIP, TaskBodyGenCB1);
- assert(TaskIP1 && "unexpected error");
- Builder.restoreIP(*TaskIP1);
+ OpenMPIRBuilder::InsertPointTy TaskIP1 =
+ cantFail(OMPBuilder.createTask(Loc, AllocaIP, TaskBodyGenCB1));
+ Builder.restoreIP(TaskIP1);
auto TaskBodyGenCB2 = [&](InsertPointTy AllocaIP, InsertPointTy CodeGenIP) {
Builder.restoreIP(CodeGenIP);
@@ -7071,10 +6981,9 @@ TEST_F(OpenMPIRBuilderTest, CreateTaskgroupWithTasks) {
return Error::success();
};
OpenMPIRBuilder::LocationDescription Loc2(Builder.saveIP(), DL);
- OpenMPIRBuilder::InsertPointOrErrorTy TaskIP2 =
- OMPBuilder.createTask(Loc2, AllocaIP, TaskBodyGenCB2);
- assert(TaskIP2 && "unexpected error");
- Builder.restoreIP(*TaskIP2);
+ OpenMPIRBuilder::InsertPointTy TaskIP2 =
+ cantFail(OMPBuilder.createTask(Loc2, AllocaIP, TaskBodyGenCB2));
+ Builder.restoreIP(TaskIP2);
return Error::success();
};
@@ -7082,10 +6991,10 @@ TEST_F(OpenMPIRBuilderTest, CreateTaskgroupWithTasks) {
BasicBlock *BodyBB = splitBB(Builder, /*CreateBranch=*/true, "alloca.split");
OpenMPIRBuilder::LocationDescription Loc(
InsertPointTy(BodyBB, BodyBB->getFirstInsertionPt()), DL);
- OpenMPIRBuilder::InsertPointOrErrorTy AfterIP = OMPBuilder.createTaskgroup(
- Loc, InsertPointTy(AllocaBB, AllocaBB->getFirstInsertionPt()), BodyGenCB);
- assert(AfterIP && "unexpected error");
- Builder.restoreIP(*AfterIP);
+ OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(OMPBuilder.createTaskgroup(
+ Loc, InsertPointTy(AllocaBB, AllocaBB->getFirstInsertionPt()),
+ BodyGenCB));
+ Builder.restoreIP(AfterIP);
OMPBuilder.finalize();
Builder.CreateRetVoid();
>From a66140fbe5df9b05fcb1d89fce1dbc3d3528ef8a Mon Sep 17 00:00:00 2001
From: Sergio Afonso <safonsof at amd.com>
Date: Wed, 13 Nov 2024 14:24:00 +0000
Subject: [PATCH 2/3] Replace cantFail() uses in tests to ensure they work
properly on release builds
---
.../Frontend/OpenMPIRBuilderTest.cpp | 382 ++++++++++--------
1 file changed, 209 insertions(+), 173 deletions(-)
diff --git a/llvm/unittests/Frontend/OpenMPIRBuilderTest.cpp b/llvm/unittests/Frontend/OpenMPIRBuilderTest.cpp
index dfdb5d69501ed4..4f8e9da5902ff0 100644
--- a/llvm/unittests/Frontend/OpenMPIRBuilderTest.cpp
+++ b/llvm/unittests/Frontend/OpenMPIRBuilderTest.cpp
@@ -236,7 +236,7 @@ class OpenMPIRBuilderTest : public testing::Test {
return Error::success();
};
- CanonicalLoopInfo *Loop = cantFail(
+ CanonicalLoopInfo *Loop = *expectedToOptional(
OMPBuilder.createCanonicalLoop(Loc, LoopBodyGenCB, CastedTripCount));
// Finalize the function.
@@ -344,14 +344,15 @@ TEST_F(OpenMPIRBuilderTest, CreateBarrier) {
IRBuilder<> Builder(BB);
- cantFail(OMPBuilder.createBarrier({IRBuilder<>::InsertPoint()}, OMPD_for));
+ ASSERT_TRUE(expectedToOptional(
+ OMPBuilder.createBarrier({IRBuilder<>::InsertPoint()}, OMPD_for)));
EXPECT_TRUE(M->global_empty());
EXPECT_EQ(M->size(), 1U);
EXPECT_EQ(F->size(), 1U);
EXPECT_EQ(BB->size(), 0U);
OpenMPIRBuilder::LocationDescription Loc({Builder.saveIP()});
- cantFail(OMPBuilder.createBarrier(Loc, OMPD_for));
+ ASSERT_TRUE(expectedToOptional(OMPBuilder.createBarrier(Loc, OMPD_for)));
EXPECT_FALSE(M->global_empty());
EXPECT_EQ(M->size(), 3U);
EXPECT_EQ(F->size(), 1U);
@@ -395,7 +396,7 @@ TEST_F(OpenMPIRBuilderTest, CreateCancel) {
OpenMPIRBuilder::LocationDescription Loc({Builder.saveIP()});
OpenMPIRBuilder::InsertPointTy NewIP =
- cantFail(OMPBuilder.createCancel(Loc, nullptr, OMPD_parallel));
+ *expectedToOptional(OMPBuilder.createCancel(Loc, nullptr, OMPD_parallel));
Builder.restoreIP(NewIP);
EXPECT_FALSE(M->global_empty());
EXPECT_EQ(M->size(), 4U);
@@ -462,8 +463,8 @@ TEST_F(OpenMPIRBuilderTest, CreateCancelIfCond) {
IRBuilder<> Builder(BB);
OpenMPIRBuilder::LocationDescription Loc({Builder.saveIP()});
- OpenMPIRBuilder::InsertPointTy NewIP =
- cantFail(OMPBuilder.createCancel(Loc, Builder.getTrue(), OMPD_parallel));
+ OpenMPIRBuilder::InsertPointTy NewIP = *expectedToOptional(
+ OMPBuilder.createCancel(Loc, Builder.getTrue(), OMPD_parallel));
Builder.restoreIP(NewIP);
EXPECT_FALSE(M->global_empty());
EXPECT_EQ(M->size(), 4U);
@@ -537,7 +538,7 @@ TEST_F(OpenMPIRBuilderTest, CreateCancelBarrier) {
OpenMPIRBuilder::LocationDescription Loc({Builder.saveIP()});
OpenMPIRBuilder::InsertPointTy NewIP =
- cantFail(OMPBuilder.createBarrier(Loc, OMPD_for));
+ *expectedToOptional(OMPBuilder.createBarrier(Loc, OMPD_for));
Builder.restoreIP(NewIP);
EXPECT_FALSE(M->global_empty());
EXPECT_EQ(M->size(), 3U);
@@ -583,7 +584,7 @@ TEST_F(OpenMPIRBuilderTest, DbgLoc) {
IRBuilder<> Builder(BB);
OpenMPIRBuilder::LocationDescription Loc({Builder.saveIP(), DL});
- cantFail(OMPBuilder.createBarrier(Loc, OMPD_for));
+ ASSERT_TRUE(expectedToOptional(OMPBuilder.createBarrier(Loc, OMPD_for)));
CallInst *GTID = dyn_cast<CallInst>(&BB->front());
CallInst *Barrier = dyn_cast<CallInst>(GTID->getNextNode());
EXPECT_EQ(GTID->getDebugLoc(), DL);
@@ -682,9 +683,10 @@ TEST_F(OpenMPIRBuilderTest, ParallelSimpleGPU) {
IRBuilder<>::InsertPoint AllocaIP(&F->getEntryBlock(),
F->getEntryBlock().getFirstInsertionPt());
- OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(OMPBuilder.createParallel(
- Loc, AllocaIP, BodyGenCB, PrivCB, FiniCB, nullptr, nullptr,
- OMP_PROC_BIND_default, false));
+ OpenMPIRBuilder::InsertPointTy AfterIP =
+ *expectedToOptional(OMPBuilder.createParallel(
+ Loc, AllocaIP, BodyGenCB, PrivCB, FiniCB, nullptr, nullptr,
+ OMP_PROC_BIND_default, false));
EXPECT_EQ(NumBodiesGenerated, 1U);
EXPECT_EQ(NumPrivatizedVars, 1U);
@@ -794,9 +796,10 @@ TEST_F(OpenMPIRBuilderTest, ParallelSimple) {
IRBuilder<>::InsertPoint AllocaIP(&F->getEntryBlock(),
F->getEntryBlock().getFirstInsertionPt());
- OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(OMPBuilder.createParallel(
- Loc, AllocaIP, BodyGenCB, PrivCB, FiniCB, nullptr, nullptr,
- OMP_PROC_BIND_default, false));
+ OpenMPIRBuilder::InsertPointTy AfterIP =
+ *expectedToOptional(OMPBuilder.createParallel(
+ Loc, AllocaIP, BodyGenCB, PrivCB, FiniCB, nullptr, nullptr,
+ OMP_PROC_BIND_default, false));
EXPECT_EQ(NumBodiesGenerated, 1U);
EXPECT_EQ(NumPrivatizedVars, 1U);
EXPECT_EQ(NumFinalizationPoints, 1U);
@@ -882,9 +885,10 @@ TEST_F(OpenMPIRBuilderTest, ParallelNested) {
BasicBlock *NewBB = SplitBlock(CGBB, &*CodeGenIP.getPoint());
CGBB->getTerminator()->eraseFromParent();
- OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(OMPBuilder.createParallel(
- InsertPointTy(CGBB, CGBB->end()), AllocaIP, InnerBodyGenCB, PrivCB,
- FiniCB, nullptr, nullptr, OMP_PROC_BIND_default, false));
+ OpenMPIRBuilder::InsertPointTy AfterIP =
+ *expectedToOptional(OMPBuilder.createParallel(
+ InsertPointTy(CGBB, CGBB->end()), AllocaIP, InnerBodyGenCB, PrivCB,
+ FiniCB, nullptr, nullptr, OMP_PROC_BIND_default, false));
Builder.restoreIP(AfterIP);
Builder.CreateBr(NewBB);
@@ -893,9 +897,10 @@ TEST_F(OpenMPIRBuilderTest, ParallelNested) {
IRBuilder<>::InsertPoint AllocaIP(&F->getEntryBlock(),
F->getEntryBlock().getFirstInsertionPt());
- OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(OMPBuilder.createParallel(
- Loc, AllocaIP, OuterBodyGenCB, PrivCB, FiniCB, nullptr, nullptr,
- OMP_PROC_BIND_default, false));
+ OpenMPIRBuilder::InsertPointTy AfterIP =
+ *expectedToOptional(OMPBuilder.createParallel(
+ Loc, AllocaIP, OuterBodyGenCB, PrivCB, FiniCB, nullptr, nullptr,
+ OMP_PROC_BIND_default, false));
EXPECT_EQ(NumInnerBodiesGenerated, 1U);
EXPECT_EQ(NumOuterBodiesGenerated, 1U);
@@ -985,7 +990,7 @@ TEST_F(OpenMPIRBuilderTest, ParallelNested2Inner) {
;
OpenMPIRBuilder::InsertPointTy AfterIP1 =
- cantFail(OMPBuilder.createParallel(
+ *expectedToOptional(OMPBuilder.createParallel(
InsertPointTy(CGBB, CGBB->end()), AllocaIP, InnerBodyGenCB, PrivCB,
FiniCB, nullptr, nullptr, OMP_PROC_BIND_default, false));
@@ -993,7 +998,7 @@ TEST_F(OpenMPIRBuilderTest, ParallelNested2Inner) {
Builder.CreateBr(NewBB1);
OpenMPIRBuilder::InsertPointTy AfterIP2 =
- cantFail(OMPBuilder.createParallel(
+ *expectedToOptional(OMPBuilder.createParallel(
InsertPointTy(NewBB1, NewBB1->end()), AllocaIP, InnerBodyGenCB,
PrivCB, FiniCB, nullptr, nullptr, OMP_PROC_BIND_default, false));
@@ -1004,9 +1009,10 @@ TEST_F(OpenMPIRBuilderTest, ParallelNested2Inner) {
IRBuilder<>::InsertPoint AllocaIP(&F->getEntryBlock(),
F->getEntryBlock().getFirstInsertionPt());
- OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(OMPBuilder.createParallel(
- Loc, AllocaIP, OuterBodyGenCB, PrivCB, FiniCB, nullptr, nullptr,
- OMP_PROC_BIND_default, false));
+ OpenMPIRBuilder::InsertPointTy AfterIP =
+ *expectedToOptional(OMPBuilder.createParallel(
+ Loc, AllocaIP, OuterBodyGenCB, PrivCB, FiniCB, nullptr, nullptr,
+ OMP_PROC_BIND_default, false));
EXPECT_EQ(NumInnerBodiesGenerated, 2U);
EXPECT_EQ(NumOuterBodiesGenerated, 1U);
@@ -1118,7 +1124,7 @@ TEST_F(OpenMPIRBuilderTest, ParallelIfCond) {
IRBuilder<>::InsertPoint AllocaIP(&F->getEntryBlock(),
F->getEntryBlock().getFirstInsertionPt());
- OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(
+ OpenMPIRBuilder::InsertPointTy AfterIP = *expectedToOptional(
OMPBuilder.createParallel(Loc, AllocaIP, BodyGenCB, PrivCB, FiniCB,
Builder.CreateIsNotNull(F->arg_begin()),
nullptr, OMP_PROC_BIND_default, false));
@@ -1182,8 +1188,8 @@ TEST_F(OpenMPIRBuilderTest, ParallelCancelBarrier) {
// Create three barriers, two cancel barriers but only one checked.
Function *CBFn, *BFn;
- OpenMPIRBuilder::InsertPointTy BarrierIP1 =
- cantFail(OMPBuilder.createBarrier(Builder.saveIP(), OMPD_parallel));
+ OpenMPIRBuilder::InsertPointTy BarrierIP1 = *expectedToOptional(
+ OMPBuilder.createBarrier(Builder.saveIP(), OMPD_parallel));
Builder.restoreIP(BarrierIP1);
CBFn = M->getFunction("__kmpc_cancel_barrier");
@@ -1195,7 +1201,7 @@ TEST_F(OpenMPIRBuilderTest, ParallelCancelBarrier) {
ASSERT_EQ(CBFn->user_back()->getNumUses(), 1U);
CheckedBarrier = cast<CallInst>(CBFn->user_back());
- OpenMPIRBuilder::InsertPointTy BarrierIP2 = cantFail(
+ OpenMPIRBuilder::InsertPointTy BarrierIP2 = *expectedToOptional(
OMPBuilder.createBarrier(Builder.saveIP(), OMPD_parallel, true));
Builder.restoreIP(BarrierIP2);
CBFn = M->getFunction("__kmpc_cancel_barrier");
@@ -1208,8 +1214,8 @@ TEST_F(OpenMPIRBuilderTest, ParallelCancelBarrier) {
ASSERT_EQ(BFn->user_back()->getNumUses(), 0U);
OpenMPIRBuilder::InsertPointTy BarrierIP3 =
- cantFail(OMPBuilder.createBarrier(Builder.saveIP(), OMPD_parallel,
- false, false));
+ *expectedToOptional(OMPBuilder.createBarrier(
+ Builder.saveIP(), OMPD_parallel, false, false));
Builder.restoreIP(BarrierIP3);
ASSERT_EQ(CBFn->getNumUses(), 2U);
ASSERT_EQ(BFn->getNumUses(), 1U);
@@ -1240,10 +1246,11 @@ TEST_F(OpenMPIRBuilderTest, ParallelCancelBarrier) {
IRBuilder<>::InsertPoint AllocaIP(&F->getEntryBlock(),
F->getEntryBlock().getFirstInsertionPt());
- OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(OMPBuilder.createParallel(
- Loc, AllocaIP, BODYGENCB_WRAPPER(BodyGenCB), PrivCB, FiniCB,
- Builder.CreateIsNotNull(F->arg_begin()), nullptr, OMP_PROC_BIND_default,
- true));
+ OpenMPIRBuilder::InsertPointTy AfterIP =
+ *expectedToOptional(OMPBuilder.createParallel(
+ Loc, AllocaIP, BODYGENCB_WRAPPER(BodyGenCB), PrivCB, FiniCB,
+ Builder.CreateIsNotNull(F->arg_begin()), nullptr,
+ OMP_PROC_BIND_default, true));
EXPECT_EQ(NumBodiesGenerated, 1U);
EXPECT_EQ(NumPrivatizedVars, 0U);
@@ -1326,9 +1333,10 @@ TEST_F(OpenMPIRBuilderTest, ParallelForwardAsPointers) {
IRBuilder<>::InsertPoint AllocaIP(&F->getEntryBlock(),
F->getEntryBlock().getFirstInsertionPt());
- OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(OMPBuilder.createParallel(
- Loc, AllocaIP, BodyGenCB, PrivCB, FiniCB, nullptr, nullptr,
- OMP_PROC_BIND_default, false));
+ OpenMPIRBuilder::InsertPointTy AfterIP =
+ *expectedToOptional(OMPBuilder.createParallel(
+ Loc, AllocaIP, BodyGenCB, PrivCB, FiniCB, nullptr, nullptr,
+ OMP_PROC_BIND_default, false));
Builder.restoreIP(AfterIP);
Builder.CreateRetVoid();
@@ -1362,8 +1370,8 @@ TEST_F(OpenMPIRBuilderTest, CanonicalLoopSimple) {
return Error::success();
};
- CanonicalLoopInfo *Loop =
- cantFail(OMPBuilder.createCanonicalLoop(Loc, LoopBodyGenCB, TripCount));
+ CanonicalLoopInfo *Loop = *expectedToOptional(
+ OMPBuilder.createCanonicalLoop(Loc, LoopBodyGenCB, TripCount));
Builder.restoreIP(Loop->getAfterIP());
ReturnInst *RetInst = Builder.CreateRetVoid();
@@ -1418,7 +1426,7 @@ TEST_F(OpenMPIRBuilderTest, CanonicalLoopBounds) {
auto LoopBodyGenCB = [&](InsertPointTy CodeGenIP, llvm::Value *LC) {
return Error::success();
};
- CanonicalLoopInfo *Loop = cantFail(
+ CanonicalLoopInfo *Loop = *expectedToOptional(
OMPBuilder.createCanonicalLoop(Loc, LoopBodyGenCB, StartVal, StopVal,
StepVal, IsSigned, InclusiveStop));
Loop->assertOK();
@@ -1515,7 +1523,7 @@ TEST_F(OpenMPIRBuilderTest, CollapseNestedLoops) {
Call = createPrintfCall(Builder, "body i=%d j=%d\\n", {OuterLC, InnerLC});
return Error::success();
};
- InnerLoop = cantFail(OMPBuilder.createCanonicalLoop(
+ InnerLoop = *expectedToOptional(OMPBuilder.createCanonicalLoop(
Builder.saveIP(), InnerLoopBodyGenCB, InnerTripCount, "inner"));
Builder.restoreIP(InnerLoop->getAfterIP());
@@ -1523,8 +1531,9 @@ TEST_F(OpenMPIRBuilderTest, CollapseNestedLoops) {
createPrintfCall(Builder, "In-between trail i=%d\\n", {OuterLC});
return Error::success();
};
- CanonicalLoopInfo *OuterLoop = cantFail(OMPBuilder.createCanonicalLoop(
- OuterLoc, OuterLoopBodyGenCB, OuterTripCount, "outer"));
+ CanonicalLoopInfo *OuterLoop =
+ *expectedToOptional(OMPBuilder.createCanonicalLoop(
+ OuterLoc, OuterLoopBodyGenCB, OuterTripCount, "outer"));
// Finish the function.
Builder.restoreIP(OuterLoop->getAfterIP());
@@ -1636,12 +1645,13 @@ TEST_F(OpenMPIRBuilderTest, TileNestedLoops) {
createPrintfCall(Builder, "i=%d j=%d\\n", {OuterLC, InnerLC});
return Error::success();
};
- InnerLoop = cantFail(OMPBuilder.createCanonicalLoop(
+ InnerLoop = *expectedToOptional(OMPBuilder.createCanonicalLoop(
OuterCodeGenIP, InnerLoopBodyGenCB, TripCount, "inner"));
return Error::success();
};
- CanonicalLoopInfo *OuterLoop = cantFail(OMPBuilder.createCanonicalLoop(
- Loc, OuterLoopBodyGenCB, TripCount, "outer"));
+ CanonicalLoopInfo *OuterLoop =
+ *expectedToOptional(OMPBuilder.createCanonicalLoop(
+ Loc, OuterLoopBodyGenCB, TripCount, "outer"));
// Finalize the function.
Builder.restoreIP(OuterLoop->getAfterIP());
@@ -1738,14 +1748,15 @@ TEST_F(OpenMPIRBuilderTest, TileNestedLoopsWithBounds) {
Call = createPrintfCall(Builder, "i=%d j=%d\\n", {OuterLC, InnerLC});
return Error::success();
};
- InnerLoop = cantFail(OMPBuilder.createCanonicalLoop(
+ InnerLoop = *expectedToOptional(OMPBuilder.createCanonicalLoop(
OuterCodeGenIP, InnerLoopBodyGenCB, InnerStartVal, InnerStopVal,
InnerStep, false, false, ComputeIP, "inner"));
return Error::success();
};
- CanonicalLoopInfo *OuterLoop = cantFail(OMPBuilder.createCanonicalLoop(
- Loc, OuterLoopBodyGenCB, OuterStartVal, OuterStopVal, OuterStep, false,
- false, ComputeIP, "outer"));
+ CanonicalLoopInfo *OuterLoop =
+ *expectedToOptional(OMPBuilder.createCanonicalLoop(
+ Loc, OuterLoopBodyGenCB, OuterStartVal, OuterStopVal, OuterStep,
+ false, false, ComputeIP, "outer"));
// Finalize the function
Builder.restoreIP(OuterLoop->getAfterIP());
@@ -1852,7 +1863,7 @@ TEST_F(OpenMPIRBuilderTest, TileSingleLoopCounts) {
auto LoopBodyGenCB = [&](InsertPointTy CodeGenIP, llvm::Value *LC) {
return Error::success();
};
- CanonicalLoopInfo *Loop = cantFail(
+ CanonicalLoopInfo *Loop = *expectedToOptional(
OMPBuilder.createCanonicalLoop(Loc, LoopBodyGenCB, StartVal, StopVal,
StepVal, IsSigned, InclusiveStop));
InsertPointTy AfterIP = Loop->getAfterIP();
@@ -2305,7 +2316,7 @@ TEST_F(OpenMPIRBuilderTest, StaticWorkshareLoopTarget) {
Value *StepVal = ConstantInt::get(LCTy, 2);
auto LoopBodyGen = [&](InsertPointTy, Value *) { return Error::success(); };
- CanonicalLoopInfo *CLI = cantFail(OMPBuilder.createCanonicalLoop(
+ CanonicalLoopInfo *CLI = *expectedToOptional(OMPBuilder.createCanonicalLoop(
Loc, LoopBodyGen, StartVal, StopVal, StepVal, false, false));
BasicBlock *Preheader = CLI->getPreheader();
Value *TripCount = CLI->getTripCount();
@@ -2313,7 +2324,7 @@ TEST_F(OpenMPIRBuilderTest, StaticWorkshareLoopTarget) {
Builder.SetInsertPoint(BB, BB->getFirstInsertionPt());
OpenMPIRBuilder::InsertPointTy AfterIP =
- cantFail(OMPBuilder.applyWorkshareLoop(
+ *expectedToOptional(OMPBuilder.applyWorkshareLoop(
DL, CLI, AllocaIP, true, OMP_SCHEDULE_Static, nullptr, false, false,
false, false, WorksharingLoopType::ForStaticLoop));
Builder.restoreIP(AfterIP);
@@ -2369,7 +2380,7 @@ TEST_F(OpenMPIRBuilderTest, StaticWorkShareLoop) {
return Error::success();
};
- CanonicalLoopInfo *CLI = cantFail(OMPBuilder.createCanonicalLoop(
+ CanonicalLoopInfo *CLI = *expectedToOptional(OMPBuilder.createCanonicalLoop(
Loc, LoopBodyGen, StartVal, StopVal, StepVal,
/*IsSigned=*/false, /*InclusiveStop=*/false));
BasicBlock *Preheader = CLI->getPreheader();
@@ -2380,8 +2391,8 @@ TEST_F(OpenMPIRBuilderTest, StaticWorkShareLoop) {
Builder.SetInsertPoint(BB, BB->getFirstInsertionPt());
InsertPointTy AllocaIP = Builder.saveIP();
- cantFail(OMPBuilder.applyWorkshareLoop(
- DL, CLI, AllocaIP, /*NeedsBarrier=*/true, OMP_SCHEDULE_Static));
+ ASSERT_TRUE(expectedToOptional(OMPBuilder.applyWorkshareLoop(
+ DL, CLI, AllocaIP, /*NeedsBarrier=*/true, OMP_SCHEDULE_Static)));
BasicBlock *Cond = Body->getSinglePredecessor();
Instruction *Cmp = &*Cond->begin();
@@ -2473,9 +2484,9 @@ TEST_P(OpenMPIRBuilderTestWithIVBits, StaticChunkedWorkshareLoop) {
Value *ChunkSize = ConstantInt::get(LCTy, 5);
InsertPointTy AllocaIP{&F->getEntryBlock(),
F->getEntryBlock().getFirstInsertionPt()};
- cantFail(OMPBuilder.applyWorkshareLoop(DL, CLI, AllocaIP,
- /*NeedsBarrier=*/true,
- OMP_SCHEDULE_Static, ChunkSize));
+ ASSERT_TRUE(expectedToOptional(OMPBuilder.applyWorkshareLoop(
+ DL, CLI, AllocaIP,
+ /*NeedsBarrier=*/true, OMP_SCHEDULE_Static, ChunkSize)));
OMPBuilder.finalize();
EXPECT_FALSE(verifyModule(*M, &errs()));
@@ -2566,7 +2577,7 @@ TEST_P(OpenMPIRBuilderTestWithParams, DynamicWorkShareLoop) {
return Error::success();
};
- CanonicalLoopInfo *CLI = cantFail(OMPBuilder.createCanonicalLoop(
+ CanonicalLoopInfo *CLI = *expectedToOptional(OMPBuilder.createCanonicalLoop(
Loc, LoopBodyGen, StartVal, StopVal, StepVal,
/*IsSigned=*/false, /*InclusiveStop=*/false));
@@ -2581,14 +2592,15 @@ TEST_P(OpenMPIRBuilderTestWithParams, DynamicWorkShareLoop) {
BasicBlock *LatchBlock = CLI->getLatch();
Value *IV = CLI->getIndVar();
- OpenMPIRBuilder::InsertPointTy EndIP = cantFail(OMPBuilder.applyWorkshareLoop(
- DL, CLI, AllocaIP, /*NeedsBarrier=*/true, getSchedKind(SchedType),
- ChunkVal, /*Simd=*/false,
- (SchedType & omp::OMPScheduleType::ModifierMonotonic) ==
- omp::OMPScheduleType::ModifierMonotonic,
- (SchedType & omp::OMPScheduleType::ModifierNonmonotonic) ==
- omp::OMPScheduleType::ModifierNonmonotonic,
- /*Ordered=*/false));
+ OpenMPIRBuilder::InsertPointTy EndIP =
+ *expectedToOptional(OMPBuilder.applyWorkshareLoop(
+ DL, CLI, AllocaIP, /*NeedsBarrier=*/true, getSchedKind(SchedType),
+ ChunkVal, /*Simd=*/false,
+ (SchedType & omp::OMPScheduleType::ModifierMonotonic) ==
+ omp::OMPScheduleType::ModifierMonotonic,
+ (SchedType & omp::OMPScheduleType::ModifierNonmonotonic) ==
+ omp::OMPScheduleType::ModifierNonmonotonic,
+ /*Ordered=*/false));
// The returned value should be the "after" point.
ASSERT_EQ(EndIP.getBlock(), AfterIP.getBlock());
@@ -2710,7 +2722,7 @@ TEST_F(OpenMPIRBuilderTest, DynamicWorkShareLoopOrdered) {
return llvm::Error::success();
};
- CanonicalLoopInfo *CLI = cantFail(OMPBuilder.createCanonicalLoop(
+ CanonicalLoopInfo *CLI = *expectedToOptional(OMPBuilder.createCanonicalLoop(
Loc, LoopBodyGen, StartVal, StopVal, StepVal,
/*IsSigned=*/false, /*InclusiveStop=*/false));
@@ -2724,11 +2736,13 @@ TEST_F(OpenMPIRBuilderTest, DynamicWorkShareLoopOrdered) {
BasicBlock *LatchBlock = CLI->getLatch();
Value *IV = CLI->getIndVar();
- OpenMPIRBuilder::InsertPointTy EndIP = cantFail(OMPBuilder.applyWorkshareLoop(
- DL, CLI, AllocaIP, /*NeedsBarrier=*/true, OMP_SCHEDULE_Static, ChunkVal,
- /*HasSimdModifier=*/false, /*HasMonotonicModifier=*/false,
- /*HasNonmonotonicModifier=*/false,
- /*HasOrderedClause=*/true));
+ OpenMPIRBuilder::InsertPointTy EndIP =
+ *expectedToOptional(OMPBuilder.applyWorkshareLoop(
+ DL, CLI, AllocaIP, /*NeedsBarrier=*/true, OMP_SCHEDULE_Static,
+ ChunkVal,
+ /*HasSimdModifier=*/false, /*HasMonotonicModifier=*/false,
+ /*HasNonmonotonicModifier=*/false,
+ /*HasOrderedClause=*/true));
// Add a termination to our block and check that it is internally consistent.
Builder.restoreIP(EndIP);
@@ -2815,8 +2829,9 @@ TEST_F(OpenMPIRBuilderTest, MasterDirective) {
EXPECT_NE(IPBB->end(), IP.getPoint());
};
- OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(OMPBuilder.createMaster(
- Builder, BODYGENCB_WRAPPER(BodyGenCB), FINICB_WRAPPER(FiniCB)));
+ OpenMPIRBuilder::InsertPointTy AfterIP =
+ *expectedToOptional(OMPBuilder.createMaster(
+ Builder, BODYGENCB_WRAPPER(BodyGenCB), FINICB_WRAPPER(FiniCB)));
Builder.restoreIP(AfterIP);
Value *EntryBBTI = EntryBB->getTerminator();
EXPECT_NE(EntryBBTI, nullptr);
@@ -2895,8 +2910,9 @@ TEST_F(OpenMPIRBuilderTest, MaskedDirective) {
};
Constant *Filter = ConstantInt::get(Type::getInt32Ty(M->getContext()), 0);
- OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(OMPBuilder.createMasked(
- Builder, BODYGENCB_WRAPPER(BodyGenCB), FINICB_WRAPPER(FiniCB), Filter));
+ OpenMPIRBuilder::InsertPointTy AfterIP = *expectedToOptional(
+ OMPBuilder.createMasked(Builder, BODYGENCB_WRAPPER(BodyGenCB),
+ FINICB_WRAPPER(FiniCB), Filter));
Builder.restoreIP(AfterIP);
Value *EntryBBTI = EntryBB->getTerminator();
EXPECT_NE(EntryBBTI, nullptr);
@@ -2962,7 +2978,7 @@ TEST_F(OpenMPIRBuilderTest, CriticalDirective) {
};
BasicBlock *EntryBB = Builder.GetInsertBlock();
- OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(
+ OpenMPIRBuilder::InsertPointTy AfterIP = *expectedToOptional(
OMPBuilder.createCritical(Builder, BODYGENCB_WRAPPER(BodyGenCB),
FINICB_WRAPPER(FiniCB), "testCRT", nullptr));
Builder.restoreIP(AfterIP);
@@ -3213,7 +3229,7 @@ TEST_F(OpenMPIRBuilderTest, OrderedDirectiveThreads) {
// Test for "#omp ordered [threads]"
BasicBlock *EntryBB = Builder.GetInsertBlock();
OpenMPIRBuilder::InsertPointTy AfterIP =
- cantFail(OMPBuilder.createOrderedThreadsSimd(
+ *expectedToOptional(OMPBuilder.createOrderedThreadsSimd(
Builder, BODYGENCB_WRAPPER(BodyGenCB), FINICB_WRAPPER(FiniCB), true));
Builder.restoreIP(AfterIP);
@@ -3285,7 +3301,7 @@ TEST_F(OpenMPIRBuilderTest, OrderedDirectiveSimd) {
// Test for "#omp ordered simd"
BasicBlock *EntryBB = Builder.GetInsertBlock();
- OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(
+ OpenMPIRBuilder::InsertPointTy AfterIP = *expectedToOptional(
OMPBuilder.createOrderedThreadsSimd(Builder, BODYGENCB_WRAPPER(BodyGenCB),
FINICB_WRAPPER(FiniCB), false));
Builder.restoreIP(AfterIP);
@@ -3401,7 +3417,7 @@ TEST_F(OpenMPIRBuilderTest, SingleDirective) {
EXPECT_NE(IPBB->end(), IP.getPoint());
};
- OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(
+ OpenMPIRBuilder::InsertPointTy AfterIP = *expectedToOptional(
OMPBuilder.createSingle(Builder, BODYGENCB_WRAPPER(BodyGenCB),
FINICB_WRAPPER(FiniCB), /*IsNowait*/ false));
Builder.restoreIP(AfterIP);
@@ -3493,7 +3509,7 @@ TEST_F(OpenMPIRBuilderTest, SingleDirectiveNowait) {
EXPECT_NE(IPBB->end(), IP.getPoint());
};
- OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(
+ OpenMPIRBuilder::InsertPointTy AfterIP = *expectedToOptional(
OMPBuilder.createSingle(Builder, BODYGENCB_WRAPPER(BodyGenCB),
FINICB_WRAPPER(FiniCB), /*IsNowait*/ true));
Builder.restoreIP(AfterIP);
@@ -3614,9 +3630,10 @@ TEST_F(OpenMPIRBuilderTest, SingleDirectiveCopyPrivate) {
EXPECT_NE(IPBB->end(), IP.getPoint());
};
- OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(OMPBuilder.createSingle(
- Builder, BODYGENCB_WRAPPER(BodyGenCB), FINICB_WRAPPER(FiniCB),
- /*IsNowait*/ false, {CPVar}, {CopyFunc}));
+ OpenMPIRBuilder::InsertPointTy AfterIP =
+ *expectedToOptional(OMPBuilder.createSingle(
+ Builder, BODYGENCB_WRAPPER(BodyGenCB), FINICB_WRAPPER(FiniCB),
+ /*IsNowait*/ false, {CPVar}, {CopyFunc}));
Builder.restoreIP(AfterIP);
Value *EntryBBTI = EntryBB->getTerminator();
EXPECT_NE(EntryBBTI, nullptr);
@@ -3879,8 +3896,8 @@ TEST_F(OpenMPIRBuilderTest, OMPAtomicUpdate) {
return Sub;
};
OpenMPIRBuilder::InsertPointTy AfterIP =
- cantFail(OMPBuilder.createAtomicUpdate(Builder, AllocaIP, X, Expr, AO,
- RMWOp, UpdateOp, IsXLHSInRHSPart));
+ *expectedToOptional(OMPBuilder.createAtomicUpdate(
+ Builder, AllocaIP, X, Expr, AO, RMWOp, UpdateOp, IsXLHSInRHSPart));
Builder.restoreIP(AfterIP);
BasicBlock *ContBB = EntryBB->getSingleSuccessor();
BranchInst *ContTI = dyn_cast<BranchInst>(ContBB->getTerminator());
@@ -3948,8 +3965,8 @@ TEST_F(OpenMPIRBuilderTest, OMPAtomicUpdateFloat) {
return Sub;
};
OpenMPIRBuilder::InsertPointTy AfterIP =
- cantFail(OMPBuilder.createAtomicUpdate(Builder, AllocaIP, X, Expr, AO,
- RMWOp, UpdateOp, IsXLHSInRHSPart));
+ *expectedToOptional(OMPBuilder.createAtomicUpdate(
+ Builder, AllocaIP, X, Expr, AO, RMWOp, UpdateOp, IsXLHSInRHSPart));
Builder.restoreIP(AfterIP);
BasicBlock *ContBB = EntryBB->getSingleSuccessor();
BranchInst *ContTI = dyn_cast<BranchInst>(ContBB->getTerminator());
@@ -4016,8 +4033,8 @@ TEST_F(OpenMPIRBuilderTest, OMPAtomicUpdateIntr) {
return Sub;
};
OpenMPIRBuilder::InsertPointTy AfterIP =
- cantFail(OMPBuilder.createAtomicUpdate(Builder, AllocaIP, X, Expr, AO,
- RMWOp, UpdateOp, IsXLHSInRHSPart));
+ *expectedToOptional(OMPBuilder.createAtomicUpdate(
+ Builder, AllocaIP, X, Expr, AO, RMWOp, UpdateOp, IsXLHSInRHSPart));
Builder.restoreIP(AfterIP);
BasicBlock *ContBB = EntryBB->getSingleSuccessor();
BranchInst *ContTI = dyn_cast<BranchInst>(ContBB->getTerminator());
@@ -4090,7 +4107,7 @@ TEST_F(OpenMPIRBuilderTest, OMPAtomicCapture) {
auto UpdateOp = [&](Value *Atomic, IRBuilder<> &IRB) { return nullptr; };
OpenMPIRBuilder::InsertPointTy AfterIP =
- cantFail(OMPBuilder.createAtomicCapture(
+ *expectedToOptional(OMPBuilder.createAtomicCapture(
Builder, AllocaIP, X, V, Expr, AO, RMWOp, UpdateOp, UpdateExpr,
IsPostfixUpdate, IsXLHSInRHSPart));
Builder.restoreIP(AfterIP);
@@ -4453,9 +4470,10 @@ TEST_F(OpenMPIRBuilderTest, CreateTeams) {
};
OpenMPIRBuilder::LocationDescription Loc({Builder.saveIP(), DL});
- OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(OMPBuilder.createTeams(
- Builder, BodyGenCB, /*NumTeamsLower=*/nullptr, /*NumTeamsUpper=*/nullptr,
- /*ThreadLimit=*/nullptr, /*IfExpr=*/nullptr));
+ OpenMPIRBuilder::InsertPointTy AfterIP = *expectedToOptional(
+ OMPBuilder.createTeams(Builder, BodyGenCB, /*NumTeamsLower=*/nullptr,
+ /*NumTeamsUpper=*/nullptr,
+ /*ThreadLimit=*/nullptr, /*IfExpr=*/nullptr));
Builder.restoreIP(AfterIP);
OMPBuilder.finalize();
@@ -4517,10 +4535,11 @@ TEST_F(OpenMPIRBuilderTest, CreateTeamsWithThreadLimit) {
};
// `F` has an argument - an integer, so we use that as the thread limit.
- OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(OMPBuilder.createTeams(
- /*=*/Builder, BodyGenCB, /*NumTeamsLower=*/nullptr,
- /*NumTeamsUpper=*/nullptr, /*ThreadLimit=*/F->arg_begin(),
- /*IfExpr=*/nullptr));
+ OpenMPIRBuilder::InsertPointTy AfterIP =
+ *expectedToOptional(OMPBuilder.createTeams(
+ /*=*/Builder, BodyGenCB, /*NumTeamsLower=*/nullptr,
+ /*NumTeamsUpper=*/nullptr, /*ThreadLimit=*/F->arg_begin(),
+ /*IfExpr=*/nullptr));
Builder.restoreIP(AfterIP);
Builder.CreateRetVoid();
@@ -4570,12 +4589,12 @@ TEST_F(OpenMPIRBuilderTest, CreateTeamsWithNumTeamsUpper) {
// `F` already has an integer argument, so we use that as upper bound to
// `num_teams`
- OpenMPIRBuilder::InsertPointTy AfterIP =
- cantFail(OMPBuilder.createTeams(Builder, BodyGenCB,
- /*NumTeamsLower=*/nullptr,
- /*NumTeamsUpper=*/F->arg_begin(),
- /*ThreadLimit=*/nullptr,
- /*IfExpr=*/nullptr));
+ OpenMPIRBuilder::InsertPointTy AfterIP = *expectedToOptional(
+ OMPBuilder.createTeams(Builder, BodyGenCB,
+ /*NumTeamsLower=*/nullptr,
+ /*NumTeamsUpper=*/F->arg_begin(),
+ /*ThreadLimit=*/nullptr,
+ /*IfExpr=*/nullptr));
Builder.restoreIP(AfterIP);
Builder.CreateRetVoid();
@@ -4630,7 +4649,7 @@ TEST_F(OpenMPIRBuilderTest, CreateTeamsWithNumTeamsBoth) {
// `F` already has an integer argument, so we use that as upper bound to
// `num_teams`
- OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(
+ OpenMPIRBuilder::InsertPointTy AfterIP = *expectedToOptional(
OMPBuilder.createTeams(Builder, BodyGenCB, NumTeamsLower, NumTeamsUpper,
/*ThreadLimit=*/nullptr, /*IfExpr=*/nullptr));
Builder.restoreIP(AfterIP);
@@ -4693,8 +4712,9 @@ TEST_F(OpenMPIRBuilderTest, CreateTeamsWithNumTeamsAndThreadLimit) {
};
OpenMPIRBuilder::LocationDescription Loc({Builder.saveIP(), DL});
- OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(OMPBuilder.createTeams(
- Builder, BodyGenCB, NumTeamsLower, NumTeamsUpper, ThreadLimit, nullptr));
+ OpenMPIRBuilder::InsertPointTy AfterIP = *expectedToOptional(
+ OMPBuilder.createTeams(Builder, BodyGenCB, NumTeamsLower, NumTeamsUpper,
+ ThreadLimit, nullptr));
Builder.restoreIP(AfterIP);
Builder.CreateRetVoid();
@@ -4747,9 +4767,10 @@ TEST_F(OpenMPIRBuilderTest, CreateTeamsWithIfCondition) {
// `F` already has an integer argument, so we use that as upper bound to
// `num_teams`
- OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(OMPBuilder.createTeams(
- Builder, BodyGenCB, /*NumTeamsLower=*/nullptr, /*NumTeamsUpper=*/nullptr,
- /*ThreadLimit=*/nullptr, IfExpr));
+ OpenMPIRBuilder::InsertPointTy AfterIP = *expectedToOptional(
+ OMPBuilder.createTeams(Builder, BodyGenCB, /*NumTeamsLower=*/nullptr,
+ /*NumTeamsUpper=*/nullptr,
+ /*ThreadLimit=*/nullptr, IfExpr));
Builder.restoreIP(AfterIP);
Builder.CreateRetVoid();
@@ -4812,8 +4833,9 @@ TEST_F(OpenMPIRBuilderTest, CreateTeamsWithIfConditionAndNumTeams) {
// `F` already has an integer argument, so we use that as upper bound to
// `num_teams`
- OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(OMPBuilder.createTeams(
- Builder, BodyGenCB, NumTeamsLower, NumTeamsUpper, ThreadLimit, IfExpr));
+ OpenMPIRBuilder::InsertPointTy AfterIP = *expectedToOptional(
+ OMPBuilder.createTeams(Builder, BodyGenCB, NumTeamsLower, NumTeamsUpper,
+ ThreadLimit, IfExpr));
Builder.restoreIP(AfterIP);
Builder.CreateRetVoid();
@@ -5074,7 +5096,7 @@ TEST_F(OpenMPIRBuilderTest, CreateReductions) {
// Do nothing in finalization.
auto FiniCB = [&](InsertPointTy CodeGenIP) { return Error::success(); };
- OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(
+ OpenMPIRBuilder::InsertPointTy AfterIP = *expectedToOptional(
OMPBuilder.createParallel(Loc, OuterAllocaIP, BodyGenCB, PrivCB, FiniCB,
/* IfCondition */ nullptr,
/* NumThreads */ nullptr, OMP_PROC_BIND_default,
@@ -5091,8 +5113,8 @@ TEST_F(OpenMPIRBuilderTest, CreateReductions) {
OMPBuilder.Config.setIsGPU(false);
bool ReduceVariableByRef[] = {false, false};
- cantFail(OMPBuilder.createReductions(BodyIP, BodyAllocaIP, ReductionInfos,
- ReduceVariableByRef));
+ ASSERT_TRUE(expectedToOptional(OMPBuilder.createReductions(
+ BodyIP, BodyAllocaIP, ReductionInfos, ReduceVariableByRef)));
Builder.restoreIP(AfterIP);
Builder.CreateRetVoid();
@@ -5330,34 +5352,35 @@ TEST_F(OpenMPIRBuilderTest, CreateTwoReductions) {
// Do nothing in finalization.
auto FiniCB = [&](InsertPointTy CodeGenIP) { return Error::success(); };
- OpenMPIRBuilder::InsertPointTy AfterIP1 = cantFail(
+ OpenMPIRBuilder::InsertPointTy AfterIP1 = *expectedToOptional(
OMPBuilder.createParallel(Loc, OuterAllocaIP, FirstBodyGenCB, PrivCB,
FiniCB, /* IfCondition */ nullptr,
/* NumThreads */ nullptr, OMP_PROC_BIND_default,
/* IsCancellable */ false));
Builder.restoreIP(AfterIP1);
- OpenMPIRBuilder::InsertPointTy AfterIP2 = cantFail(OMPBuilder.createParallel(
- {Builder.saveIP(), DL}, OuterAllocaIP, SecondBodyGenCB, PrivCB, FiniCB,
- /* IfCondition */ nullptr,
- /* NumThreads */ nullptr, OMP_PROC_BIND_default,
- /* IsCancellable */ false));
+ OpenMPIRBuilder::InsertPointTy AfterIP2 = *expectedToOptional(
+ OMPBuilder.createParallel({Builder.saveIP(), DL}, OuterAllocaIP,
+ SecondBodyGenCB, PrivCB, FiniCB,
+ /* IfCondition */ nullptr,
+ /* NumThreads */ nullptr, OMP_PROC_BIND_default,
+ /* IsCancellable */ false));
Builder.restoreIP(AfterIP2);
OMPBuilder.Config.setIsGPU(false);
bool ReduceVariableByRef[] = {false};
- cantFail(OMPBuilder.createReductions(
+ ASSERT_TRUE(expectedToOptional(OMPBuilder.createReductions(
FirstBodyIP, FirstBodyAllocaIP,
{{SumType, SumReduced, SumPrivatized,
/*EvaluationKind=*/OpenMPIRBuilder::EvalKind::Scalar, sumReduction,
/*ReductionGenClang=*/nullptr, sumAtomicReduction}},
- ReduceVariableByRef));
- cantFail(OMPBuilder.createReductions(
+ ReduceVariableByRef)));
+ ASSERT_TRUE(expectedToOptional(OMPBuilder.createReductions(
SecondBodyIP, SecondBodyAllocaIP,
{{XorType, XorReduced, XorPrivatized,
/*EvaluationKind=*/OpenMPIRBuilder::EvalKind::Scalar, xorReduction,
/*ReductionGenClang=*/nullptr, xorAtomicReduction}},
- ReduceVariableByRef));
+ ReduceVariableByRef)));
Builder.restoreIP(AfterIP2);
Builder.CreateRetVoid();
@@ -5437,8 +5460,9 @@ TEST_F(OpenMPIRBuilderTest, CreateSectionsSimple) {
llvm::Value *&ReplVal) { return CodeGenIP; };
IRBuilder<>::InsertPoint AllocaIP(&F->getEntryBlock(),
F->getEntryBlock().getFirstInsertionPt());
- OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(OMPBuilder.createSections(
- Loc, AllocaIP, SectionCBVector, PrivCB, FiniCB, false, false));
+ OpenMPIRBuilder::InsertPointTy AfterIP =
+ *expectedToOptional(OMPBuilder.createSections(
+ Loc, AllocaIP, SectionCBVector, PrivCB, FiniCB, false, false));
Builder.restoreIP(AfterIP);
Builder.CreateRetVoid(); // Required at the end of the function
EXPECT_NE(F->getEntryBlock().getTerminator(), nullptr);
@@ -5493,9 +5517,9 @@ TEST_F(OpenMPIRBuilderTest, CreateSections) {
IRBuilder<>::InsertPoint AllocaIP(&F->getEntryBlock(),
F->getEntryBlock().getFirstInsertionPt());
- OpenMPIRBuilder::InsertPointTy AfterIP =
- cantFail(OMPBuilder.createSections(Loc, AllocaIP, SectionCBVector, PrivCB,
- FINICB_WRAPPER(FiniCB), false, false));
+ OpenMPIRBuilder::InsertPointTy AfterIP = *expectedToOptional(
+ OMPBuilder.createSections(Loc, AllocaIP, SectionCBVector, PrivCB,
+ FINICB_WRAPPER(FiniCB), false, false));
Builder.restoreIP(AfterIP);
Builder.CreateRetVoid(); // Required at the end of the function
@@ -5582,8 +5606,9 @@ TEST_F(OpenMPIRBuilderTest, CreateSectionsNoWait) {
llvm::Value *&ReplVal) { return CodeGenIP; };
auto FiniCB = [&](InsertPointTy IP) { return Error::success(); };
- OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(OMPBuilder.createSections(
- Loc, AllocaIP, SectionCBVector, PrivCB, FiniCB, false, true));
+ OpenMPIRBuilder::InsertPointTy AfterIP =
+ *expectedToOptional(OMPBuilder.createSections(
+ Loc, AllocaIP, SectionCBVector, PrivCB, FiniCB, false, true));
Builder.restoreIP(AfterIP);
Builder.CreateRetVoid(); // Required at the end of the function
for (auto &Inst : instructions(*F)) {
@@ -5805,9 +5830,10 @@ TEST_F(OpenMPIRBuilderTest, TargetEnterData) {
OMPBuilder.Config.setIsGPU(true);
llvm::omp::RuntimeFunction RTLFunc = OMPRTL___tgt_target_data_begin_mapper;
- OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(OMPBuilder.createTargetData(
- Loc, AllocaIP, Builder.saveIP(), Builder.getInt64(DeviceID),
- /* IfCond= */ nullptr, Info, GenMapInfoCB, &RTLFunc));
+ OpenMPIRBuilder::InsertPointTy AfterIP =
+ *expectedToOptional(OMPBuilder.createTargetData(
+ Loc, AllocaIP, Builder.saveIP(), Builder.getInt64(DeviceID),
+ /* IfCond= */ nullptr, Info, GenMapInfoCB, &RTLFunc));
Builder.restoreIP(AfterIP);
CallInst *TargetDataCall = dyn_cast<CallInst>(&BB->back());
@@ -5865,9 +5891,10 @@ TEST_F(OpenMPIRBuilderTest, TargetExitData) {
OMPBuilder.Config.setIsGPU(true);
llvm::omp::RuntimeFunction RTLFunc = OMPRTL___tgt_target_data_end_mapper;
- OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(OMPBuilder.createTargetData(
- Loc, AllocaIP, Builder.saveIP(), Builder.getInt64(DeviceID),
- /* IfCond= */ nullptr, Info, GenMapInfoCB, &RTLFunc));
+ OpenMPIRBuilder::InsertPointTy AfterIP =
+ *expectedToOptional(OMPBuilder.createTargetData(
+ Loc, AllocaIP, Builder.saveIP(), Builder.getInt64(DeviceID),
+ /* IfCond= */ nullptr, Info, GenMapInfoCB, &RTLFunc));
Builder.restoreIP(AfterIP);
CallInst *TargetDataCall = dyn_cast<CallInst>(&BB->back());
@@ -5975,7 +6002,7 @@ TEST_F(OpenMPIRBuilderTest, TargetDataRegion) {
};
OpenMPIRBuilder::InsertPointTy TargetDataIP1 =
- cantFail(OMPBuilder.createTargetData(
+ *expectedToOptional(OMPBuilder.createTargetData(
Loc, AllocaIP, Builder.saveIP(), Builder.getInt64(DeviceID),
/* IfCond= */ nullptr, Info, GenMapInfoCB, nullptr, BodyCB));
Builder.restoreIP(TargetDataIP1);
@@ -6002,7 +6029,7 @@ TEST_F(OpenMPIRBuilderTest, TargetDataRegion) {
return Builder.saveIP();
};
OpenMPIRBuilder::InsertPointTy TargetDataIP2 =
- cantFail(OMPBuilder.createTargetData(
+ *expectedToOptional(OMPBuilder.createTargetData(
Loc, AllocaIP, Builder.saveIP(), Builder.getInt64(DeviceID),
/* IfCond= */ nullptr, Info, GenMapInfoCB, nullptr, BodyTargetCB));
Builder.restoreIP(TargetDataIP2);
@@ -6100,9 +6127,10 @@ TEST_F(OpenMPIRBuilderTest, TargetRegion) {
TargetRegionEntryInfo EntryInfo("func", 42, 4711, 17);
OpenMPIRBuilder::LocationDescription OmpLoc({Builder.saveIP(), DL});
- OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(OMPBuilder.createTarget(
- OmpLoc, /*IsOffloadEntry=*/true, Builder.saveIP(), Builder.saveIP(),
- EntryInfo, -1, 0, Inputs, GenMapInfoCB, BodyGenCB, SimpleArgAccessorCB));
+ OpenMPIRBuilder::InsertPointTy AfterIP = *expectedToOptional(
+ OMPBuilder.createTarget(OmpLoc, /*IsOffloadEntry=*/true, Builder.saveIP(),
+ Builder.saveIP(), EntryInfo, -1, 0, Inputs,
+ GenMapInfoCB, BodyGenCB, SimpleArgAccessorCB));
Builder.restoreIP(AfterIP);
OMPBuilder.finalize();
Builder.CreateRetVoid();
@@ -6209,7 +6237,7 @@ TEST_F(OpenMPIRBuilderTest, TargetRegionDevice) {
TargetRegionEntryInfo EntryInfo("parent", /*DeviceID=*/1, /*FileID=*/2,
/*Line=*/3, /*Count=*/0);
- OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(
+ OpenMPIRBuilder::InsertPointTy AfterIP = *expectedToOptional(
OMPBuilder.createTarget(Loc, /*IsOffloadEntry=*/true, EntryIP, EntryIP,
EntryInfo, /*NumTeams=*/-1,
/*NumThreads=*/0, CapturedArgs, GenMapInfoCB,
@@ -6359,7 +6387,7 @@ TEST_F(OpenMPIRBuilderTest, ConstantAllocaRaise) {
TargetRegionEntryInfo EntryInfo("parent", /*DeviceID=*/1, /*FileID=*/2,
/*Line=*/3, /*Count=*/0);
- OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(
+ OpenMPIRBuilder::InsertPointTy AfterIP = *expectedToOptional(
OMPBuilder.createTarget(Loc, /*IsOffloadEntry=*/true, EntryIP, EntryIP,
EntryInfo, /*NumTeams=*/-1,
/*NumThreads=*/0, CapturedArgs, GenMapInfoCB,
@@ -6483,9 +6511,10 @@ TEST_F(OpenMPIRBuilderTest, CreateTask) {
BasicBlock *BodyBB = splitBB(Builder, /*CreateBranch=*/true, "alloca.split");
OpenMPIRBuilder::LocationDescription Loc(
InsertPointTy(BodyBB, BodyBB->getFirstInsertionPt()), DL);
- OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(OMPBuilder.createTask(
- Loc, InsertPointTy(AllocaBB, AllocaBB->getFirstInsertionPt()),
- BodyGenCB));
+ OpenMPIRBuilder::InsertPointTy AfterIP =
+ *expectedToOptional(OMPBuilder.createTask(
+ Loc, InsertPointTy(AllocaBB, AllocaBB->getFirstInsertionPt()),
+ BodyGenCB));
Builder.restoreIP(AfterIP);
OMPBuilder.finalize();
Builder.CreateRetVoid();
@@ -6592,9 +6621,10 @@ TEST_F(OpenMPIRBuilderTest, CreateTaskNoArgs) {
BasicBlock *BodyBB = splitBB(Builder, /*CreateBranch=*/true, "alloca.split");
OpenMPIRBuilder::LocationDescription Loc(
InsertPointTy(BodyBB, BodyBB->getFirstInsertionPt()), DL);
- OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(OMPBuilder.createTask(
- Loc, InsertPointTy(AllocaBB, AllocaBB->getFirstInsertionPt()),
- BodyGenCB));
+ OpenMPIRBuilder::InsertPointTy AfterIP =
+ *expectedToOptional(OMPBuilder.createTask(
+ Loc, InsertPointTy(AllocaBB, AllocaBB->getFirstInsertionPt()),
+ BodyGenCB));
Builder.restoreIP(AfterIP);
OMPBuilder.finalize();
Builder.CreateRetVoid();
@@ -6624,9 +6654,11 @@ TEST_F(OpenMPIRBuilderTest, CreateTaskUntied) {
BasicBlock *BodyBB = splitBB(Builder, /*CreateBranch=*/true, "alloca.split");
OpenMPIRBuilder::LocationDescription Loc(
InsertPointTy(BodyBB, BodyBB->getFirstInsertionPt()), DL);
- OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(OMPBuilder.createTask(
- Loc, InsertPointTy(AllocaBB, AllocaBB->getFirstInsertionPt()), BodyGenCB,
- /*Tied=*/false));
+ OpenMPIRBuilder::InsertPointTy AfterIP =
+ *expectedToOptional(OMPBuilder.createTask(
+ Loc, InsertPointTy(AllocaBB, AllocaBB->getFirstInsertionPt()),
+ BodyGenCB,
+ /*Tied=*/false));
Builder.restoreIP(AfterIP);
OMPBuilder.finalize();
Builder.CreateRetVoid();
@@ -6664,9 +6696,11 @@ TEST_F(OpenMPIRBuilderTest, CreateTaskDepend) {
Type::getInt32Ty(M->getContext()), InDep);
DDS.push_back(DDIn);
}
- OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(OMPBuilder.createTask(
- Loc, InsertPointTy(AllocaBB, AllocaBB->getFirstInsertionPt()), BodyGenCB,
- /*Tied=*/false, /*Final*/ nullptr, /*IfCondition*/ nullptr, DDS));
+ OpenMPIRBuilder::InsertPointTy AfterIP =
+ *expectedToOptional(OMPBuilder.createTask(
+ Loc, InsertPointTy(AllocaBB, AllocaBB->getFirstInsertionPt()),
+ BodyGenCB,
+ /*Tied=*/false, /*Final*/ nullptr, /*IfCondition*/ nullptr, DDS));
Builder.restoreIP(AfterIP);
OMPBuilder.finalize();
Builder.CreateRetVoid();
@@ -6738,8 +6772,8 @@ TEST_F(OpenMPIRBuilderTest, CreateTaskFinal) {
ConstantInt::get(Type::getInt32Ty(M->getContext()), 0U));
OpenMPIRBuilder::LocationDescription Loc(Builder.saveIP(), DL);
OpenMPIRBuilder::InsertPointTy AfterIP =
- cantFail(OMPBuilder.createTask(Loc, AllocaIP, BodyGenCB,
- /*Tied=*/false, Final));
+ *expectedToOptional(OMPBuilder.createTask(Loc, AllocaIP, BodyGenCB,
+ /*Tied=*/false, Final));
Builder.restoreIP(AfterIP);
OMPBuilder.finalize();
Builder.CreateRetVoid();
@@ -6795,7 +6829,7 @@ TEST_F(OpenMPIRBuilderTest, CreateTaskIfCondition) {
CmpInst::Predicate::ICMP_EQ, F->getArg(0),
ConstantInt::get(Type::getInt32Ty(M->getContext()), 0U));
OpenMPIRBuilder::LocationDescription Loc(Builder.saveIP(), DL);
- OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(
+ OpenMPIRBuilder::InsertPointTy AfterIP = *expectedToOptional(
OMPBuilder.createTask(Loc, AllocaIP, BodyGenCB,
/*Tied=*/false, /*Final=*/nullptr, IfCondition));
Builder.restoreIP(AfterIP);
@@ -6889,9 +6923,10 @@ TEST_F(OpenMPIRBuilderTest, CreateTaskgroup) {
BasicBlock *BodyBB = splitBB(Builder, /*CreateBranch=*/true, "alloca.split");
OpenMPIRBuilder::LocationDescription Loc(
InsertPointTy(BodyBB, BodyBB->getFirstInsertionPt()), DL);
- OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(OMPBuilder.createTaskgroup(
- Loc, InsertPointTy(AllocaBB, AllocaBB->getFirstInsertionPt()),
- BodyGenCB));
+ OpenMPIRBuilder::InsertPointTy AfterIP =
+ *expectedToOptional(OMPBuilder.createTaskgroup(
+ Loc, InsertPointTy(AllocaBB, AllocaBB->getFirstInsertionPt()),
+ BodyGenCB));
Builder.restoreIP(AfterIP);
OMPBuilder.finalize();
Builder.CreateRetVoid();
@@ -6968,8 +7003,8 @@ TEST_F(OpenMPIRBuilderTest, CreateTaskgroupWithTasks) {
return Error::success();
};
OpenMPIRBuilder::LocationDescription Loc(Builder.saveIP(), DL);
- OpenMPIRBuilder::InsertPointTy TaskIP1 =
- cantFail(OMPBuilder.createTask(Loc, AllocaIP, TaskBodyGenCB1));
+ OpenMPIRBuilder::InsertPointTy TaskIP1 = *expectedToOptional(
+ OMPBuilder.createTask(Loc, AllocaIP, TaskBodyGenCB1));
Builder.restoreIP(TaskIP1);
auto TaskBodyGenCB2 = [&](InsertPointTy AllocaIP, InsertPointTy CodeGenIP) {
@@ -6981,8 +7016,8 @@ TEST_F(OpenMPIRBuilderTest, CreateTaskgroupWithTasks) {
return Error::success();
};
OpenMPIRBuilder::LocationDescription Loc2(Builder.saveIP(), DL);
- OpenMPIRBuilder::InsertPointTy TaskIP2 =
- cantFail(OMPBuilder.createTask(Loc2, AllocaIP, TaskBodyGenCB2));
+ OpenMPIRBuilder::InsertPointTy TaskIP2 = *expectedToOptional(
+ OMPBuilder.createTask(Loc2, AllocaIP, TaskBodyGenCB2));
Builder.restoreIP(TaskIP2);
return Error::success();
};
@@ -6991,9 +7026,10 @@ TEST_F(OpenMPIRBuilderTest, CreateTaskgroupWithTasks) {
BasicBlock *BodyBB = splitBB(Builder, /*CreateBranch=*/true, "alloca.split");
OpenMPIRBuilder::LocationDescription Loc(
InsertPointTy(BodyBB, BodyBB->getFirstInsertionPt()), DL);
- OpenMPIRBuilder::InsertPointTy AfterIP = cantFail(OMPBuilder.createTaskgroup(
- Loc, InsertPointTy(AllocaBB, AllocaBB->getFirstInsertionPt()),
- BodyGenCB));
+ OpenMPIRBuilder::InsertPointTy AfterIP =
+ *expectedToOptional(OMPBuilder.createTaskgroup(
+ Loc, InsertPointTy(AllocaBB, AllocaBB->getFirstInsertionPt()),
+ BodyGenCB));
Builder.restoreIP(AfterIP);
OMPBuilder.finalize();
Builder.CreateRetVoid();
>From a338fa6acf0297aff625ef2cfa777e515d5881a6 Mon Sep 17 00:00:00 2001
From: Sergio Afonso <safonsof at amd.com>
Date: Fri, 15 Nov 2024 12:55:19 +0000
Subject: [PATCH 3/3] Replace expectedToOptional() with macros
---
.../Frontend/OpenMPIRBuilderTest.cpp | 561 +++++++++++-------
1 file changed, 336 insertions(+), 225 deletions(-)
diff --git a/llvm/unittests/Frontend/OpenMPIRBuilderTest.cpp b/llvm/unittests/Frontend/OpenMPIRBuilderTest.cpp
index 4f8e9da5902ff0..d6793ac80d0d9a 100644
--- a/llvm/unittests/Frontend/OpenMPIRBuilderTest.cpp
+++ b/llvm/unittests/Frontend/OpenMPIRBuilderTest.cpp
@@ -19,6 +19,7 @@
#include "llvm/IR/Verifier.h"
#include "llvm/Passes/PassBuilder.h"
#include "llvm/Support/Casting.h"
+#include "llvm/Testing/Support/Error.h"
#include "llvm/Transforms/Utils/BasicBlockUtils.h"
#include "gmock/gmock.h"
#include "gtest/gtest.h"
@@ -27,6 +28,23 @@
using namespace llvm;
using namespace omp;
+// Helper that intends to be functionally equivalent to `VarType VarName = Init`
+// for an `Init` that returns an `Expected<VarType>` value. It produces an error
+// message and returns if `Init` didn't produce a valid result.
+#define ASSERT_EXPECTED_INIT(VarType, VarName, Init) \
+ auto __##Expected##VarName = Init; \
+ ASSERT_THAT_EXPECTED(__##Expected##VarName, Succeeded()); \
+ VarType VarName = *__##Expected##VarName
+
+// Similar to ASSERT_EXPECTED_INIT, but returns a given expression in case of
+// error after printing the error message.
+#define ASSERT_EXPECTED_INIT_RETURN(VarType, VarName, Init, Return) \
+ auto __##Expected##VarName = Init; \
+ EXPECT_THAT_EXPECTED(__##Expected##VarName, Succeeded()); \
+ if (!__##Expected##VarName) \
+ return Return; \
+ VarType VarName = *__##Expected##VarName
+
// Wrapper lambdas to allow using EXPECT*() macros inside of error-returning
// callbacks.
#define FINICB_WRAPPER(cb) \
@@ -41,6 +59,12 @@ using namespace omp;
return Error::success(); \
}
+#define LOOP_BODYGENCB_WRAPPER(cb) \
+ [&cb](InsertPointTy CodeGenIP, Value *LC) -> Error { \
+ cb(CodeGenIP, LC); \
+ return Error::success(); \
+ }
+
namespace {
/// Create an instruction that uses the values in \p Values. We use "printf"
@@ -236,8 +260,10 @@ class OpenMPIRBuilderTest : public testing::Test {
return Error::success();
};
- CanonicalLoopInfo *Loop = *expectedToOptional(
- OMPBuilder.createCanonicalLoop(Loc, LoopBodyGenCB, CastedTripCount));
+ ASSERT_EXPECTED_INIT_RETURN(
+ CanonicalLoopInfo *, Loop,
+ OMPBuilder.createCanonicalLoop(Loc, LoopBodyGenCB, CastedTripCount),
+ nullptr);
// Finalize the function.
Builder.restoreIP(Loop->getAfterIP());
@@ -344,15 +370,16 @@ TEST_F(OpenMPIRBuilderTest, CreateBarrier) {
IRBuilder<> Builder(BB);
- ASSERT_TRUE(expectedToOptional(
- OMPBuilder.createBarrier({IRBuilder<>::InsertPoint()}, OMPD_for)));
+ ASSERT_THAT_EXPECTED(
+ OMPBuilder.createBarrier({IRBuilder<>::InsertPoint()}, OMPD_for),
+ Succeeded());
EXPECT_TRUE(M->global_empty());
EXPECT_EQ(M->size(), 1U);
EXPECT_EQ(F->size(), 1U);
EXPECT_EQ(BB->size(), 0U);
OpenMPIRBuilder::LocationDescription Loc({Builder.saveIP()});
- ASSERT_TRUE(expectedToOptional(OMPBuilder.createBarrier(Loc, OMPD_for)));
+ ASSERT_THAT_EXPECTED(OMPBuilder.createBarrier(Loc, OMPD_for), Succeeded());
EXPECT_FALSE(M->global_empty());
EXPECT_EQ(M->size(), 3U);
EXPECT_EQ(F->size(), 1U);
@@ -395,8 +422,8 @@ TEST_F(OpenMPIRBuilderTest, CreateCancel) {
IRBuilder<> Builder(BB);
OpenMPIRBuilder::LocationDescription Loc({Builder.saveIP()});
- OpenMPIRBuilder::InsertPointTy NewIP =
- *expectedToOptional(OMPBuilder.createCancel(Loc, nullptr, OMPD_parallel));
+ ASSERT_EXPECTED_INIT(OpenMPIRBuilder::InsertPointTy, NewIP,
+ OMPBuilder.createCancel(Loc, nullptr, OMPD_parallel));
Builder.restoreIP(NewIP);
EXPECT_FALSE(M->global_empty());
EXPECT_EQ(M->size(), 4U);
@@ -463,7 +490,8 @@ TEST_F(OpenMPIRBuilderTest, CreateCancelIfCond) {
IRBuilder<> Builder(BB);
OpenMPIRBuilder::LocationDescription Loc({Builder.saveIP()});
- OpenMPIRBuilder::InsertPointTy NewIP = *expectedToOptional(
+ ASSERT_EXPECTED_INIT(
+ OpenMPIRBuilder::InsertPointTy, NewIP,
OMPBuilder.createCancel(Loc, Builder.getTrue(), OMPD_parallel));
Builder.restoreIP(NewIP);
EXPECT_FALSE(M->global_empty());
@@ -537,8 +565,8 @@ TEST_F(OpenMPIRBuilderTest, CreateCancelBarrier) {
IRBuilder<> Builder(BB);
OpenMPIRBuilder::LocationDescription Loc({Builder.saveIP()});
- OpenMPIRBuilder::InsertPointTy NewIP =
- *expectedToOptional(OMPBuilder.createBarrier(Loc, OMPD_for));
+ ASSERT_EXPECTED_INIT(OpenMPIRBuilder::InsertPointTy, NewIP,
+ OMPBuilder.createBarrier(Loc, OMPD_for));
Builder.restoreIP(NewIP);
EXPECT_FALSE(M->global_empty());
EXPECT_EQ(M->size(), 3U);
@@ -584,7 +612,7 @@ TEST_F(OpenMPIRBuilderTest, DbgLoc) {
IRBuilder<> Builder(BB);
OpenMPIRBuilder::LocationDescription Loc({Builder.saveIP(), DL});
- ASSERT_TRUE(expectedToOptional(OMPBuilder.createBarrier(Loc, OMPD_for)));
+ ASSERT_THAT_EXPECTED(OMPBuilder.createBarrier(Loc, OMPD_for), Succeeded());
CallInst *GTID = dyn_cast<CallInst>(&BB->front());
CallInst *Barrier = dyn_cast<CallInst>(GTID->getNextNode());
EXPECT_EQ(GTID->getDebugLoc(), DL);
@@ -683,10 +711,10 @@ TEST_F(OpenMPIRBuilderTest, ParallelSimpleGPU) {
IRBuilder<>::InsertPoint AllocaIP(&F->getEntryBlock(),
F->getEntryBlock().getFirstInsertionPt());
- OpenMPIRBuilder::InsertPointTy AfterIP =
- *expectedToOptional(OMPBuilder.createParallel(
- Loc, AllocaIP, BodyGenCB, PrivCB, FiniCB, nullptr, nullptr,
- OMP_PROC_BIND_default, false));
+ ASSERT_EXPECTED_INIT(OpenMPIRBuilder::InsertPointTy, AfterIP,
+ OMPBuilder.createParallel(
+ Loc, AllocaIP, BodyGenCB, PrivCB, FiniCB, nullptr,
+ nullptr, OMP_PROC_BIND_default, false));
EXPECT_EQ(NumBodiesGenerated, 1U);
EXPECT_EQ(NumPrivatizedVars, 1U);
@@ -796,10 +824,10 @@ TEST_F(OpenMPIRBuilderTest, ParallelSimple) {
IRBuilder<>::InsertPoint AllocaIP(&F->getEntryBlock(),
F->getEntryBlock().getFirstInsertionPt());
- OpenMPIRBuilder::InsertPointTy AfterIP =
- *expectedToOptional(OMPBuilder.createParallel(
- Loc, AllocaIP, BodyGenCB, PrivCB, FiniCB, nullptr, nullptr,
- OMP_PROC_BIND_default, false));
+ ASSERT_EXPECTED_INIT(OpenMPIRBuilder::InsertPointTy, AfterIP,
+ OMPBuilder.createParallel(
+ Loc, AllocaIP, BodyGenCB, PrivCB, FiniCB, nullptr,
+ nullptr, OMP_PROC_BIND_default, false));
EXPECT_EQ(NumBodiesGenerated, 1U);
EXPECT_EQ(NumPrivatizedVars, 1U);
EXPECT_EQ(NumFinalizationPoints, 1U);
@@ -885,22 +913,23 @@ TEST_F(OpenMPIRBuilderTest, ParallelNested) {
BasicBlock *NewBB = SplitBlock(CGBB, &*CodeGenIP.getPoint());
CGBB->getTerminator()->eraseFromParent();
- OpenMPIRBuilder::InsertPointTy AfterIP =
- *expectedToOptional(OMPBuilder.createParallel(
- InsertPointTy(CGBB, CGBB->end()), AllocaIP, InnerBodyGenCB, PrivCB,
- FiniCB, nullptr, nullptr, OMP_PROC_BIND_default, false));
+ ASSERT_EXPECTED_INIT(
+ OpenMPIRBuilder::InsertPointTy, AfterIP,
+ OMPBuilder.createParallel(InsertPointTy(CGBB, CGBB->end()), AllocaIP,
+ InnerBodyGenCB, PrivCB, FiniCB, nullptr,
+ nullptr, OMP_PROC_BIND_default, false));
Builder.restoreIP(AfterIP);
Builder.CreateBr(NewBB);
- return Error::success();
};
IRBuilder<>::InsertPoint AllocaIP(&F->getEntryBlock(),
F->getEntryBlock().getFirstInsertionPt());
- OpenMPIRBuilder::InsertPointTy AfterIP =
- *expectedToOptional(OMPBuilder.createParallel(
- Loc, AllocaIP, OuterBodyGenCB, PrivCB, FiniCB, nullptr, nullptr,
- OMP_PROC_BIND_default, false));
+ ASSERT_EXPECTED_INIT(OpenMPIRBuilder::InsertPointTy, AfterIP,
+ OMPBuilder.createParallel(
+ Loc, AllocaIP, BODYGENCB_WRAPPER(OuterBodyGenCB),
+ PrivCB, FiniCB, nullptr, nullptr,
+ OMP_PROC_BIND_default, false));
EXPECT_EQ(NumInnerBodiesGenerated, 1U);
EXPECT_EQ(NumOuterBodiesGenerated, 1U);
@@ -989,30 +1018,32 @@ TEST_F(OpenMPIRBuilderTest, ParallelNested2Inner) {
NewBB1->getTerminator()->eraseFromParent();
;
- OpenMPIRBuilder::InsertPointTy AfterIP1 =
- *expectedToOptional(OMPBuilder.createParallel(
- InsertPointTy(CGBB, CGBB->end()), AllocaIP, InnerBodyGenCB, PrivCB,
- FiniCB, nullptr, nullptr, OMP_PROC_BIND_default, false));
+ ASSERT_EXPECTED_INIT(
+ OpenMPIRBuilder::InsertPointTy, AfterIP1,
+ OMPBuilder.createParallel(InsertPointTy(CGBB, CGBB->end()), AllocaIP,
+ InnerBodyGenCB, PrivCB, FiniCB, nullptr,
+ nullptr, OMP_PROC_BIND_default, false));
Builder.restoreIP(AfterIP1);
Builder.CreateBr(NewBB1);
- OpenMPIRBuilder::InsertPointTy AfterIP2 =
- *expectedToOptional(OMPBuilder.createParallel(
- InsertPointTy(NewBB1, NewBB1->end()), AllocaIP, InnerBodyGenCB,
- PrivCB, FiniCB, nullptr, nullptr, OMP_PROC_BIND_default, false));
+ ASSERT_EXPECTED_INIT(OpenMPIRBuilder::InsertPointTy, AfterIP2,
+ OMPBuilder.createParallel(
+ InsertPointTy(NewBB1, NewBB1->end()), AllocaIP,
+ InnerBodyGenCB, PrivCB, FiniCB, nullptr, nullptr,
+ OMP_PROC_BIND_default, false));
Builder.restoreIP(AfterIP2);
Builder.CreateBr(NewBB2);
- return Error::success();
};
IRBuilder<>::InsertPoint AllocaIP(&F->getEntryBlock(),
F->getEntryBlock().getFirstInsertionPt());
- OpenMPIRBuilder::InsertPointTy AfterIP =
- *expectedToOptional(OMPBuilder.createParallel(
- Loc, AllocaIP, OuterBodyGenCB, PrivCB, FiniCB, nullptr, nullptr,
- OMP_PROC_BIND_default, false));
+ ASSERT_EXPECTED_INIT(OpenMPIRBuilder::InsertPointTy, AfterIP,
+ OMPBuilder.createParallel(
+ Loc, AllocaIP, BODYGENCB_WRAPPER(OuterBodyGenCB),
+ PrivCB, FiniCB, nullptr, nullptr,
+ OMP_PROC_BIND_default, false));
EXPECT_EQ(NumInnerBodiesGenerated, 2U);
EXPECT_EQ(NumOuterBodiesGenerated, 1U);
@@ -1124,7 +1155,8 @@ TEST_F(OpenMPIRBuilderTest, ParallelIfCond) {
IRBuilder<>::InsertPoint AllocaIP(&F->getEntryBlock(),
F->getEntryBlock().getFirstInsertionPt());
- OpenMPIRBuilder::InsertPointTy AfterIP = *expectedToOptional(
+ ASSERT_EXPECTED_INIT(
+ OpenMPIRBuilder::InsertPointTy, AfterIP,
OMPBuilder.createParallel(Loc, AllocaIP, BodyGenCB, PrivCB, FiniCB,
Builder.CreateIsNotNull(F->arg_begin()),
nullptr, OMP_PROC_BIND_default, false));
@@ -1188,7 +1220,8 @@ TEST_F(OpenMPIRBuilderTest, ParallelCancelBarrier) {
// Create three barriers, two cancel barriers but only one checked.
Function *CBFn, *BFn;
- OpenMPIRBuilder::InsertPointTy BarrierIP1 = *expectedToOptional(
+ ASSERT_EXPECTED_INIT(
+ OpenMPIRBuilder::InsertPointTy, BarrierIP1,
OMPBuilder.createBarrier(Builder.saveIP(), OMPD_parallel));
Builder.restoreIP(BarrierIP1);
@@ -1201,7 +1234,8 @@ TEST_F(OpenMPIRBuilderTest, ParallelCancelBarrier) {
ASSERT_EQ(CBFn->user_back()->getNumUses(), 1U);
CheckedBarrier = cast<CallInst>(CBFn->user_back());
- OpenMPIRBuilder::InsertPointTy BarrierIP2 = *expectedToOptional(
+ ASSERT_EXPECTED_INIT(
+ OpenMPIRBuilder::InsertPointTy, BarrierIP2,
OMPBuilder.createBarrier(Builder.saveIP(), OMPD_parallel, true));
Builder.restoreIP(BarrierIP2);
CBFn = M->getFunction("__kmpc_cancel_barrier");
@@ -1213,9 +1247,9 @@ TEST_F(OpenMPIRBuilderTest, ParallelCancelBarrier) {
ASSERT_TRUE(isa<CallInst>(BFn->user_back()));
ASSERT_EQ(BFn->user_back()->getNumUses(), 0U);
- OpenMPIRBuilder::InsertPointTy BarrierIP3 =
- *expectedToOptional(OMPBuilder.createBarrier(
- Builder.saveIP(), OMPD_parallel, false, false));
+ ASSERT_EXPECTED_INIT(OpenMPIRBuilder::InsertPointTy, BarrierIP3,
+ OMPBuilder.createBarrier(Builder.saveIP(),
+ OMPD_parallel, false, false));
Builder.restoreIP(BarrierIP3);
ASSERT_EQ(CBFn->getNumUses(), 2U);
ASSERT_EQ(BFn->getNumUses(), 1U);
@@ -1246,11 +1280,11 @@ TEST_F(OpenMPIRBuilderTest, ParallelCancelBarrier) {
IRBuilder<>::InsertPoint AllocaIP(&F->getEntryBlock(),
F->getEntryBlock().getFirstInsertionPt());
- OpenMPIRBuilder::InsertPointTy AfterIP =
- *expectedToOptional(OMPBuilder.createParallel(
- Loc, AllocaIP, BODYGENCB_WRAPPER(BodyGenCB), PrivCB, FiniCB,
- Builder.CreateIsNotNull(F->arg_begin()), nullptr,
- OMP_PROC_BIND_default, true));
+ ASSERT_EXPECTED_INIT(OpenMPIRBuilder::InsertPointTy, AfterIP,
+ OMPBuilder.createParallel(
+ Loc, AllocaIP, BODYGENCB_WRAPPER(BodyGenCB), PrivCB,
+ FiniCB, Builder.CreateIsNotNull(F->arg_begin()),
+ nullptr, OMP_PROC_BIND_default, true));
EXPECT_EQ(NumBodiesGenerated, 1U);
EXPECT_EQ(NumPrivatizedVars, 0U);
@@ -1333,10 +1367,10 @@ TEST_F(OpenMPIRBuilderTest, ParallelForwardAsPointers) {
IRBuilder<>::InsertPoint AllocaIP(&F->getEntryBlock(),
F->getEntryBlock().getFirstInsertionPt());
- OpenMPIRBuilder::InsertPointTy AfterIP =
- *expectedToOptional(OMPBuilder.createParallel(
- Loc, AllocaIP, BodyGenCB, PrivCB, FiniCB, nullptr, nullptr,
- OMP_PROC_BIND_default, false));
+ ASSERT_EXPECTED_INIT(OpenMPIRBuilder::InsertPointTy, AfterIP,
+ OMPBuilder.createParallel(
+ Loc, AllocaIP, BodyGenCB, PrivCB, FiniCB, nullptr,
+ nullptr, OMP_PROC_BIND_default, false));
Builder.restoreIP(AfterIP);
Builder.CreateRetVoid();
@@ -1370,7 +1404,8 @@ TEST_F(OpenMPIRBuilderTest, CanonicalLoopSimple) {
return Error::success();
};
- CanonicalLoopInfo *Loop = *expectedToOptional(
+ ASSERT_EXPECTED_INIT(
+ CanonicalLoopInfo *, Loop,
OMPBuilder.createCanonicalLoop(Loc, LoopBodyGenCB, TripCount));
Builder.restoreIP(Loop->getAfterIP());
@@ -1426,9 +1461,11 @@ TEST_F(OpenMPIRBuilderTest, CanonicalLoopBounds) {
auto LoopBodyGenCB = [&](InsertPointTy CodeGenIP, llvm::Value *LC) {
return Error::success();
};
- CanonicalLoopInfo *Loop = *expectedToOptional(
+ ASSERT_EXPECTED_INIT_RETURN(
+ CanonicalLoopInfo *, Loop,
OMPBuilder.createCanonicalLoop(Loc, LoopBodyGenCB, StartVal, StopVal,
- StepVal, IsSigned, InclusiveStop));
+ StepVal, IsSigned, InclusiveStop),
+ -1);
Loop->assertOK();
Builder.restoreIP(Loop->getAfterIP());
Value *TripCount = Loop->getTripCount();
@@ -1523,17 +1560,20 @@ TEST_F(OpenMPIRBuilderTest, CollapseNestedLoops) {
Call = createPrintfCall(Builder, "body i=%d j=%d\\n", {OuterLC, InnerLC});
return Error::success();
};
- InnerLoop = *expectedToOptional(OMPBuilder.createCanonicalLoop(
- Builder.saveIP(), InnerLoopBodyGenCB, InnerTripCount, "inner"));
+ ASSERT_EXPECTED_INIT(
+ CanonicalLoopInfo *, InnerLoopResult,
+ OMPBuilder.createCanonicalLoop(Builder.saveIP(), InnerLoopBodyGenCB,
+ InnerTripCount, "inner"));
+ InnerLoop = InnerLoopResult;
Builder.restoreIP(InnerLoop->getAfterIP());
InbetweenTrail =
createPrintfCall(Builder, "In-between trail i=%d\\n", {OuterLC});
- return Error::success();
};
- CanonicalLoopInfo *OuterLoop =
- *expectedToOptional(OMPBuilder.createCanonicalLoop(
- OuterLoc, OuterLoopBodyGenCB, OuterTripCount, "outer"));
+ ASSERT_EXPECTED_INIT(CanonicalLoopInfo *, OuterLoop,
+ OMPBuilder.createCanonicalLoop(
+ OuterLoc, LOOP_BODYGENCB_WRAPPER(OuterLoopBodyGenCB),
+ OuterTripCount, "outer"));
// Finish the function.
Builder.restoreIP(OuterLoop->getAfterIP());
@@ -1584,6 +1624,7 @@ TEST_F(OpenMPIRBuilderTest, TileSingleLoop) {
BasicBlock *BodyCode;
CanonicalLoopInfo *Loop =
buildSingleLoopFunction(DL, OMPBuilder, 32, &Call, &BodyCode);
+ ASSERT_NE(Loop, nullptr);
Instruction *OrigIndVar = Loop->getIndVar();
EXPECT_EQ(Call->getOperand(1), OrigIndVar);
@@ -1645,13 +1686,16 @@ TEST_F(OpenMPIRBuilderTest, TileNestedLoops) {
createPrintfCall(Builder, "i=%d j=%d\\n", {OuterLC, InnerLC});
return Error::success();
};
- InnerLoop = *expectedToOptional(OMPBuilder.createCanonicalLoop(
- OuterCodeGenIP, InnerLoopBodyGenCB, TripCount, "inner"));
- return Error::success();
+ ASSERT_EXPECTED_INIT(CanonicalLoopInfo *, InnerLoopResult,
+ OMPBuilder.createCanonicalLoop(OuterCodeGenIP,
+ InnerLoopBodyGenCB,
+ TripCount, "inner"));
+ InnerLoop = InnerLoopResult;
};
- CanonicalLoopInfo *OuterLoop =
- *expectedToOptional(OMPBuilder.createCanonicalLoop(
- Loc, OuterLoopBodyGenCB, TripCount, "outer"));
+ ASSERT_EXPECTED_INIT(
+ CanonicalLoopInfo *, OuterLoop,
+ OMPBuilder.createCanonicalLoop(
+ Loc, LOOP_BODYGENCB_WRAPPER(OuterLoopBodyGenCB), TripCount, "outer"));
// Finalize the function.
Builder.restoreIP(OuterLoop->getAfterIP());
@@ -1748,15 +1792,18 @@ TEST_F(OpenMPIRBuilderTest, TileNestedLoopsWithBounds) {
Call = createPrintfCall(Builder, "i=%d j=%d\\n", {OuterLC, InnerLC});
return Error::success();
};
- InnerLoop = *expectedToOptional(OMPBuilder.createCanonicalLoop(
- OuterCodeGenIP, InnerLoopBodyGenCB, InnerStartVal, InnerStopVal,
- InnerStep, false, false, ComputeIP, "inner"));
- return Error::success();
+ ASSERT_EXPECTED_INIT(
+ CanonicalLoopInfo *, InnerLoopResult,
+ OMPBuilder.createCanonicalLoop(OuterCodeGenIP, InnerLoopBodyGenCB,
+ InnerStartVal, InnerStopVal, InnerStep,
+ false, false, ComputeIP, "inner"));
+ InnerLoop = InnerLoopResult;
};
- CanonicalLoopInfo *OuterLoop =
- *expectedToOptional(OMPBuilder.createCanonicalLoop(
- Loc, OuterLoopBodyGenCB, OuterStartVal, OuterStopVal, OuterStep,
- false, false, ComputeIP, "outer"));
+ ASSERT_EXPECTED_INIT(CanonicalLoopInfo *, OuterLoop,
+ OMPBuilder.createCanonicalLoop(
+ Loc, LOOP_BODYGENCB_WRAPPER(OuterLoopBodyGenCB),
+ OuterStartVal, OuterStopVal, OuterStep, false, false,
+ ComputeIP, "outer"));
// Finalize the function
Builder.restoreIP(OuterLoop->getAfterIP());
@@ -1863,9 +1910,11 @@ TEST_F(OpenMPIRBuilderTest, TileSingleLoopCounts) {
auto LoopBodyGenCB = [&](InsertPointTy CodeGenIP, llvm::Value *LC) {
return Error::success();
};
- CanonicalLoopInfo *Loop = *expectedToOptional(
+ ASSERT_EXPECTED_INIT_RETURN(
+ CanonicalLoopInfo *, Loop,
OMPBuilder.createCanonicalLoop(Loc, LoopBodyGenCB, StartVal, StopVal,
- StepVal, IsSigned, InclusiveStop));
+ StepVal, IsSigned, InclusiveStop),
+ (unsigned)-1);
InsertPointTy AfterIP = Loop->getAfterIP();
// Tile the loop.
@@ -1932,6 +1981,7 @@ TEST_F(OpenMPIRBuilderTest, ApplySimd) {
OpenMPIRBuilder OMPBuilder(*M);
MapVector<Value *, Value *> AlignedVars;
CanonicalLoopInfo *CLI = buildSingleLoopFunction(DL, OMPBuilder, 32);
+ ASSERT_NE(CLI, nullptr);
// Simd-ize the loop.
OMPBuilder.applySimd(CLI, AlignedVars, /* IfCond */ nullptr,
@@ -1973,6 +2023,7 @@ TEST_F(OpenMPIRBuilderTest, ApplySimdCustomAligned) {
AlignedVars.insert({Load1, Builder.getInt64(AlignmentValue)});
CanonicalLoopInfo *CLI = buildSingleLoopFunction(DL, OMPBuilder, 32);
+ ASSERT_NE(CLI, nullptr);
// Simd-ize the loop.
OMPBuilder.applySimd(CLI, AlignedVars, /* IfCond */ nullptr,
@@ -2028,6 +2079,7 @@ TEST_F(OpenMPIRBuilderTest, ApplySimdlen) {
OpenMPIRBuilder OMPBuilder(*M);
MapVector<Value *, Value *> AlignedVars;
CanonicalLoopInfo *CLI = buildSingleLoopFunction(DL, OMPBuilder, 32);
+ ASSERT_NE(CLI, nullptr);
// Simd-ize the loop.
OMPBuilder.applySimd(CLI, AlignedVars,
@@ -2064,6 +2116,7 @@ TEST_F(OpenMPIRBuilderTest, ApplySafelenOrderConcurrent) {
MapVector<Value *, Value *> AlignedVars;
CanonicalLoopInfo *CLI = buildSingleLoopFunction(DL, OMPBuilder, 32);
+ ASSERT_NE(CLI, nullptr);
// Simd-ize the loop.
OMPBuilder.applySimd(
@@ -2101,6 +2154,7 @@ TEST_F(OpenMPIRBuilderTest, ApplySafelen) {
MapVector<Value *, Value *> AlignedVars;
CanonicalLoopInfo *CLI = buildSingleLoopFunction(DL, OMPBuilder, 32);
+ ASSERT_NE(CLI, nullptr);
OMPBuilder.applySimd(
CLI, AlignedVars, /* IfCond */ nullptr, OrderKind::OMP_ORDER_unknown,
@@ -2135,6 +2189,7 @@ TEST_F(OpenMPIRBuilderTest, ApplySimdlenSafelen) {
MapVector<Value *, Value *> AlignedVars;
CanonicalLoopInfo *CLI = buildSingleLoopFunction(DL, OMPBuilder, 32);
+ ASSERT_NE(CLI, nullptr);
OMPBuilder.applySimd(CLI, AlignedVars, /* IfCond */ nullptr,
OrderKind::OMP_ORDER_unknown,
@@ -2181,6 +2236,7 @@ TEST_F(OpenMPIRBuilderTest, ApplySimdIf) {
Value *IfCmp = Builder.CreateICmpNE(Load1, Load2);
CanonicalLoopInfo *CLI = buildSingleLoopFunction(DL, OMPBuilder, 32);
+ ASSERT_NE(CLI, nullptr);
// Simd-ize the loop with if condition
OMPBuilder.applySimd(CLI, AlignedVars, IfCmp, OrderKind::OMP_ORDER_unknown,
@@ -2220,6 +2276,7 @@ TEST_F(OpenMPIRBuilderTest, UnrollLoopFull) {
OpenMPIRBuilder OMPBuilder(*M);
CanonicalLoopInfo *CLI = buildSingleLoopFunction(DL, OMPBuilder, 32);
+ ASSERT_NE(CLI, nullptr);
// Unroll the loop.
OMPBuilder.unrollLoopFull(DL, CLI);
@@ -2243,6 +2300,7 @@ TEST_F(OpenMPIRBuilderTest, UnrollLoopFull) {
TEST_F(OpenMPIRBuilderTest, UnrollLoopPartial) {
OpenMPIRBuilder OMPBuilder(*M);
CanonicalLoopInfo *CLI = buildSingleLoopFunction(DL, OMPBuilder, 32);
+ ASSERT_NE(CLI, nullptr);
// Unroll the loop.
CanonicalLoopInfo *UnrolledLoop = nullptr;
@@ -2277,6 +2335,7 @@ TEST_F(OpenMPIRBuilderTest, UnrollLoopHeuristic) {
OpenMPIRBuilder OMPBuilder(*M);
CanonicalLoopInfo *CLI = buildSingleLoopFunction(DL, OMPBuilder, 32);
+ ASSERT_NE(CLI, nullptr);
// Unroll the loop.
OMPBuilder.unrollLoopHeuristic(DL, CLI);
@@ -2316,17 +2375,20 @@ TEST_F(OpenMPIRBuilderTest, StaticWorkshareLoopTarget) {
Value *StepVal = ConstantInt::get(LCTy, 2);
auto LoopBodyGen = [&](InsertPointTy, Value *) { return Error::success(); };
- CanonicalLoopInfo *CLI = *expectedToOptional(OMPBuilder.createCanonicalLoop(
- Loc, LoopBodyGen, StartVal, StopVal, StepVal, false, false));
+ ASSERT_EXPECTED_INIT(CanonicalLoopInfo *, CLI,
+ OMPBuilder.createCanonicalLoop(Loc, LoopBodyGen,
+ StartVal, StopVal,
+ StepVal, false, false));
BasicBlock *Preheader = CLI->getPreheader();
Value *TripCount = CLI->getTripCount();
Builder.SetInsertPoint(BB, BB->getFirstInsertionPt());
- OpenMPIRBuilder::InsertPointTy AfterIP =
- *expectedToOptional(OMPBuilder.applyWorkshareLoop(
- DL, CLI, AllocaIP, true, OMP_SCHEDULE_Static, nullptr, false, false,
- false, false, WorksharingLoopType::ForStaticLoop));
+ ASSERT_EXPECTED_INIT(OpenMPIRBuilder::InsertPointTy, AfterIP,
+ OMPBuilder.applyWorkshareLoop(
+ DL, CLI, AllocaIP, true, OMP_SCHEDULE_Static,
+ nullptr, false, false, false, false,
+ WorksharingLoopType::ForStaticLoop));
Builder.restoreIP(AfterIP);
Builder.CreateRetVoid();
@@ -2379,10 +2441,10 @@ TEST_F(OpenMPIRBuilderTest, StaticWorkShareLoop) {
auto LoopBodyGen = [&](InsertPointTy, llvm::Value *) {
return Error::success();
};
-
- CanonicalLoopInfo *CLI = *expectedToOptional(OMPBuilder.createCanonicalLoop(
- Loc, LoopBodyGen, StartVal, StopVal, StepVal,
- /*IsSigned=*/false, /*InclusiveStop=*/false));
+ ASSERT_EXPECTED_INIT(CanonicalLoopInfo *, CLI,
+ OMPBuilder.createCanonicalLoop(
+ Loc, LoopBodyGen, StartVal, StopVal, StepVal,
+ /*IsSigned=*/false, /*InclusiveStop=*/false));
BasicBlock *Preheader = CLI->getPreheader();
BasicBlock *Body = CLI->getBody();
Value *IV = CLI->getIndVar();
@@ -2391,8 +2453,10 @@ TEST_F(OpenMPIRBuilderTest, StaticWorkShareLoop) {
Builder.SetInsertPoint(BB, BB->getFirstInsertionPt());
InsertPointTy AllocaIP = Builder.saveIP();
- ASSERT_TRUE(expectedToOptional(OMPBuilder.applyWorkshareLoop(
- DL, CLI, AllocaIP, /*NeedsBarrier=*/true, OMP_SCHEDULE_Static)));
+ ASSERT_THAT_EXPECTED(OMPBuilder.applyWorkshareLoop(DL, CLI, AllocaIP,
+ /*NeedsBarrier=*/true,
+ OMP_SCHEDULE_Static),
+ Succeeded());
BasicBlock *Cond = Body->getSinglePredecessor();
Instruction *Cmp = &*Cond->begin();
@@ -2476,6 +2540,7 @@ TEST_P(OpenMPIRBuilderTestWithIVBits, StaticChunkedWorkshareLoop) {
CallInst *Call;
CanonicalLoopInfo *CLI =
buildSingleLoopFunction(DL, OMPBuilder, IVBits, &Call, &Body);
+ ASSERT_NE(CLI, nullptr);
Instruction *OrigIndVar = CLI->getIndVar();
EXPECT_EQ(Call->getOperand(1), OrigIndVar);
@@ -2484,9 +2549,11 @@ TEST_P(OpenMPIRBuilderTestWithIVBits, StaticChunkedWorkshareLoop) {
Value *ChunkSize = ConstantInt::get(LCTy, 5);
InsertPointTy AllocaIP{&F->getEntryBlock(),
F->getEntryBlock().getFirstInsertionPt()};
- ASSERT_TRUE(expectedToOptional(OMPBuilder.applyWorkshareLoop(
- DL, CLI, AllocaIP,
- /*NeedsBarrier=*/true, OMP_SCHEDULE_Static, ChunkSize)));
+ ASSERT_THAT_EXPECTED(OMPBuilder.applyWorkshareLoop(DL, CLI, AllocaIP,
+ /*NeedsBarrier=*/true,
+ OMP_SCHEDULE_Static,
+ ChunkSize),
+ Succeeded());
OMPBuilder.finalize();
EXPECT_FALSE(verifyModule(*M, &errs()));
@@ -2577,9 +2644,10 @@ TEST_P(OpenMPIRBuilderTestWithParams, DynamicWorkShareLoop) {
return Error::success();
};
- CanonicalLoopInfo *CLI = *expectedToOptional(OMPBuilder.createCanonicalLoop(
- Loc, LoopBodyGen, StartVal, StopVal, StepVal,
- /*IsSigned=*/false, /*InclusiveStop=*/false));
+ ASSERT_EXPECTED_INIT(CanonicalLoopInfo *, CLI,
+ OMPBuilder.createCanonicalLoop(
+ Loc, LoopBodyGen, StartVal, StopVal, StepVal,
+ /*IsSigned=*/false, /*InclusiveStop=*/false));
Builder.SetInsertPoint(BB, BB->getFirstInsertionPt());
InsertPointTy AllocaIP = Builder.saveIP();
@@ -2592,8 +2660,9 @@ TEST_P(OpenMPIRBuilderTestWithParams, DynamicWorkShareLoop) {
BasicBlock *LatchBlock = CLI->getLatch();
Value *IV = CLI->getIndVar();
- OpenMPIRBuilder::InsertPointTy EndIP =
- *expectedToOptional(OMPBuilder.applyWorkshareLoop(
+ ASSERT_EXPECTED_INIT(
+ OpenMPIRBuilder::InsertPointTy, EndIP,
+ OMPBuilder.applyWorkshareLoop(
DL, CLI, AllocaIP, /*NeedsBarrier=*/true, getSchedKind(SchedType),
ChunkVal, /*Simd=*/false,
(SchedType & omp::OMPScheduleType::ModifierMonotonic) ==
@@ -2722,9 +2791,10 @@ TEST_F(OpenMPIRBuilderTest, DynamicWorkShareLoopOrdered) {
return llvm::Error::success();
};
- CanonicalLoopInfo *CLI = *expectedToOptional(OMPBuilder.createCanonicalLoop(
- Loc, LoopBodyGen, StartVal, StopVal, StepVal,
- /*IsSigned=*/false, /*InclusiveStop=*/false));
+ ASSERT_EXPECTED_INIT(CanonicalLoopInfo *, CLI,
+ OMPBuilder.createCanonicalLoop(
+ Loc, LoopBodyGen, StartVal, StopVal, StepVal,
+ /*IsSigned=*/false, /*InclusiveStop=*/false));
Builder.SetInsertPoint(BB, BB->getFirstInsertionPt());
InsertPointTy AllocaIP = Builder.saveIP();
@@ -2736,13 +2806,14 @@ TEST_F(OpenMPIRBuilderTest, DynamicWorkShareLoopOrdered) {
BasicBlock *LatchBlock = CLI->getLatch();
Value *IV = CLI->getIndVar();
- OpenMPIRBuilder::InsertPointTy EndIP =
- *expectedToOptional(OMPBuilder.applyWorkshareLoop(
- DL, CLI, AllocaIP, /*NeedsBarrier=*/true, OMP_SCHEDULE_Static,
- ChunkVal,
- /*HasSimdModifier=*/false, /*HasMonotonicModifier=*/false,
- /*HasNonmonotonicModifier=*/false,
- /*HasOrderedClause=*/true));
+ ASSERT_EXPECTED_INIT(
+ OpenMPIRBuilder::InsertPointTy, EndIP,
+ OMPBuilder.applyWorkshareLoop(DL, CLI, AllocaIP, /*NeedsBarrier=*/true,
+ OMP_SCHEDULE_Static, ChunkVal,
+ /*HasSimdModifier=*/false,
+ /*HasMonotonicModifier=*/false,
+ /*HasNonmonotonicModifier=*/false,
+ /*HasOrderedClause=*/true));
// Add a termination to our block and check that it is internally consistent.
Builder.restoreIP(EndIP);
@@ -2829,9 +2900,10 @@ TEST_F(OpenMPIRBuilderTest, MasterDirective) {
EXPECT_NE(IPBB->end(), IP.getPoint());
};
- OpenMPIRBuilder::InsertPointTy AfterIP =
- *expectedToOptional(OMPBuilder.createMaster(
- Builder, BODYGENCB_WRAPPER(BodyGenCB), FINICB_WRAPPER(FiniCB)));
+ ASSERT_EXPECTED_INIT(OpenMPIRBuilder::InsertPointTy, AfterIP,
+ OMPBuilder.createMaster(Builder,
+ BODYGENCB_WRAPPER(BodyGenCB),
+ FINICB_WRAPPER(FiniCB)));
Builder.restoreIP(AfterIP);
Value *EntryBBTI = EntryBB->getTerminator();
EXPECT_NE(EntryBBTI, nullptr);
@@ -2910,9 +2982,10 @@ TEST_F(OpenMPIRBuilderTest, MaskedDirective) {
};
Constant *Filter = ConstantInt::get(Type::getInt32Ty(M->getContext()), 0);
- OpenMPIRBuilder::InsertPointTy AfterIP = *expectedToOptional(
- OMPBuilder.createMasked(Builder, BODYGENCB_WRAPPER(BodyGenCB),
- FINICB_WRAPPER(FiniCB), Filter));
+ ASSERT_EXPECTED_INIT(OpenMPIRBuilder::InsertPointTy, AfterIP,
+ OMPBuilder.createMasked(Builder,
+ BODYGENCB_WRAPPER(BodyGenCB),
+ FINICB_WRAPPER(FiniCB), Filter));
Builder.restoreIP(AfterIP);
Value *EntryBBTI = EntryBB->getTerminator();
EXPECT_NE(EntryBBTI, nullptr);
@@ -2978,7 +3051,8 @@ TEST_F(OpenMPIRBuilderTest, CriticalDirective) {
};
BasicBlock *EntryBB = Builder.GetInsertBlock();
- OpenMPIRBuilder::InsertPointTy AfterIP = *expectedToOptional(
+ ASSERT_EXPECTED_INIT(
+ OpenMPIRBuilder::InsertPointTy, AfterIP,
OMPBuilder.createCritical(Builder, BODYGENCB_WRAPPER(BodyGenCB),
FINICB_WRAPPER(FiniCB), "testCRT", nullptr));
Builder.restoreIP(AfterIP);
@@ -3228,9 +3302,10 @@ TEST_F(OpenMPIRBuilderTest, OrderedDirectiveThreads) {
// Test for "#omp ordered [threads]"
BasicBlock *EntryBB = Builder.GetInsertBlock();
- OpenMPIRBuilder::InsertPointTy AfterIP =
- *expectedToOptional(OMPBuilder.createOrderedThreadsSimd(
- Builder, BODYGENCB_WRAPPER(BodyGenCB), FINICB_WRAPPER(FiniCB), true));
+ ASSERT_EXPECTED_INIT(
+ OpenMPIRBuilder::InsertPointTy, AfterIP,
+ OMPBuilder.createOrderedThreadsSimd(Builder, BODYGENCB_WRAPPER(BodyGenCB),
+ FINICB_WRAPPER(FiniCB), true));
Builder.restoreIP(AfterIP);
Builder.CreateRetVoid();
@@ -3301,7 +3376,8 @@ TEST_F(OpenMPIRBuilderTest, OrderedDirectiveSimd) {
// Test for "#omp ordered simd"
BasicBlock *EntryBB = Builder.GetInsertBlock();
- OpenMPIRBuilder::InsertPointTy AfterIP = *expectedToOptional(
+ ASSERT_EXPECTED_INIT(
+ OpenMPIRBuilder::InsertPointTy, AfterIP,
OMPBuilder.createOrderedThreadsSimd(Builder, BODYGENCB_WRAPPER(BodyGenCB),
FINICB_WRAPPER(FiniCB), false));
Builder.restoreIP(AfterIP);
@@ -3417,7 +3493,8 @@ TEST_F(OpenMPIRBuilderTest, SingleDirective) {
EXPECT_NE(IPBB->end(), IP.getPoint());
};
- OpenMPIRBuilder::InsertPointTy AfterIP = *expectedToOptional(
+ ASSERT_EXPECTED_INIT(
+ OpenMPIRBuilder::InsertPointTy, AfterIP,
OMPBuilder.createSingle(Builder, BODYGENCB_WRAPPER(BodyGenCB),
FINICB_WRAPPER(FiniCB), /*IsNowait*/ false));
Builder.restoreIP(AfterIP);
@@ -3509,7 +3586,8 @@ TEST_F(OpenMPIRBuilderTest, SingleDirectiveNowait) {
EXPECT_NE(IPBB->end(), IP.getPoint());
};
- OpenMPIRBuilder::InsertPointTy AfterIP = *expectedToOptional(
+ ASSERT_EXPECTED_INIT(
+ OpenMPIRBuilder::InsertPointTy, AfterIP,
OMPBuilder.createSingle(Builder, BODYGENCB_WRAPPER(BodyGenCB),
FINICB_WRAPPER(FiniCB), /*IsNowait*/ true));
Builder.restoreIP(AfterIP);
@@ -3630,10 +3708,11 @@ TEST_F(OpenMPIRBuilderTest, SingleDirectiveCopyPrivate) {
EXPECT_NE(IPBB->end(), IP.getPoint());
};
- OpenMPIRBuilder::InsertPointTy AfterIP =
- *expectedToOptional(OMPBuilder.createSingle(
- Builder, BODYGENCB_WRAPPER(BodyGenCB), FINICB_WRAPPER(FiniCB),
- /*IsNowait*/ false, {CPVar}, {CopyFunc}));
+ ASSERT_EXPECTED_INIT(
+ OpenMPIRBuilder::InsertPointTy, AfterIP,
+ OMPBuilder.createSingle(Builder, BODYGENCB_WRAPPER(BodyGenCB),
+ FINICB_WRAPPER(FiniCB),
+ /*IsNowait*/ false, {CPVar}, {CopyFunc}));
Builder.restoreIP(AfterIP);
Value *EntryBBTI = EntryBB->getTerminator();
EXPECT_NE(EntryBBTI, nullptr);
@@ -3895,9 +3974,10 @@ TEST_F(OpenMPIRBuilderTest, OMPAtomicUpdate) {
Sub = IRB.CreateSub(ConstVal, Atomic);
return Sub;
};
- OpenMPIRBuilder::InsertPointTy AfterIP =
- *expectedToOptional(OMPBuilder.createAtomicUpdate(
- Builder, AllocaIP, X, Expr, AO, RMWOp, UpdateOp, IsXLHSInRHSPart));
+ ASSERT_EXPECTED_INIT(OpenMPIRBuilder::InsertPointTy, AfterIP,
+ OMPBuilder.createAtomicUpdate(Builder, AllocaIP, X, Expr,
+ AO, RMWOp, UpdateOp,
+ IsXLHSInRHSPart));
Builder.restoreIP(AfterIP);
BasicBlock *ContBB = EntryBB->getSingleSuccessor();
BranchInst *ContTI = dyn_cast<BranchInst>(ContBB->getTerminator());
@@ -3964,9 +4044,10 @@ TEST_F(OpenMPIRBuilderTest, OMPAtomicUpdateFloat) {
Sub = IRB.CreateFSub(ConstVal, Atomic);
return Sub;
};
- OpenMPIRBuilder::InsertPointTy AfterIP =
- *expectedToOptional(OMPBuilder.createAtomicUpdate(
- Builder, AllocaIP, X, Expr, AO, RMWOp, UpdateOp, IsXLHSInRHSPart));
+ ASSERT_EXPECTED_INIT(OpenMPIRBuilder::InsertPointTy, AfterIP,
+ OMPBuilder.createAtomicUpdate(Builder, AllocaIP, X, Expr,
+ AO, RMWOp, UpdateOp,
+ IsXLHSInRHSPart));
Builder.restoreIP(AfterIP);
BasicBlock *ContBB = EntryBB->getSingleSuccessor();
BranchInst *ContTI = dyn_cast<BranchInst>(ContBB->getTerminator());
@@ -4032,9 +4113,10 @@ TEST_F(OpenMPIRBuilderTest, OMPAtomicUpdateIntr) {
Sub = IRB.CreateSub(ConstVal, Atomic);
return Sub;
};
- OpenMPIRBuilder::InsertPointTy AfterIP =
- *expectedToOptional(OMPBuilder.createAtomicUpdate(
- Builder, AllocaIP, X, Expr, AO, RMWOp, UpdateOp, IsXLHSInRHSPart));
+ ASSERT_EXPECTED_INIT(OpenMPIRBuilder::InsertPointTy, AfterIP,
+ OMPBuilder.createAtomicUpdate(Builder, AllocaIP, X, Expr,
+ AO, RMWOp, UpdateOp,
+ IsXLHSInRHSPart));
Builder.restoreIP(AfterIP);
BasicBlock *ContBB = EntryBB->getSingleSuccessor();
BranchInst *ContTI = dyn_cast<BranchInst>(ContBB->getTerminator());
@@ -4106,10 +4188,10 @@ TEST_F(OpenMPIRBuilderTest, OMPAtomicCapture) {
// integer update - not used
auto UpdateOp = [&](Value *Atomic, IRBuilder<> &IRB) { return nullptr; };
- OpenMPIRBuilder::InsertPointTy AfterIP =
- *expectedToOptional(OMPBuilder.createAtomicCapture(
- Builder, AllocaIP, X, V, Expr, AO, RMWOp, UpdateOp, UpdateExpr,
- IsPostfixUpdate, IsXLHSInRHSPart));
+ ASSERT_EXPECTED_INIT(OpenMPIRBuilder::InsertPointTy, AfterIP,
+ OMPBuilder.createAtomicCapture(
+ Builder, AllocaIP, X, V, Expr, AO, RMWOp, UpdateOp,
+ UpdateExpr, IsPostfixUpdate, IsXLHSInRHSPart));
Builder.restoreIP(AfterIP);
EXPECT_EQ(EntryBB->getParent()->size(), 1U);
AtomicRMWInst *ARWM = dyn_cast<AtomicRMWInst>(Init->getNextNode());
@@ -4470,7 +4552,8 @@ TEST_F(OpenMPIRBuilderTest, CreateTeams) {
};
OpenMPIRBuilder::LocationDescription Loc({Builder.saveIP(), DL});
- OpenMPIRBuilder::InsertPointTy AfterIP = *expectedToOptional(
+ ASSERT_EXPECTED_INIT(
+ OpenMPIRBuilder::InsertPointTy, AfterIP,
OMPBuilder.createTeams(Builder, BodyGenCB, /*NumTeamsLower=*/nullptr,
/*NumTeamsUpper=*/nullptr,
/*ThreadLimit=*/nullptr, /*IfExpr=*/nullptr));
@@ -4535,11 +4618,12 @@ TEST_F(OpenMPIRBuilderTest, CreateTeamsWithThreadLimit) {
};
// `F` has an argument - an integer, so we use that as the thread limit.
- OpenMPIRBuilder::InsertPointTy AfterIP =
- *expectedToOptional(OMPBuilder.createTeams(
- /*=*/Builder, BodyGenCB, /*NumTeamsLower=*/nullptr,
- /*NumTeamsUpper=*/nullptr, /*ThreadLimit=*/F->arg_begin(),
- /*IfExpr=*/nullptr));
+ ASSERT_EXPECTED_INIT(OpenMPIRBuilder::InsertPointTy, AfterIP,
+ OMPBuilder.createTeams(
+ /*=*/Builder, BodyGenCB, /*NumTeamsLower=*/nullptr,
+ /*NumTeamsUpper=*/nullptr,
+ /*ThreadLimit=*/F->arg_begin(),
+ /*IfExpr=*/nullptr));
Builder.restoreIP(AfterIP);
Builder.CreateRetVoid();
@@ -4589,12 +4673,12 @@ TEST_F(OpenMPIRBuilderTest, CreateTeamsWithNumTeamsUpper) {
// `F` already has an integer argument, so we use that as upper bound to
// `num_teams`
- OpenMPIRBuilder::InsertPointTy AfterIP = *expectedToOptional(
- OMPBuilder.createTeams(Builder, BodyGenCB,
- /*NumTeamsLower=*/nullptr,
- /*NumTeamsUpper=*/F->arg_begin(),
- /*ThreadLimit=*/nullptr,
- /*IfExpr=*/nullptr));
+ ASSERT_EXPECTED_INIT(OpenMPIRBuilder::InsertPointTy, AfterIP,
+ OMPBuilder.createTeams(Builder, BodyGenCB,
+ /*NumTeamsLower=*/nullptr,
+ /*NumTeamsUpper=*/F->arg_begin(),
+ /*ThreadLimit=*/nullptr,
+ /*IfExpr=*/nullptr));
Builder.restoreIP(AfterIP);
Builder.CreateRetVoid();
@@ -4649,7 +4733,8 @@ TEST_F(OpenMPIRBuilderTest, CreateTeamsWithNumTeamsBoth) {
// `F` already has an integer argument, so we use that as upper bound to
// `num_teams`
- OpenMPIRBuilder::InsertPointTy AfterIP = *expectedToOptional(
+ ASSERT_EXPECTED_INIT(
+ OpenMPIRBuilder::InsertPointTy, AfterIP,
OMPBuilder.createTeams(Builder, BodyGenCB, NumTeamsLower, NumTeamsUpper,
/*ThreadLimit=*/nullptr, /*IfExpr=*/nullptr));
Builder.restoreIP(AfterIP);
@@ -4712,9 +4797,10 @@ TEST_F(OpenMPIRBuilderTest, CreateTeamsWithNumTeamsAndThreadLimit) {
};
OpenMPIRBuilder::LocationDescription Loc({Builder.saveIP(), DL});
- OpenMPIRBuilder::InsertPointTy AfterIP = *expectedToOptional(
- OMPBuilder.createTeams(Builder, BodyGenCB, NumTeamsLower, NumTeamsUpper,
- ThreadLimit, nullptr));
+ ASSERT_EXPECTED_INIT(OpenMPIRBuilder::InsertPointTy, AfterIP,
+ OMPBuilder.createTeams(Builder, BodyGenCB, NumTeamsLower,
+ NumTeamsUpper, ThreadLimit,
+ nullptr));
Builder.restoreIP(AfterIP);
Builder.CreateRetVoid();
@@ -4767,10 +4853,11 @@ TEST_F(OpenMPIRBuilderTest, CreateTeamsWithIfCondition) {
// `F` already has an integer argument, so we use that as upper bound to
// `num_teams`
- OpenMPIRBuilder::InsertPointTy AfterIP = *expectedToOptional(
- OMPBuilder.createTeams(Builder, BodyGenCB, /*NumTeamsLower=*/nullptr,
- /*NumTeamsUpper=*/nullptr,
- /*ThreadLimit=*/nullptr, IfExpr));
+ ASSERT_EXPECTED_INIT(OpenMPIRBuilder::InsertPointTy, AfterIP,
+ OMPBuilder.createTeams(Builder, BodyGenCB,
+ /*NumTeamsLower=*/nullptr,
+ /*NumTeamsUpper=*/nullptr,
+ /*ThreadLimit=*/nullptr, IfExpr));
Builder.restoreIP(AfterIP);
Builder.CreateRetVoid();
@@ -4833,9 +4920,10 @@ TEST_F(OpenMPIRBuilderTest, CreateTeamsWithIfConditionAndNumTeams) {
// `F` already has an integer argument, so we use that as upper bound to
// `num_teams`
- OpenMPIRBuilder::InsertPointTy AfterIP = *expectedToOptional(
- OMPBuilder.createTeams(Builder, BodyGenCB, NumTeamsLower, NumTeamsUpper,
- ThreadLimit, IfExpr));
+ ASSERT_EXPECTED_INIT(OpenMPIRBuilder::InsertPointTy, AfterIP,
+ OMPBuilder.createTeams(Builder, BodyGenCB, NumTeamsLower,
+ NumTeamsUpper, ThreadLimit,
+ IfExpr));
Builder.restoreIP(AfterIP);
Builder.CreateRetVoid();
@@ -5096,7 +5184,8 @@ TEST_F(OpenMPIRBuilderTest, CreateReductions) {
// Do nothing in finalization.
auto FiniCB = [&](InsertPointTy CodeGenIP) { return Error::success(); };
- OpenMPIRBuilder::InsertPointTy AfterIP = *expectedToOptional(
+ ASSERT_EXPECTED_INIT(
+ OpenMPIRBuilder::InsertPointTy, AfterIP,
OMPBuilder.createParallel(Loc, OuterAllocaIP, BodyGenCB, PrivCB, FiniCB,
/* IfCondition */ nullptr,
/* NumThreads */ nullptr, OMP_PROC_BIND_default,
@@ -5113,8 +5202,10 @@ TEST_F(OpenMPIRBuilderTest, CreateReductions) {
OMPBuilder.Config.setIsGPU(false);
bool ReduceVariableByRef[] = {false, false};
- ASSERT_TRUE(expectedToOptional(OMPBuilder.createReductions(
- BodyIP, BodyAllocaIP, ReductionInfos, ReduceVariableByRef)));
+ ASSERT_THAT_EXPECTED(OMPBuilder.createReductions(BodyIP, BodyAllocaIP,
+ ReductionInfos,
+ ReduceVariableByRef),
+ Succeeded());
Builder.restoreIP(AfterIP);
Builder.CreateRetVoid();
@@ -5352,13 +5443,15 @@ TEST_F(OpenMPIRBuilderTest, CreateTwoReductions) {
// Do nothing in finalization.
auto FiniCB = [&](InsertPointTy CodeGenIP) { return Error::success(); };
- OpenMPIRBuilder::InsertPointTy AfterIP1 = *expectedToOptional(
+ ASSERT_EXPECTED_INIT(
+ OpenMPIRBuilder::InsertPointTy, AfterIP1,
OMPBuilder.createParallel(Loc, OuterAllocaIP, FirstBodyGenCB, PrivCB,
FiniCB, /* IfCondition */ nullptr,
/* NumThreads */ nullptr, OMP_PROC_BIND_default,
/* IsCancellable */ false));
Builder.restoreIP(AfterIP1);
- OpenMPIRBuilder::InsertPointTy AfterIP2 = *expectedToOptional(
+ ASSERT_EXPECTED_INIT(
+ OpenMPIRBuilder::InsertPointTy, AfterIP2,
OMPBuilder.createParallel({Builder.saveIP(), DL}, OuterAllocaIP,
SecondBodyGenCB, PrivCB, FiniCB,
/* IfCondition */ nullptr,
@@ -5369,18 +5462,22 @@ TEST_F(OpenMPIRBuilderTest, CreateTwoReductions) {
OMPBuilder.Config.setIsGPU(false);
bool ReduceVariableByRef[] = {false};
- ASSERT_TRUE(expectedToOptional(OMPBuilder.createReductions(
- FirstBodyIP, FirstBodyAllocaIP,
- {{SumType, SumReduced, SumPrivatized,
- /*EvaluationKind=*/OpenMPIRBuilder::EvalKind::Scalar, sumReduction,
- /*ReductionGenClang=*/nullptr, sumAtomicReduction}},
- ReduceVariableByRef)));
- ASSERT_TRUE(expectedToOptional(OMPBuilder.createReductions(
- SecondBodyIP, SecondBodyAllocaIP,
- {{XorType, XorReduced, XorPrivatized,
- /*EvaluationKind=*/OpenMPIRBuilder::EvalKind::Scalar, xorReduction,
- /*ReductionGenClang=*/nullptr, xorAtomicReduction}},
- ReduceVariableByRef)));
+ ASSERT_THAT_EXPECTED(
+ OMPBuilder.createReductions(
+ FirstBodyIP, FirstBodyAllocaIP,
+ {{SumType, SumReduced, SumPrivatized,
+ /*EvaluationKind=*/OpenMPIRBuilder::EvalKind::Scalar, sumReduction,
+ /*ReductionGenClang=*/nullptr, sumAtomicReduction}},
+ ReduceVariableByRef),
+ Succeeded());
+ ASSERT_THAT_EXPECTED(
+ OMPBuilder.createReductions(
+ SecondBodyIP, SecondBodyAllocaIP,
+ {{XorType, XorReduced, XorPrivatized,
+ /*EvaluationKind=*/OpenMPIRBuilder::EvalKind::Scalar, xorReduction,
+ /*ReductionGenClang=*/nullptr, xorAtomicReduction}},
+ ReduceVariableByRef),
+ Succeeded());
Builder.restoreIP(AfterIP2);
Builder.CreateRetVoid();
@@ -5460,9 +5557,9 @@ TEST_F(OpenMPIRBuilderTest, CreateSectionsSimple) {
llvm::Value *&ReplVal) { return CodeGenIP; };
IRBuilder<>::InsertPoint AllocaIP(&F->getEntryBlock(),
F->getEntryBlock().getFirstInsertionPt());
- OpenMPIRBuilder::InsertPointTy AfterIP =
- *expectedToOptional(OMPBuilder.createSections(
- Loc, AllocaIP, SectionCBVector, PrivCB, FiniCB, false, false));
+ ASSERT_EXPECTED_INIT(OpenMPIRBuilder::InsertPointTy, AfterIP,
+ OMPBuilder.createSections(Loc, AllocaIP, SectionCBVector,
+ PrivCB, FiniCB, false, false));
Builder.restoreIP(AfterIP);
Builder.CreateRetVoid(); // Required at the end of the function
EXPECT_NE(F->getEntryBlock().getTerminator(), nullptr);
@@ -5517,9 +5614,10 @@ TEST_F(OpenMPIRBuilderTest, CreateSections) {
IRBuilder<>::InsertPoint AllocaIP(&F->getEntryBlock(),
F->getEntryBlock().getFirstInsertionPt());
- OpenMPIRBuilder::InsertPointTy AfterIP = *expectedToOptional(
- OMPBuilder.createSections(Loc, AllocaIP, SectionCBVector, PrivCB,
- FINICB_WRAPPER(FiniCB), false, false));
+ ASSERT_EXPECTED_INIT(OpenMPIRBuilder::InsertPointTy, AfterIP,
+ OMPBuilder.createSections(Loc, AllocaIP, SectionCBVector,
+ PrivCB, FINICB_WRAPPER(FiniCB),
+ false, false));
Builder.restoreIP(AfterIP);
Builder.CreateRetVoid(); // Required at the end of the function
@@ -5606,9 +5704,9 @@ TEST_F(OpenMPIRBuilderTest, CreateSectionsNoWait) {
llvm::Value *&ReplVal) { return CodeGenIP; };
auto FiniCB = [&](InsertPointTy IP) { return Error::success(); };
- OpenMPIRBuilder::InsertPointTy AfterIP =
- *expectedToOptional(OMPBuilder.createSections(
- Loc, AllocaIP, SectionCBVector, PrivCB, FiniCB, false, true));
+ ASSERT_EXPECTED_INIT(OpenMPIRBuilder::InsertPointTy, AfterIP,
+ OMPBuilder.createSections(Loc, AllocaIP, SectionCBVector,
+ PrivCB, FiniCB, false, true));
Builder.restoreIP(AfterIP);
Builder.CreateRetVoid(); // Required at the end of the function
for (auto &Inst : instructions(*F)) {
@@ -5830,8 +5928,9 @@ TEST_F(OpenMPIRBuilderTest, TargetEnterData) {
OMPBuilder.Config.setIsGPU(true);
llvm::omp::RuntimeFunction RTLFunc = OMPRTL___tgt_target_data_begin_mapper;
- OpenMPIRBuilder::InsertPointTy AfterIP =
- *expectedToOptional(OMPBuilder.createTargetData(
+ ASSERT_EXPECTED_INIT(
+ OpenMPIRBuilder::InsertPointTy, AfterIP,
+ OMPBuilder.createTargetData(
Loc, AllocaIP, Builder.saveIP(), Builder.getInt64(DeviceID),
/* IfCond= */ nullptr, Info, GenMapInfoCB, &RTLFunc));
Builder.restoreIP(AfterIP);
@@ -5891,8 +5990,9 @@ TEST_F(OpenMPIRBuilderTest, TargetExitData) {
OMPBuilder.Config.setIsGPU(true);
llvm::omp::RuntimeFunction RTLFunc = OMPRTL___tgt_target_data_end_mapper;
- OpenMPIRBuilder::InsertPointTy AfterIP =
- *expectedToOptional(OMPBuilder.createTargetData(
+ ASSERT_EXPECTED_INIT(
+ OpenMPIRBuilder::InsertPointTy, AfterIP,
+ OMPBuilder.createTargetData(
Loc, AllocaIP, Builder.saveIP(), Builder.getInt64(DeviceID),
/* IfCond= */ nullptr, Info, GenMapInfoCB, &RTLFunc));
Builder.restoreIP(AfterIP);
@@ -6001,8 +6101,9 @@ TEST_F(OpenMPIRBuilderTest, TargetDataRegion) {
return Builder.saveIP();
};
- OpenMPIRBuilder::InsertPointTy TargetDataIP1 =
- *expectedToOptional(OMPBuilder.createTargetData(
+ ASSERT_EXPECTED_INIT(
+ OpenMPIRBuilder::InsertPointTy, TargetDataIP1,
+ OMPBuilder.createTargetData(
Loc, AllocaIP, Builder.saveIP(), Builder.getInt64(DeviceID),
/* IfCond= */ nullptr, Info, GenMapInfoCB, nullptr, BodyCB));
Builder.restoreIP(TargetDataIP1);
@@ -6028,8 +6129,9 @@ TEST_F(OpenMPIRBuilderTest, TargetDataRegion) {
EXPECT_EQ(TargetDataCall, nullptr);
return Builder.saveIP();
};
- OpenMPIRBuilder::InsertPointTy TargetDataIP2 =
- *expectedToOptional(OMPBuilder.createTargetData(
+ ASSERT_EXPECTED_INIT(
+ OpenMPIRBuilder::InsertPointTy, TargetDataIP2,
+ OMPBuilder.createTargetData(
Loc, AllocaIP, Builder.saveIP(), Builder.getInt64(DeviceID),
/* IfCond= */ nullptr, Info, GenMapInfoCB, nullptr, BodyTargetCB));
Builder.restoreIP(TargetDataIP2);
@@ -6127,7 +6229,8 @@ TEST_F(OpenMPIRBuilderTest, TargetRegion) {
TargetRegionEntryInfo EntryInfo("func", 42, 4711, 17);
OpenMPIRBuilder::LocationDescription OmpLoc({Builder.saveIP(), DL});
- OpenMPIRBuilder::InsertPointTy AfterIP = *expectedToOptional(
+ ASSERT_EXPECTED_INIT(
+ OpenMPIRBuilder::InsertPointTy, AfterIP,
OMPBuilder.createTarget(OmpLoc, /*IsOffloadEntry=*/true, Builder.saveIP(),
Builder.saveIP(), EntryInfo, -1, 0, Inputs,
GenMapInfoCB, BodyGenCB, SimpleArgAccessorCB));
@@ -6237,7 +6340,8 @@ TEST_F(OpenMPIRBuilderTest, TargetRegionDevice) {
TargetRegionEntryInfo EntryInfo("parent", /*DeviceID=*/1, /*FileID=*/2,
/*Line=*/3, /*Count=*/0);
- OpenMPIRBuilder::InsertPointTy AfterIP = *expectedToOptional(
+ ASSERT_EXPECTED_INIT(
+ OpenMPIRBuilder::InsertPointTy, AfterIP,
OMPBuilder.createTarget(Loc, /*IsOffloadEntry=*/true, EntryIP, EntryIP,
EntryInfo, /*NumTeams=*/-1,
/*NumThreads=*/0, CapturedArgs, GenMapInfoCB,
@@ -6387,7 +6491,8 @@ TEST_F(OpenMPIRBuilderTest, ConstantAllocaRaise) {
TargetRegionEntryInfo EntryInfo("parent", /*DeviceID=*/1, /*FileID=*/2,
/*Line=*/3, /*Count=*/0);
- OpenMPIRBuilder::InsertPointTy AfterIP = *expectedToOptional(
+ ASSERT_EXPECTED_INIT(
+ OpenMPIRBuilder::InsertPointTy, AfterIP,
OMPBuilder.createTarget(Loc, /*IsOffloadEntry=*/true, EntryIP, EntryIP,
EntryInfo, /*NumTeams=*/-1,
/*NumThreads=*/0, CapturedArgs, GenMapInfoCB,
@@ -6511,8 +6616,9 @@ TEST_F(OpenMPIRBuilderTest, CreateTask) {
BasicBlock *BodyBB = splitBB(Builder, /*CreateBranch=*/true, "alloca.split");
OpenMPIRBuilder::LocationDescription Loc(
InsertPointTy(BodyBB, BodyBB->getFirstInsertionPt()), DL);
- OpenMPIRBuilder::InsertPointTy AfterIP =
- *expectedToOptional(OMPBuilder.createTask(
+ ASSERT_EXPECTED_INIT(
+ OpenMPIRBuilder::InsertPointTy, AfterIP,
+ OMPBuilder.createTask(
Loc, InsertPointTy(AllocaBB, AllocaBB->getFirstInsertionPt()),
BodyGenCB));
Builder.restoreIP(AfterIP);
@@ -6621,8 +6727,9 @@ TEST_F(OpenMPIRBuilderTest, CreateTaskNoArgs) {
BasicBlock *BodyBB = splitBB(Builder, /*CreateBranch=*/true, "alloca.split");
OpenMPIRBuilder::LocationDescription Loc(
InsertPointTy(BodyBB, BodyBB->getFirstInsertionPt()), DL);
- OpenMPIRBuilder::InsertPointTy AfterIP =
- *expectedToOptional(OMPBuilder.createTask(
+ ASSERT_EXPECTED_INIT(
+ OpenMPIRBuilder::InsertPointTy, AfterIP,
+ OMPBuilder.createTask(
Loc, InsertPointTy(AllocaBB, AllocaBB->getFirstInsertionPt()),
BodyGenCB));
Builder.restoreIP(AfterIP);
@@ -6654,8 +6761,9 @@ TEST_F(OpenMPIRBuilderTest, CreateTaskUntied) {
BasicBlock *BodyBB = splitBB(Builder, /*CreateBranch=*/true, "alloca.split");
OpenMPIRBuilder::LocationDescription Loc(
InsertPointTy(BodyBB, BodyBB->getFirstInsertionPt()), DL);
- OpenMPIRBuilder::InsertPointTy AfterIP =
- *expectedToOptional(OMPBuilder.createTask(
+ ASSERT_EXPECTED_INIT(
+ OpenMPIRBuilder::InsertPointTy, AfterIP,
+ OMPBuilder.createTask(
Loc, InsertPointTy(AllocaBB, AllocaBB->getFirstInsertionPt()),
BodyGenCB,
/*Tied=*/false));
@@ -6696,8 +6804,9 @@ TEST_F(OpenMPIRBuilderTest, CreateTaskDepend) {
Type::getInt32Ty(M->getContext()), InDep);
DDS.push_back(DDIn);
}
- OpenMPIRBuilder::InsertPointTy AfterIP =
- *expectedToOptional(OMPBuilder.createTask(
+ ASSERT_EXPECTED_INIT(
+ OpenMPIRBuilder::InsertPointTy, AfterIP,
+ OMPBuilder.createTask(
Loc, InsertPointTy(AllocaBB, AllocaBB->getFirstInsertionPt()),
BodyGenCB,
/*Tied=*/false, /*Final*/ nullptr, /*IfCondition*/ nullptr, DDS));
@@ -6771,9 +6880,9 @@ TEST_F(OpenMPIRBuilderTest, CreateTaskFinal) {
CmpInst::Predicate::ICMP_EQ, F->getArg(0),
ConstantInt::get(Type::getInt32Ty(M->getContext()), 0U));
OpenMPIRBuilder::LocationDescription Loc(Builder.saveIP(), DL);
- OpenMPIRBuilder::InsertPointTy AfterIP =
- *expectedToOptional(OMPBuilder.createTask(Loc, AllocaIP, BodyGenCB,
- /*Tied=*/false, Final));
+ ASSERT_EXPECTED_INIT(OpenMPIRBuilder::InsertPointTy, AfterIP,
+ OMPBuilder.createTask(Loc, AllocaIP, BodyGenCB,
+ /*Tied=*/false, Final));
Builder.restoreIP(AfterIP);
OMPBuilder.finalize();
Builder.CreateRetVoid();
@@ -6829,9 +6938,10 @@ TEST_F(OpenMPIRBuilderTest, CreateTaskIfCondition) {
CmpInst::Predicate::ICMP_EQ, F->getArg(0),
ConstantInt::get(Type::getInt32Ty(M->getContext()), 0U));
OpenMPIRBuilder::LocationDescription Loc(Builder.saveIP(), DL);
- OpenMPIRBuilder::InsertPointTy AfterIP = *expectedToOptional(
- OMPBuilder.createTask(Loc, AllocaIP, BodyGenCB,
- /*Tied=*/false, /*Final=*/nullptr, IfCondition));
+ ASSERT_EXPECTED_INIT(OpenMPIRBuilder::InsertPointTy, AfterIP,
+ OMPBuilder.createTask(Loc, AllocaIP, BodyGenCB,
+ /*Tied=*/false, /*Final=*/nullptr,
+ IfCondition));
Builder.restoreIP(AfterIP);
OMPBuilder.finalize();
Builder.CreateRetVoid();
@@ -6923,8 +7033,9 @@ TEST_F(OpenMPIRBuilderTest, CreateTaskgroup) {
BasicBlock *BodyBB = splitBB(Builder, /*CreateBranch=*/true, "alloca.split");
OpenMPIRBuilder::LocationDescription Loc(
InsertPointTy(BodyBB, BodyBB->getFirstInsertionPt()), DL);
- OpenMPIRBuilder::InsertPointTy AfterIP =
- *expectedToOptional(OMPBuilder.createTaskgroup(
+ ASSERT_EXPECTED_INIT(
+ OpenMPIRBuilder::InsertPointTy, AfterIP,
+ OMPBuilder.createTaskgroup(
Loc, InsertPointTy(AllocaBB, AllocaBB->getFirstInsertionPt()),
BodyGenCB));
Builder.restoreIP(AfterIP);
@@ -7003,8 +7114,8 @@ TEST_F(OpenMPIRBuilderTest, CreateTaskgroupWithTasks) {
return Error::success();
};
OpenMPIRBuilder::LocationDescription Loc(Builder.saveIP(), DL);
- OpenMPIRBuilder::InsertPointTy TaskIP1 = *expectedToOptional(
- OMPBuilder.createTask(Loc, AllocaIP, TaskBodyGenCB1));
+ ASSERT_EXPECTED_INIT(OpenMPIRBuilder::InsertPointTy, TaskIP1,
+ OMPBuilder.createTask(Loc, AllocaIP, TaskBodyGenCB1));
Builder.restoreIP(TaskIP1);
auto TaskBodyGenCB2 = [&](InsertPointTy AllocaIP, InsertPointTy CodeGenIP) {
@@ -7016,20 +7127,20 @@ TEST_F(OpenMPIRBuilderTest, CreateTaskgroupWithTasks) {
return Error::success();
};
OpenMPIRBuilder::LocationDescription Loc2(Builder.saveIP(), DL);
- OpenMPIRBuilder::InsertPointTy TaskIP2 = *expectedToOptional(
- OMPBuilder.createTask(Loc2, AllocaIP, TaskBodyGenCB2));
+ ASSERT_EXPECTED_INIT(OpenMPIRBuilder::InsertPointTy, TaskIP2,
+ OMPBuilder.createTask(Loc2, AllocaIP, TaskBodyGenCB2));
Builder.restoreIP(TaskIP2);
- return Error::success();
};
BasicBlock *AllocaBB = Builder.GetInsertBlock();
BasicBlock *BodyBB = splitBB(Builder, /*CreateBranch=*/true, "alloca.split");
OpenMPIRBuilder::LocationDescription Loc(
InsertPointTy(BodyBB, BodyBB->getFirstInsertionPt()), DL);
- OpenMPIRBuilder::InsertPointTy AfterIP =
- *expectedToOptional(OMPBuilder.createTaskgroup(
+ ASSERT_EXPECTED_INIT(
+ OpenMPIRBuilder::InsertPointTy, AfterIP,
+ OMPBuilder.createTaskgroup(
Loc, InsertPointTy(AllocaBB, AllocaBB->getFirstInsertionPt()),
- BodyGenCB));
+ BODYGENCB_WRAPPER(BodyGenCB)));
Builder.restoreIP(AfterIP);
OMPBuilder.finalize();
Builder.CreateRetVoid();
More information about the cfe-commits
mailing list