Diagnose on storing differentiable value into non-differentiable location. (#2681)

1 files changed, 23 insertions, 0 deletions

diff --git a/source/slang/slang-ir-autodiff.cpp b/source/slang/slang-ir-autodiff.cpp
index b630b798d..fcfbf3bee 100644
--- a/source/slang/slang-ir-autodiff.cpp
+++ b/source/slang/slang-ir-autodiff.cpp
@@ -933,6 +933,27 @@ bool processAutodiffCalls(
     return modified;
 }
 
+struct RemoveDetachInstsPass : InstPassBase
+{
+    RemoveDetachInstsPass(IRModule* module) :
+        InstPassBase(module)
+    {
+    }
+    void processModule()
+    {
+        processInstsOfType<IRDetachDerivative>(kIROp_DetachDerivative, [&](IRDetachDerivative* detach)
+            {
+                detach->replaceUsesWith(detach->getBase());
+            });
+    }
+};
+
+void removeDetachInsts(IRModule* module)
+{
+    RemoveDetachInstsPass pass(module);
+    pass.processModule();
+}
+
 bool finalizeAutoDiffPass(IRModule* module)
 {
     bool modified = false;
@@ -947,6 +968,8 @@ bool finalizeAutoDiffPass(IRModule* module)
     // 
     modified |= processPairTypes(&autodiffContext);
 
+    removeDetachInsts(module);
+
     stripNoDiffTypeAttribute(module);
 
     // Remove auto-diff related decorations.