Don't inline callees with custom derivative before autodiff. (#3196)

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

1 files changed, 2 insertions, 1 deletions

diff --git a/source/slang/slang-ir-autodiff-rev.cpp b/source/slang/slang-ir-autodiff-rev.cpp
index 8d7582373..681c69cd3 100644
--- a/source/slang/slang-ir-autodiff-rev.cpp
+++ b/source/slang/slang-ir-autodiff-rev.cpp
@@ -351,6 +351,7 @@ namespace Slang
             builder.setInsertBefore(diffFunc->getFirstDecorationOrChild());
             cloneInst(&cloneEnv, &builder, dictDecor);
         }
+        copyOriginalDecorations(origFunc, diffFunc);
         builder.addFloatingModeOverrideDecoration(diffFunc, FloatingPointMode::Fast);
         return InstPair(primalFunc, diffFunc);
     }
@@ -532,7 +533,7 @@ namespace Slang
     {
         removeLinkageDecorations(func);
 
-        performForceInlining(func);
+        performPreAutoDiffForceInlining(func);
 
         DifferentiableTypeConformanceContext diffTypeContext(autoDiffSharedContext);
         diffTypeContext.setFunc(func);