Searched refs:dynamicSharedMemorySize (Results 1 – 4 of 4) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Dialect/GPU/IR/ |
| H A D | GPUDialect.cpp | 395 Value dynamicSharedMemorySize, Type asyncTokenType, in build() argument
404 if (dynamicSharedMemorySize) in build()
405 result.addOperands(dynamicSharedMemorySize); in build()
417 segmentSizes.back() = dynamicSharedMemorySize ? 1 : 0; in build()
515 if (dynamicSharedMemorySize()) in print()
517 << dynamicSharedMemorySize(); in print()
603 OpAsmParser::UnresolvedOperand dynamicSharedMemorySize; in parse() local
608 if (parser.parseOperand(dynamicSharedMemorySize) || in parse()
609 parser.resolveOperand(dynamicSharedMemorySize, in parse()
699 if (dynamicSharedMemorySize) in build()
[all …]
|
| /llvm-project-15.0.7/mlir/lib/Conversion/GPUCommon/ |
| H A D | GPUToLLVMConversion.cpp | 771 Value dynamicSharedMemorySize = launchOp.dynamicSharedMemorySize() in matchAndRewrite() local
772 ? launchOp.dynamicSharedMemorySize() in matchAndRewrite()
778 adaptor.blockSizeZ(), dynamicSharedMemorySize, stream, kernelParams, in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/GPU/Transforms/ |
| H A D | KernelOutlining.cpp | 231 launchOp.getBlockSizeOperandValues(), launchOp.dynamicSharedMemorySize(), in convertToLaunchFuncOp()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/GPU/IR/ |
| H A D | GPUOps.td | 340 Optional<I32>:$dynamicSharedMemorySize,
429 "KernelDim3":$blockSize, "Value":$dynamicSharedMemorySize,
468 (`dynamic_shared_memory_size` $dynamicSharedMemorySize^)?
480 Optional<I32>:$dynamicSharedMemorySize)>,
565 CArg<"Value", "nullptr">:$dynamicSharedMemorySize,
|