From 46a4d98baa1d43b33717b4377aefeeaf46b9c2ff Mon Sep 17 00:00:00 2001
From: Yong He <yonghe@outlook.com>
Date: Mon, 23 Jan 2023 06:59:25 -0800
Subject: Full address insts elimination for backward autodiff. (#2604)

Co-authored-by: Yong He <yhe@nvidia.com>
---
 source/slang/slang-ir-simplify-cfg.cpp | 4 ++++
 1 file changed, 4 insertions(+)

(limited to 'source/slang/slang-ir-simplify-cfg.cpp')

diff --git a/source/slang/slang-ir-simplify-cfg.cpp b/source/slang/slang-ir-simplify-cfg.cpp
index 1e247d1d9..54a1f7e08 100644
--- a/source/slang/slang-ir-simplify-cfg.cpp
+++ b/source/slang/slang-ir-simplify-cfg.cpp
@@ -196,6 +196,10 @@ bool simplifyCFG(IRModule* module)
     bool changed = false;
     for (auto inst : module->getGlobalInsts())
     {
+        if (auto genericInst = as<IRGeneric>(inst))
+        {
+            inst = findGenericReturnVal(genericInst);
+        }
         if (auto func = as<IRFunc>(inst))
         {
             changed |= processFunc(func);
-- 
cgit v1.2.3