diff options
Diffstat (limited to 'source/slang/slang-ir-specialize.cpp')
| -rw-r--r-- | source/slang/slang-ir-specialize.cpp | 8 |
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; } |