Home
last modified time | relevance | path

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

/llvm-project/mlir/include/mlir/Dialect/GPU/IR/
H A DGPUOps.td625 Results<(outs Optional<GPU_AsyncToken>:$asyncToken)> {
783 custom<AsyncDependencies>(type($asyncToken), $asyncDependencies)
809 Results<(outs Optional<GPU_AsyncToken>:$asyncToken)> {
1574 let results = (outs Optional<GPU_AsyncToken>:$asyncToken);
1577 custom<AsyncDependencies>(type($asyncToken), $asyncDependencies) attr-dict
1614 Optional<GPU_AsyncToken>:$asyncToken);
1621 custom<AsyncDependencies>(type($asyncToken), $asyncDependencies) (` ` `host_shared` $hostShared^)? ` `
1654 let results = (outs Optional<GPU_AsyncToken>:$asyncToken);
1657 custom<AsyncDependencies>(type($asyncToken), $asyncDependencies)
1686 let results = (outs Optional<GPU_AsyncToken>:$asyncToken);
[all...]
/llvm-project/mlir/lib/Dialect/Async/Transforms/
H A DAsyncToAsyncRuntime.cpp91 std::optional<Value> asyncToken; // returned completion token member
251 machinery.asyncToken = retToken; in setupCoroMachinery()
275 if (coro.asyncToken) in setupSetErrorBlock()
276 builder.create<RuntimeSetErrorOp>(*coro.asyncToken); in setupSetErrorBlock()
530 if (coro.asyncToken) in matchAndRewrite()
532 rewriter.create<RuntimeSetAvailableOp>(loc, *coro.asyncToken); in matchAndRewrite()
722 if (coro.asyncToken) in matchAndRewrite()
724 rewriter.create<RuntimeSetAvailableOp>(loc, *coro.asyncToken); in matchAndRewrite()
/llvm-project/mlir/test/Dialect/GPU/
H A Dcanonicalize.mlir32 %memref, %asyncToken = gpu.alloc async [%0] () : memref<5xf16>
52 %memref, %asyncToken = gpu.alloc async [%2] () : memref<2xf16>
76 %memref, %asyncToken = gpu.alloc async [%2] () : memref<2xf16>
94 %memref, %asyncToken = gpu.alloc async [%2] () : memref<2xf16>
183 %memref, %asyncToken = gpu.alloc async [%1] () : memref<2x16x16xf32>
/llvm-project/mlir/lib/Dialect/GPU/Transforms/
H A DKernelOutlining.cpp288 Value asyncToken = launchOp.getAsyncToken(); in convertToLaunchFuncOp() local
295 asyncToken ? asyncToken.getType() : nullptr, in convertToLaunchFuncOp()
H A DAsyncRegionRewriter.cpp239 // Add the `asyncToken` as dependency as needed after `op`.
251 for (auto asyncToken : asyncTokens) in addAsyncDependencyAfter() local
253 builder.create<async::AwaitOp>(loc, asyncToken).getResult()); in addAsyncDependencyAfter()
/llvm-project/mlir/test/Dialect/NVGPU/
H A Dtmaload-transform.mlir15 %memref, %asyncToken = gpu.alloc async [%0] () : memref<64x32xf32>
/llvm-project/mlir/test/Integration/GPU/CUDA/sm90/
H A Dtma_load_128x64_swizzle128b.mlir67 %memref, %asyncToken = gpu.alloc async [%0] () : !lhs
H A Dtma_load_64x8_8x128_noswizzle.mlir62 %memref, %asyncToken = gpu.alloc async [%0] () : memref<64x8xf32>
H A Dtma_load_64x64_swizzle128b.mlir84 %d_glbmem_lhs, %asyncToken = gpu.alloc async [%0] () : !lhs
/llvm-project/mlir/test/Integration/GPU/CUDA/sm90/transform-dialect/
H A Dtma_load_64x8_8x128_noswizzle-transform.mlir82 %memref, %asyncToken = gpu.alloc async [%0] () : memref<64x8xf32>
/llvm-project/mlir/include/mlir/Dialect/NVGPU/IR/
H A DNVGPU.td438 let results = (outs NVGPU_DeviceAsyncToken:$asyncToken);
473 let results = (outs NVGPU_DeviceAsyncToken:$asyncToken);