Searched refs:waitOp (Results 1 – 5 of 5) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Dialect/GPU/Transforms/ |
| H A D | AsyncRegionRewriter.cpp | 69 if (auto waitOp = llvm::dyn_cast<gpu::WaitOp>(op)) { in visit() local
71 waitOp.addAsyncDependency(currentToken); in visit()
72 currentToken = waitOp.asyncToken(); in visit()
188 if (auto waitOp = dyn_cast<gpu::WaitOp>(op)) { in operator ()() local
189 if (!waitOp.asyncToken()) in operator ()()
190 worklist.push_back(waitOp); in operator ()()
201 auto waitOp = worklist[i]; in ~DeferWaitCallback() local
202 auto executeOp = waitOp->getParentOfType<async::ExecuteOp>(); in ~DeferWaitCallback()
206 waitOp.erase(); in ~DeferWaitCallback()
277 auto waitOp = builder.create<gpu::WaitOp>(loc, Type{}, tokens); in addAsyncDependencyAfter() local
[all …]
|
| /llvm-project-15.0.7/mlir/lib/Conversion/GPUCommon/ |
| H A D | GPUToLLVMConversion.cpp | 258 matchAndRewrite(gpu::WaitOp waitOp, OpAdaptor adaptor,
272 matchAndRewrite(gpu::WaitOp waitOp, OpAdaptor adaptor,
553 gpu::WaitOp waitOp, OpAdaptor adaptor, in matchAndRewrite() argument
555 if (waitOp.asyncToken()) in matchAndRewrite()
558 Location loc = waitOp.getLoc(); in matchAndRewrite()
573 rewriter.eraseOp(waitOp); in matchAndRewrite()
583 gpu::WaitOp waitOp, OpAdaptor adaptor, in matchAndRewrite() argument
585 if (!waitOp.asyncToken()) in matchAndRewrite()
588 Location loc = waitOp.getLoc(); in matchAndRewrite()
593 llvm::zip(waitOp.asyncDependencies(), adaptor.getOperands())) { in matchAndRewrite()
[all …]
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Affine/Transforms/ |
| H A D | PipelineDataTransfer.cpp | 140 static bool checkTagMatch(AffineDmaStartOp startOp, AffineDmaWaitOp waitOp) { in checkTagMatch() argument
141 if (startOp.getTagMemRef() != waitOp.getTagMemRef()) in checkTagMatch()
144 auto waitIndices = waitOp.getTagIndices(); in checkTagMatch()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/GPU/IR/ |
| H A D | GPUDialect.cpp | 1287 auto waitOp = value.getDefiningOp<WaitOp>(); in matchAndRewrite() local
1288 return waitOp && waitOp->getNumOperands() == 0; in matchAndRewrite()
|
| /llvm-project-15.0.7/flang/lib/Lower/ |
| H A D | OpenACC.cpp | 1008 mlir::acc::WaitOp waitOp = createSimpleOp<mlir::acc::WaitOp>( in genACC() local
1012 waitOp.asyncAttr(firOpBuilder.getUnitAttr()); in genACC()
|