Searched refs:waitOp (Results 1 – 6 of 6) sorted by relevance
| /external/llvm-project/mlir/lib/Dialect/GPU/Transforms/ |
| D | AsyncRegionRewriter.cpp | 129 if (auto waitOp = dyn_cast<gpu::WaitOp>(op)) { in operator ()() local
130 if (!waitOp.asyncToken()) in operator ()()
131 worklist.push_back(waitOp); in operator ()()
142 auto waitOp = worklist[i]; in ~DeferWaitCallback() local
143 auto executeOp = waitOp->getParentOfType<async::ExecuteOp>(); in ~DeferWaitCallback()
144 auto numDependencies = waitOp.asyncDependencies().size(); in ~DeferWaitCallback()
149 waitOp.asyncDependencies()); in ~DeferWaitCallback()
150 waitOp.erase(); in ~DeferWaitCallback()
246 auto waitOp = builder.create<gpu::WaitOp>(loc, Type{}, tokens); in addAsyncDependencyAfter() local
253 worklist.push_back(waitOp); in addAsyncDependencyAfter()
|
| /external/llvm-project/mlir/lib/Conversion/GPUCommon/ |
| D | ConvertLaunchFuncToRuntimeCalls.cpp | 208 matchAndRewrite(gpu::WaitOp waitOp, ArrayRef<Value> operands,
222 matchAndRewrite(gpu::WaitOp waitOp, ArrayRef<Value> operands,
414 gpu::WaitOp waitOp, ArrayRef<Value> operands, in matchAndRewrite() argument
416 if (waitOp.asyncToken()) in matchAndRewrite()
417 return rewriter.notifyMatchFailure(waitOp, "Cannot convert async op."); in matchAndRewrite()
419 Location loc = waitOp.getLoc(); in matchAndRewrite()
426 rewriter.eraseOp(waitOp); in matchAndRewrite()
437 gpu::WaitOp waitOp, ArrayRef<Value> operands, in matchAndRewrite() argument
439 if (!waitOp.asyncToken()) in matchAndRewrite()
440 return rewriter.notifyMatchFailure(waitOp, "Can only convert async op."); in matchAndRewrite()
[all …]
|
| /external/llvm-project/mlir/lib/Dialect/OpenACC/IR/ |
| D | OpenACC.cpp | 773 static LogicalResult verify(acc::WaitOp waitOp) { in verify() argument
776 if (waitOp.asyncOperand() && waitOp.async()) in verify()
777 return waitOp.emitError("async attribute cannot appear with asyncOperand"); in verify()
779 if (waitOp.waitDevnum() && waitOp.waitOperands().empty()) in verify()
780 return waitOp.emitError("wait_devnum cannot appear without waitOperands"); in verify()
|
| /external/llvm-project/mlir/lib/Transforms/ |
| D | PipelineDataTransfer.cpp | 141 static bool checkTagMatch(AffineDmaStartOp startOp, AffineDmaWaitOp waitOp) { in checkTagMatch() argument
142 if (startOp.getTagMemRef() != waitOp.getTagMemRef()) in checkTagMatch()
145 auto waitIndices = waitOp.getTagIndices(); in checkTagMatch()
|
| /external/llvm-project/flang/lib/Lower/ |
| D | OpenACC.cpp | 987 auto waitOp = createSimpleOp<mlir::acc::WaitOp>(firOpBuilder, currentLocation, in genACC() local
991 waitOp.asyncAttr(firOpBuilder.getUnitAttr()); in genACC()
|
| /external/skqp/src/gpu/ |
| D | GrRenderTargetContext.cpp | 1655 std::unique_ptr<GrOp> waitOp(GrSemaphoreOp::MakeWait(fContext, sema, in waitOnSemaphores() local
1657 this->getRTOpList()->addOp(std::move(waitOp), *this->caps()); in waitOnSemaphores()
|