Searched defs:gridSize (Results 1 – 2 of 2) sorted by relevance
| /llvm-project/mlir/lib/Dialect/GPU/IR/ |
| H A D | GPUDialect.cpp | 1099 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
1139 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
1154 build(OpBuilder & builder,OperationState & result,SymbolRefAttr kernel,KernelDim3 gridSize,KernelDim3 getBlockSize,Value dynamicSharedMemorySize,ValueRange kernelOperands,Value asyncObject,std::optional<KernelDim3> clusterSize) build() argument
[all...] |
| /llvm-project/mlir/lib/Dialect/SparseTensor/Transforms/ |
| H A D | SparseGPUCodegen.cpp | 99 gpu::KernelDim3 gridSize = {one, one, one}; in genLaunchGPUFunc() local
|