Support custom backward derivative attribute. (#2594)

1 files changed, 2 insertions, 2 deletions

diff --git a/source/slang/slang-check-modifier.cpp b/source/slang/slang-check-modifier.cpp
index f505b1321..f3623f19f 100644
--- a/source/slang/slang-check-modifier.cpp
+++ b/source/slang/slang-check-modifier.cpp
@@ -613,7 +613,7 @@ namespace Slang
 
             hitObjectAttributesAttr->location = (int32_t)val->value;
         }
-        else if (auto forwardDerivativeAttr = as<ForwardDerivativeAttribute>(attr))
+        else if (auto derivativeAttr = as<UserDefinedDerivativeAttribute>(attr))
         {
             SLANG_ASSERT(attr->args.getCount() == 1);
             SLANG_ASSERT(as<Decl>(attrTarget));
@@ -633,7 +633,7 @@ namespace Slang
             // 
             // Set type to null to indicate that this needs expr needs to be further resolved.
             diffExpr->type.type = nullptr;
-            forwardDerivativeAttr->funcExpr = diffExpr;
+            derivativeAttr->funcExpr = diffExpr;
         }
         else if (auto derivativeOfAttr = as<DerivativeOfAttribute>(attr))
         {