[clang] [llvm] [llvm][opt][Transforms][SPIR-V] Enable `InferAddressSpaces` for SPIR-V (PR #110897)
Matt Arsenault via cfe-commits
cfe-commits at lists.llvm.org
Sun Dec 8 07:53:08 PST 2024
================
@@ -92,6 +98,65 @@ SPIRVTargetMachine::SPIRVTargetMachine(const Target &T, const Triple &TT,
setRequiresStructuredCFG(false);
}
+enum AddressSpace {
+ Function = storageClassToAddressSpace(SPIRV::StorageClass::Function),
+ CrossWorkgroup =
+ storageClassToAddressSpace(SPIRV::StorageClass::CrossWorkgroup),
+ UniformConstant =
+ storageClassToAddressSpace(SPIRV::StorageClass::UniformConstant),
+ Workgroup = storageClassToAddressSpace(SPIRV::StorageClass::Workgroup),
+ Generic = storageClassToAddressSpace(SPIRV::StorageClass::Generic)
+};
+
+unsigned SPIRVTargetMachine::getAssumedAddrSpace(const Value *V) const {
+ // TODO: we only enable this for AMDGCN flavoured SPIR-V, where we know it to
+ // be correct; this might be relaxed in the future.
+ if (getTargetTriple().getVendor() != Triple::VendorType::AMD)
----------------
arsenm wrote:
Please move this whole hook to a separate PR. I also do not think we should have any vendor checks
https://github.com/llvm/llvm-project/pull/110897
More information about the cfe-commits
mailing list