Home
last modified time | relevance | path

Searched refs:asyncToken (Results 1 – 9 of 9) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/GPU/Transforms/
H A DAsyncRegionRewriter.cpp72 currentToken = waitOp.asyncToken(); in visit()
127 return builder.create<gpu::WaitOp>(loc, resultType, operands).asyncToken(); in createWaitOp()
189 if (!waitOp.asyncToken()) in operator ()()
243 for (auto asyncToken : asyncTokens) in addAsyncDependencyAfter() local
245 builder.create<async::AwaitOp>(loc, asyncToken).result()); in addAsyncDependencyAfter()
H A DKernelOutlining.cpp228 Value asyncToken = launchOp.asyncToken(); in convertToLaunchFuncOp() local
232 operands, asyncToken ? asyncToken.getType() : nullptr, in convertToLaunchFuncOp()
/llvm-project-15.0.7/mlir/test/Dialect/GPU/
H A Dcanonicalize.mlir18 %memref, %asyncToken = gpu.alloc async [%0] () : memref<5xf16>
36 %memref, %asyncToken = gpu.alloc async [%2] () : memref<2xf16>
58 %memref, %asyncToken = gpu.alloc async [%2] () : memref<2xf16>
74 %memref, %asyncToken = gpu.alloc async [%2] () : memref<2xf16>
146 %memref, %asyncToken = gpu.alloc async [%1] () : memref<2x16x16xf32>
/llvm-project-15.0.7/mlir/lib/Dialect/GPU/IR/
H A DGPUDialect.cpp484 if (getNumResults() == 0 && asyncToken()) in verifyRegions()
503 if (asyncToken()) { in print()
1136 ((op.asyncDependencies().empty() && op.asyncToken()) || in matchAndRewrite()
1137 (!op.asyncDependencies().empty() && !op.asyncToken()))) in matchAndRewrite()
1318 if (op.asyncDependencies().empty() && !op.asyncToken()) { in matchAndRewrite()
1323 if (llvm::hasSingleElement(op.asyncDependencies()) && op.asyncToken()) { in matchAndRewrite()
1328 if (op.asyncToken() && op.asyncToken().use_empty()) { in matchAndRewrite()
/llvm-project-15.0.7/mlir/include/mlir/Dialect/GPU/IR/
H A DGPUOps.td342 Results<(outs Optional<GPU_AsyncToken>:$asyncToken)> {
464 custom<AsyncDependencies>(type($asyncToken), $asyncDependencies)
481 Results<(outs Optional<GPU_AsyncToken>:$asyncToken)> {
902 let results = (outs Optional<GPU_AsyncToken>:$asyncToken);
938 Optional<GPU_AsyncToken>:$asyncToken);
978 let results = (outs Optional<GPU_AsyncToken>:$asyncToken);
981 custom<AsyncDependencies>(type($asyncToken), $asyncDependencies)
1010 let results = (outs Optional<GPU_AsyncToken>:$asyncToken);
1013 custom<AsyncDependencies>(type($asyncToken), $asyncDependencies)
1046 let results = (outs Optional<GPU_AsyncToken>:$asyncToken);
[all …]
H A DGPUBase.td119 return op.asyncToken().template dyn_cast_or_null<OpResult>();
/llvm-project-15.0.7/mlir/lib/Dialect/Async/Transforms/
H A DAsyncToAsyncRuntime.cpp68 Value asyncToken; // token representing completion of the async region member
201 machinery.asyncToken = retToken; in setupCoroMachinery()
224 builder.create<RuntimeSetErrorOp>(coro.asyncToken); in setupSetErrorBlock()
554 rewriter.replaceOpWithNewOp<RuntimeSetAvailableOp>(op, coro.asyncToken); in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Conversion/GPUCommon/
H A DGPUToLLVMConversion.cpp555 if (waitOp.asyncToken()) in matchAndRewrite()
585 if (!waitOp.asyncToken()) in matchAndRewrite()
728 if (!launchOp.asyncToken() && !launchOp.asyncDependencies().empty()) in matchAndRewrite()
781 if (launchOp.asyncToken()) { in matchAndRewrite()
/llvm-project-15.0.7/mlir/include/mlir/Dialect/NVGPU/IR/
H A DNVGPU.td170 let results = (outs NVGPU_DeviceAsyncToken:$asyncToken);
204 let results = (outs NVGPU_DeviceAsyncToken:$asyncToken);