Searched refs:KernelDim3 (Results 1 – 4 of 4) sorted by relevance
| /external/llvm-project/mlir/lib/Dialect/GPU/IR/ |
| D | GPUDialect.cpp | 262 KernelDim3 LaunchOp::getBlockIds() { in getBlockIds()
265 return KernelDim3{args[0], args[1], args[2]}; in getBlockIds()
268 KernelDim3 LaunchOp::getThreadIds() { in getThreadIds()
271 return KernelDim3{args[3], args[4], args[5]}; in getThreadIds()
274 KernelDim3 LaunchOp::getGridSize() { in getGridSize()
277 return KernelDim3{args[6], args[7], args[8]}; in getGridSize()
280 KernelDim3 LaunchOp::getBlockSize() { in getBlockSize()
283 return KernelDim3{args[9], args[10], args[11]}; in getBlockSize()
286 KernelDim3 LaunchOp::getGridSizeOperandValues() { in getGridSizeOperandValues()
287 return KernelDim3{getOperand(0), getOperand(1), getOperand(2)}; in getGridSizeOperandValues()
[all …]
|
| /external/llvm-project/mlir/include/mlir/Dialect/GPU/ |
| D | GPUDialect.h | 33 struct KernelDim3 { struct
|
| D | GPUOps.td | 380 OpBuilderDAG<(ins "GPUFuncOp":$kernelFunc, "KernelDim3":$gridSize,
381 "KernelDim3":$blockSize, "ValueRange":$kernelOperands)>
398 KernelDim3 getGridSizeOperandValues();
401 KernelDim3 getBlockSizeOperandValues();
506 KernelDim3 getBlockIds();
508 KernelDim3 getThreadIds();
510 KernelDim3 getGridSize();
512 KernelDim3 getBlockSize();
515 KernelDim3 getGridSizeOperandValues();
517 KernelDim3 getBlockSizeOperandValues();
|
| /external/llvm-project/mlir/lib/Conversion/SCFToGPU/ |
| D | SCFToGPU.cpp | 40 static Value getDim3Value(const gpu::KernelDim3 &dim3, unsigned pos) { in getDim3Value()
|