Searched refs:asyncToken (Results 1 – 6 of 6) sorted by relevance
| /external/llvm-project/mlir/lib/Dialect/GPU/Transforms/ |
| D | AsyncRegionRewriter.cpp | 106 return builder.create<gpu::WaitOp>(loc, resultType, operands).asyncToken(); in createWaitOp()
130 if (!waitOp.asyncToken()) in operator ()()
214 for (auto asyncToken : asyncTokens) in addAsyncDependencyAfter() local
216 builder.create<async::AwaitOp>(loc, asyncToken).result()); in addAsyncDependencyAfter()
|
| /external/llvm-project/mlir/lib/Conversion/GPUCommon/ |
| D | ConvertLaunchFuncToRuntimeCalls.cpp | 416 if (waitOp.asyncToken()) in matchAndRewrite()
439 if (!waitOp.asyncToken()) in matchAndRewrite()
578 if (!launchOp.asyncToken() && !launchOp.asyncDependencies().empty()) in matchAndRewrite()
630 if (launchOp.asyncToken()) { in matchAndRewrite()
|
| /external/llvm-project/mlir/include/mlir/Dialect/GPU/ |
| D | GPUOps.td | 301 Results<(outs Optional<GPU_AsyncToken>:$asyncToken)> {
418 custom<AsyncDependencies>(type($asyncToken), $asyncDependencies)
800 let results = (outs Optional<GPU_AsyncToken>:$asyncToken);
803 custom<AsyncDependencies>(type($asyncToken), $asyncDependencies) attr-dict
835 Optional<GPU_AsyncToken>:$asyncToken);
842 custom<AsyncDependencies>(type($asyncToken), $asyncDependencies) ` `
874 let results = (outs Optional<GPU_AsyncToken>:$asyncToken);
877 custom<AsyncDependencies>(type($asyncToken), $asyncDependencies)
|
| D | GPUBase.td | 99 return op.asyncToken().template dyn_cast_or_null<OpResult>();
|
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/ |
| D | tf_kernel_to_llvm_pass.cc | 147 if (!launch_op.asyncDependencies().empty() || launch_op.asyncToken()) { in matchAndRewrite()
|
| /external/llvm-project/mlir/lib/Conversion/AsyncToLLVM/ |
| D | AsyncToLLVM.cpp | 255 Value asyncToken; member
543 builder.create<CallOp>(loc, kEmplaceToken, TypeRange(), coro.asyncToken); in outlineExecuteOp()
|