Home
last modified time | relevance | path

Searched refs:asyncDependencies (Results 1 – 8 of 8) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/GPU/IR/
H A DGPUDialect.cpp287 return parser.parseOperandList(asyncDependencies, in parseAsyncDependencies()
299 if (asyncDependencies.empty()) in printAsyncDependencies()
304 llvm::interleaveComma(asyncDependencies, printer); in printAsyncDependencies()
397 result.addOperands(asyncDependencies); in build()
416 segmentSizes.front() = asyncDependencies.size(); in build()
505 if (!asyncDependencies().empty()) in print()
506 p << " [" << asyncDependencies() << ']'; in print()
636 segmentSizes.front() = asyncDependencies.size(); in parse()
692 result.addOperands(asyncDependencies); in build()
708 segmentSizes.front() = asyncDependencies.size(); in build()
[all …]
H A DInferIntRangeInterfaceImpls.cpp86 argRanges = argRanges.drop_front(asyncDependencies().size()); in inferResultRanges()
/llvm-project-15.0.7/mlir/lib/Conversion/GPUCommon/
H A DGPUToLLVMConversion.cpp468 auto stream = adaptor.asyncDependencies().front(); in matchAndRewrite()
498 Value stream = adaptor.asyncDependencies().front(); in matchAndRewrite()
593 llvm::zip(waitOp.asyncDependencies(), adaptor.getOperands())) { in matchAndRewrite()
721 if (launchOp.asyncDependencies().size() > 1) in matchAndRewrite()
728 if (!launchOp.asyncToken() && !launchOp.asyncDependencies().empty()) in matchAndRewrite()
765 adaptor.asyncDependencies().empty() in matchAndRewrite()
767 : adaptor.asyncDependencies().front(); in matchAndRewrite()
825 auto stream = adaptor.asyncDependencies().front(); in matchAndRewrite()
859 auto stream = adaptor.asyncDependencies().front(); in matchAndRewrite()
/llvm-project-15.0.7/mlir/include/mlir/Dialect/GPU/IR/
H A DGPUOps.td336 Arguments<(ins Variadic<GPU_AsyncToken>:$asyncDependencies,
432 CArg<"ValueRange", "{}">:$asyncDependencies)>
477 Arguments<(ins Variadic<GPU_AsyncToken>:$asyncDependencies,
567 CArg<"ValueRange", "{}">:$asyncDependencies)>
901 let arguments = (ins Variadic<GPU_AsyncToken>:$asyncDependencies);
935 let arguments = (ins Variadic<GPU_AsyncToken>:$asyncDependencies,
976 let arguments = (ins Variadic<GPU_AsyncToken>:$asyncDependencies,
981 custom<AsyncDependencies>(type($asyncToken), $asyncDependencies)
1007 let arguments = (ins Variadic<GPU_AsyncToken>:$asyncDependencies,
1013 custom<AsyncDependencies>(type($asyncToken), $asyncDependencies)
[all …]
H A DGPUBase.td103 return op.asyncDependencies();
/llvm-project-15.0.7/mlir/include/mlir/Dialect/NVGPU/IR/
H A DNVGPU.td225 let arguments = (ins NVGPU_DeviceAsyncToken:$asyncDependencies,
228 $asyncDependencies attr-dict
/llvm-project-15.0.7/mlir/lib/Dialect/GPU/Transforms/
H A DAsyncRegionRewriter.cpp205 SmallVector<Value, 4> dependencies = waitOp.asyncDependencies(); in ~DeferWaitCallback()
H A DKernelOutlining.cpp233 launchOp.asyncDependencies()); in convertToLaunchFuncOp()