Move force inlining step to before `processAutodiffCalls` (and run in loop) (#3217)

* Move auto-diff force inlining step to before `processAutodiffCalls`

* Fix `replaceUsesWith` to handle existing inst defined after current use.

* Fix.

---------

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

1 files changed, 6 insertions, 0 deletions

diff --git a/source/slang/slang-ir-inline.cpp b/source/slang/slang-ir-inline.cpp
index 06b63db52..1d308c507 100644
--- a/source/slang/slang-ir-inline.cpp
+++ b/source/slang/slang-ir-inline.cpp
@@ -872,6 +872,12 @@ bool performPreAutoDiffForceInlining(IRGlobalValueWithCode* func)
     return pass.considerAllCallSitesRec(func);
 }
 
+bool performPreAutoDiffForceInlining(IRModule* module)
+{
+    PreAutoDiffForceInliningPass pass(module);
+    return pass.considerAllCallSitesRec(module->getModuleInst());
+}
+
     // Defined in slang-ir-specialize-resource.cpp
 bool isResourceType(IRType* type);
 bool isIllegalGLSLParameterType(IRType* type);