Support custom backward derivative attribute. (#2594)

1 files changed, 1 insertions, 14 deletions

diff --git a/source/slang/slang-ir-check-differentiability.cpp b/source/slang/slang-ir-check-differentiability.cpp
index 83351d07b..8413e7e79 100644
--- a/source/slang/slang-ir-check-differentiability.cpp
+++ b/source/slang/slang-ir-check-differentiability.cpp
@@ -71,6 +71,7 @@ public:
             {
             case kIROp_ForwardDerivativeDecoration:
             case kIROp_ForwardDifferentiableDecoration:
+            case kIROp_UserDefinedBackwardDerivativeDecoration:
             case kIROp_BackwardDerivativeDecoration:
             case kIROp_BackwardDifferentiableDecoration:
                 return true;
@@ -140,20 +141,6 @@ public:
         return false;
     }
 
-    bool isBackwardDifferentiableFunc(IRInst* func)
-    {
-        for (auto decorations : func->getDecorations())
-        {
-            switch (decorations->getOp())
-            {
-            case kIROp_BackwardDerivativeDecoration:
-            case kIROp_BackwardDifferentiableDecoration:
-                return true;
-            }
-        }
-        return false;
-    }
-
     bool isDifferentiableType(DifferentiableTypeConformanceContext& context, IRInst* typeInst)
     {
         HashSet<IRInst*> processedSet;