Searched refs:kernelOperands (Results 1 – 3 of 3) sorted by relevance
| /llvm-project/mlir/lib/Conversion/SPIRVToLLVM/ |
| H A D | ConvertLaunchFuncToLLVMCalls.cpp | 216 auto kernelOperands = adaptor.getOperands().take_back(numKernelOperands); in matchAndRewrite() local
217 for (const auto &operand : llvm::enumerate(kernelOperands)) { in matchAndRewrite()
|
| /llvm-project/mlir/lib/Dialect/GPU/IR/ |
| H A D | GPUDialect.cpp | 1130 ValueRange kernelOperands, Type asyncTokenType, in build()
1146 result.addOperands(kernelOperands); in build()
1163 static_cast<int32_t>(kernelOperands.size()); in build()
1170 ValueRange kernelOperands, Type asyncTokenType, in build()
1178 dynamicSharedMemorySize, kernelOperands, asyncTokenType, in build()
1185 ValueRange kernelOperands, Value asyncObject, in build()
1194 result.addOperands(kernelOperands);
1212 static_cast<int32_t>(kernelOperands.size());
1101 build(OpBuilder & builder,OperationState & result,SymbolRefAttr kernelSymbol,KernelDim3 gridSize,KernelDim3 getBlockSize,Value dynamicSharedMemorySize,ValueRange kernelOperands,Type asyncTokenType,ValueRange asyncDependencies,std::optional<KernelDim3> clusterSize) build() argument
1141 build(OpBuilder & builder,OperationState & result,GPUFuncOp kernelFunc,KernelDim3 gridSize,KernelDim3 getBlockSize,Value dynamicSharedMemorySize,ValueRange kernelOperands,Type asyncTokenType,ValueRange asyncDependencies,std::optional<KernelDim3> clusterSize) build() argument
1156 build(OpBuilder & builder,OperationState & result,SymbolRefAttr kernel,KernelDim3 gridSize,KernelDim3 getBlockSize,Value dynamicSharedMemorySize,ValueRange kernelOperands,Value asyncObject,std::optional<KernelDim3> clusterSize) build() argument
|
| /llvm-project/mlir/include/mlir/Dialect/GPU/IR/ |
| H A D | GPUOps.td | 623 Variadic<AnyType>:$kernelOperands,
729 "ValueRange":$kernelOperands,
735 "ValueRange":$kernelOperands,
741 "ValueRange":$kernelOperands,
791 custom<LaunchFuncOperands>($kernelOperands, type($kernelOperands)) attr-dict
|