From 86fc50c5092fbccf6072dcf7bbdfafb8915f02c8 Mon Sep 17 00:00:00 2001 From: Yong He Date: Wed, 8 Mar 2023 21:52:34 -0800 Subject: Add support for `[PrimalSubstitute]` and `[PrimalSubstituteOf]`. (#2691) * Add support for `[PrimalSubstitute]` and `[PrimalSubstituteOf]`. * Fix * Fix. * Cleanup. --------- Co-authored-by: Yong He --- source/slang/slang-ir-check-differentiability.cpp | 15 +++++++++++++-- 1 file changed, 13 insertions(+), 2 deletions(-) (limited to 'source/slang/slang-ir-check-differentiability.cpp') diff --git a/source/slang/slang-ir-check-differentiability.cpp b/source/slang/slang-ir-check-differentiability.cpp index 186b0cc03..14f6394e2 100644 --- a/source/slang/slang-ir-check-differentiability.cpp +++ b/source/slang/slang-ir-check-differentiability.cpp @@ -39,12 +39,17 @@ public: return false; } - bool _isDifferentiableFuncImpl(IRInst* func, DifferentiableLevel level) { func = getResolvedInstForDecorations(func); if (!func) return false; + if (auto substDecor = func->findDecoration()) + { + func = getResolvedInstForDecorations(substDecor->getPrimalSubstituteFunc()); + if (!func) + return false; + } for (auto decorations : func->getDecorations()) { @@ -84,7 +89,13 @@ public: if (!func) return false; - + if (auto substDecor = func->findDecoration()) + { + func = getResolvedInstForDecorations(substDecor->getPrimalSubstituteFunc()); + if (!func) + return false; + } + if (auto existingLevel = differentiableFunctions.TryGetValue(func)) return *existingLevel >= level; -- cgit v1.2.3