Searched refs:localSize (Results 1 – 4 of 4) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Conversion/LinalgToSPIRV/ |
| H A D | LinalgToSPIRV.cpp | 122 DenseIntElementsAttr localSize = spirv::lookupLocalWorkGroupSize(genericOp); in matchAndRewrite() local
123 if (!localSize) in matchAndRewrite()
126 if ((*localSize.begin()).getSExtValue() != originalInputType.getDimSize(0)) in matchAndRewrite()
128 if (llvm::any_of(llvm::drop_begin(localSize.getValues<APInt>(), 1), in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/SPIRV/IR/ |
| H A D | TargetAndABI.cpp | 122 spirv::getEntryPointABIAttr(ArrayRef<int32_t> localSize, MLIRContext *context) { in getEntryPointABIAttr() argument
123 if (localSize.empty()) in getEntryPointABIAttr()
126 assert(localSize.size() == 3); in getEntryPointABIAttr()
129 VectorType::get(3, IntegerType::get(context, 32)), localSize) in getEntryPointABIAttr()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/SPIRV/IR/ |
| H A D | TargetAndABI.h | 89 EntryPointABIAttr getEntryPointABIAttr(ArrayRef<int32_t> localSize,
|
| /llvm-project-15.0.7/mlir/lib/Dialect/SPIRV/Transforms/ |
| H A D | LowerABIAttributesPass.cpp | 141 SmallVector<int32_t, 3> localSize(values); in lowerEntryPointABIAttr() local
143 funcOp.getLoc(), funcOp, spirv::ExecutionMode::LocalSize, localSize); in lowerEntryPointABIAttr()
|