Support custom backward derivative attribute. (#2594)

1 files changed, 9 insertions, 0 deletions

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;