Added simplification of IfBranch terminators.

m4rs-mt · Sep 1, 2023 · 10323d0 · 10323d0
1 parent f75ecb4
commit 10323d0
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/Src/ILGPU/IR/Construction/Terminators.cs b/Src/ILGPU/IR/Construction/Terminators.cs
@@ -1,6 +1,6 @@
 // ---------------------------------------------------------------------------------------
 //                                        ILGPU
-//                        Copyright (c) 2018-2021 ILGPU Project
+//                        Copyright (c) 2018-2023 ILGPU Project
 //                                    www.ilgpu.net
 //
 // File: Terminators.cs
@@ -87,13 +87,20 @@ public Branch CreateIfBranch(
             Value condition,
             BasicBlock trueTarget,
             BasicBlock falseTarget,
-            IfBranchFlags flags) =>
-            CreateTerminator(new IfBranch(
+            IfBranchFlags flags)
+        {
+            // Simplify unnecessary if branches and fold them to unconditional branches
+            if (trueTarget == falseTarget)
+                return CreateBranch(location, trueTarget);
+
+            // Create an if branch in all other cases
+            return CreateTerminator(new IfBranch(
                 GetInitializer(location),
                 condition,
                 trueTarget,
                 falseTarget,
                 flags));
+        }
 
         /// <summary>
         /// Creates a switch terminator builder.