From 259a015feb9d4ab65e8fbba32f6c777e92780cc7 Mon Sep 17 00:00:00 2001
From: Yong He <yonghe@outlook.com>
Date: Wed, 22 Mar 2023 21:16:35 -0700
Subject: Type legalization and autodiff bug fixes. (#2722)

* Bug fixes.

* Fix.

* Only perform autodiff for functions whose derivative is actually used.

* Fix loop optimize bug.

* Fix high order diff.

* Fix trivial diff func generation.

* Fixes.

* Cleanup.

---------

Co-authored-by: Yong He <yhe@nvidia.com>
---
 source/slang/slang-ir-loop-unroll.cpp | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

(limited to 'source/slang/slang-ir-loop-unroll.cpp')

diff --git a/source/slang/slang-ir-loop-unroll.cpp b/source/slang/slang-ir-loop-unroll.cpp
index 4f9b8d272..121665c85 100644
--- a/source/slang/slang-ir-loop-unroll.cpp
+++ b/source/slang/slang-ir-loop-unroll.cpp
@@ -67,8 +67,14 @@ List<IRBlock*> _collectBlocksInLoop(IRDominatorTree* dom, IRLoop* loopInst)
         {
             if (succ == breakBlock)
                 continue;
-            if (dom->dominates(firstBlock, succ) && !dom->dominates(breakBlock, succ))
-                addBlock(succ);
+            if (!dom->dominates(firstBlock, succ))
+                continue;
+            if (!as<IRUnreachable>(breakBlock->getTerminator()))
+            {
+                if (dom->dominates(breakBlock, succ))
+                    continue;
+            }
+            addBlock(succ);
         }
     }
     return loopBlocks;
-- 
cgit v1.2.3