More renaming in jvp pass. (#2475)

Co-authored-by: Yong He <yhe@nvidia.com>

1 files changed, 4 insertions, 4 deletions

diff --git a/source/slang/slang-ir-specialize.cpp b/source/slang/slang-ir-specialize.cpp
index 53ea99a0c..406e5157c 100644
--- a/source/slang/slang-ir-specialize.cpp
+++ b/source/slang/slang-ir-specialize.cpp
@@ -388,15 +388,15 @@ struct SpecializationContext
             auto genericReturnVal = findInnerMostGenericReturnVal(genericVal);
             if (genericReturnVal->findDecoration<IRTargetIntrinsicDecoration>())
             {
-                if (auto customDiffRef = genericReturnVal->findDecoration<IRJVPDerivativeReferenceDecoration>())
+                if (auto customDiffRef = genericReturnVal->findDecoration<IRForwardDerivativeDecoration>())
                 {
                     // If we already have a diff func on this specialize, skip.
-                    if (auto specDiffRef = specInst->findDecoration<IRJVPDerivativeReferenceDecoration>())
+                    if (auto specDiffRef = specInst->findDecoration<IRForwardDerivativeDecoration>())
                     {
                         return false;
                     }
 
-                    auto specDiffFunc = as<IRSpecialize>(customDiffRef->getJVPFunc());
+                    auto specDiffFunc = as<IRSpecialize>(customDiffRef->getForwardDerivativeFunc());
 
                     // If the base is specialized, the JVP version must be also be a specialized
                     // generic.
@@ -436,7 +436,7 @@ struct SpecializationContext
                     addToWorkList(newDiffFuncType);
                     addToWorkList(newDiffFunc);
 
-                    builder.addJVPDerivativeReferenceDecoration(specInst, newDiffFunc);
+                    builder.addForwardDerivativeDecoration(specInst, newDiffFunc);
 
                     return true;
                 }