From c5a453e56985022deb820cbbb2ff5cd6a8347e34 Mon Sep 17 00:00:00 2001 From: Yong He Date: Sat, 1 Jun 2024 19:26:14 -0700 Subject: Support different SPIRV versions. (#4254) --- source/slang/slang-lower-to-ir.cpp | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) (limited to 'source/slang/slang-lower-to-ir.cpp') diff --git a/source/slang/slang-lower-to-ir.cpp b/source/slang/slang-lower-to-ir.cpp index 80e73fe2b..403e2a921 100644 --- a/source/slang/slang-lower-to-ir.cpp +++ b/source/slang/slang-lower-to-ir.cpp @@ -11599,6 +11599,9 @@ RefPtr TargetProgram::createIRModuleForLayout(DiagnosticSink* sink) builder->addLayoutDecoration(irModule->getModuleInst(), irGlobalScopeVarLayout); + auto latestSpirvAtom = getLatestSpirvAtom(); + auto latestMetalAtom = getLatestMetalAtom(); + for( auto entryPointLayout : programLayout->entryPoints ) { auto funcDeclRef = entryPointLayout->entryPoint; @@ -11617,6 +11620,19 @@ RefPtr TargetProgram::createIRModuleForLayout(DiagnosticSink* sink) builder->addImportDecoration(irFunc, getMangledName(astBuilder, funcDeclRef).getUnownedSlice()); } + for (auto atomSet : as(funcDeclRef.getDecl())->inferredCapabilityRequirements.getAtomSets()) + { + for (auto atomVal : atomSet) + { + auto atom = (CapabilityName)atomVal; + if (atom >= CapabilityName::spirv_1_0 && atom <= latestSpirvAtom || + atom >= CapabilityName::metallib_2_3 && atom <= latestMetalAtom) + { + builder->addRequireCapabilityAtomDecoration(irFunc, atom); + } + } + } + auto irEntryPointLayout = lowerEntryPointLayout(context, entryPointLayout); builder->addLayoutDecoration(irFunc, irEntryPointLayout); -- cgit v1.2.3