From 1c9b33157322751c456bf7abbd386edccf4413c3 Mon Sep 17 00:00:00 2001 From: Yong He Date: Sat, 14 Jan 2023 22:50:57 -0800 Subject: Support custom backward derivative attribute. (#2594) --- source/slang/slang-ir.cpp | 9 +++++++++ 1 file changed, 9 insertions(+) (limited to 'source/slang/slang-ir.cpp') diff --git a/source/slang/slang-ir.cpp b/source/slang/slang-ir.cpp index b36a2ebec..e400d0a17 100644 --- a/source/slang/slang-ir.cpp +++ b/source/slang/slang-ir.cpp @@ -6671,6 +6671,8 @@ namespace Slang case kIROp_ForwardDifferentiate: case kIROp_BackwardDifferentiate: + case kIROp_BackwardDifferentiatePrimal: + case kIROp_BackwardDifferentiatePropagate: return false; } @@ -6815,6 +6817,13 @@ namespace Slang return nullptr; } + IRInst* maybeFindOuterGeneric(IRInst* inst) + { + auto outerGeneric = findOuterGeneric(inst); + if (!outerGeneric) return inst; + return outerGeneric; + } + IRInst* findOuterMostGeneric(IRInst* inst) { IRInst* currInst = inst; -- cgit v1.2.3