| /llvm-project-15.0.7/mlir/lib/Dialect/SPIRV/Transforms/ |
| H A D | UnifyAliasedResourcePass.cpp | 47 if (varOp->getAttrOfType<UnitAttr>("aliased")) { in collectAliasedResources()
48 Optional<uint32_t> set = varOp.descriptor_set(); in collectAliasedResources()
49 Optional<uint32_t> binding = varOp.binding(); in collectAliasedResources()
51 aliasedResources[{*set, *binding}].push_back(varOp); in collectAliasedResources()
207 auto canonicalOp = getCanonicalResource(varOp); in shouldUnify()
208 return canonicalOp && varOp != canonicalOp; in shouldUnify()
213 return shouldUnify(varOp); in shouldUnify()
235 spirv::GlobalVariableOp varOp) const { in getCanonicalResource()
236 auto descriptorIt = descriptorMap.find(varOp); in getCanonicalResource()
244 auto it = elementTypeMap.find(varOp); in getElementType()
[all …]
|
| H A D | SPIRVConversion.cpp | 602 for (auto varOp : body.getOps<spirv::GlobalVariableOp>()) { in getBuiltinVariable() local
603 if (auto builtinAttr = varOp->getAttrOfType<StringAttr>( in getBuiltinVariable()
608 return varOp; in getBuiltinVariable()
624 if (auto varOp = getBuiltinVariable(body, builtin)) in getOrInsertBuiltinVariable() local
625 return varOp; in getOrInsertBuiltinVariable()
671 spirv::GlobalVariableOp varOp = in getBuiltinVariableValue() local
697 for (auto varOp : body.getOps<spirv::GlobalVariableOp>()) { in getPushConstantVariable() local
698 auto ptrType = varOp.type().dyn_cast<spirv::PointerType>(); in getPushConstantVariable()
712 return varOp; in getPushConstantVariable()
725 return varOp; in getOrInsertPushConstantVariable()
[all …]
|
| H A D | DecorateCompositeTypeLayoutPass.cpp | 68 auto varOp = spirvModule.lookupSymbol<spirv::GlobalVariableOp>(varName); in matchAndRewrite() local
71 op, varOp.type(), SymbolRefAttr::get(varName.getAttr())); in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Conversion/TensorToSPIRV/ |
| H A D | TensorToSPIRV.cpp | 69 spirv::VariableOp varOp; in matchAndRewrite() local
71 varOp = rewriter.create<spirv::VariableOp>( in matchAndRewrite()
85 auto acOp = rewriter.create<spirv::AccessChainOp>(loc, varOp, index); in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Target/SPIRV/Serialization/ |
| H A D | SerializeOps.cpp | 308 Serializer::processGlobalVariableOp(spirv::GlobalVariableOp varOp) { in processGlobalVariableOp() argument
312 if (failed(processType(varOp.getLoc(), varOp.type(), resultTypeID))) { in processGlobalVariableOp()
322 auto varName = varOp.sym_name(); in processGlobalVariableOp()
331 operands.push_back(static_cast<uint32_t>(varOp.storageClass())); in processGlobalVariableOp()
334 if (auto initializer = varOp.initializer()) { in processGlobalVariableOp()
337 return emitError(varOp.getLoc(), in processGlobalVariableOp()
344 if (failed(emitDebugLine(typesGlobalValues, varOp.getLoc()))) in processGlobalVariableOp()
350 for (auto attr : varOp->getAttrs()) { in processGlobalVariableOp()
356 if (failed(processDecoration(varOp.getLoc(), resultID, attr))) { in processGlobalVariableOp()
|
| H A D | Serializer.h | 134 LogicalResult processGlobalVariableOp(spirv::GlobalVariableOp varOp);
|
| /llvm-project-15.0.7/mlir/lib/Target/SPIRV/Deserialization/ |
| H A D | DeserializeOps.cpp | 46 if (auto varOp = getGlobalVariable(id)) { in getValue() local
48 unknownLoc, varOp.type(), SymbolRefAttr::get(varOp.getOperation())); in getValue()
|
| H A D | Deserializer.cpp | 605 auto varOp = opBuilder.create<spirv::GlobalVariableOp>( in processGlobalVariable() local
612 varOp->setAttr(attr.getName(), attr.getValue()); in processGlobalVariable()
614 globalVariableMap[variableID] = varOp; in processGlobalVariable()
|
| /llvm-project-15.0.7/mlir/lib/Conversion/MemRefToSPIRV/ |
| H A D | MemRefToSPIRV.cpp | 280 spirv::GlobalVariableOp varOp; in matchAndRewrite() local
289 varOp = rewriter.create<spirv::GlobalVariableOp>(loc, spirvType, varName, in matchAndRewrite()
294 rewriter.replaceOpWithNewOp<spirv::AddressOfOp>(operation, varOp); in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Conversion/SPIRVToLLVM/ |
| H A D | SPIRVToLLVM.cpp | 1252 matchAndRewrite(spirv::VariableOp varOp, OpAdaptor adaptor, in matchAndRewrite() argument
1254 auto srcType = varOp.getType(); in matchAndRewrite()
1257 auto init = varOp.initializer(); in matchAndRewrite()
1265 Location loc = varOp.getLoc(); in matchAndRewrite()
1268 rewriter.replaceOpWithNewOp<LLVM::AllocaOp>(varOp, dstType, size); in matchAndRewrite()
1273 rewriter.replaceOp(varOp, allocated); in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/SPIRV/IR/ |
| H A D | SPIRVOps.cpp | 1122 auto varOp = dyn_cast_or_null<spirv::GlobalVariableOp>( in verify() local
1125 if (!varOp) { in verify()
1128 if (pointer().getType() != varOp.type()) { in verify()
|