WIP

Src/ILGPU/Backends/OpenCL/CLCodeGenerator.cs

@@ -199,7 +199,7 @@ protected static string GetParameterName(Parameter parameter) =>
 
         private int labelCounter;
         private readonly Dictionary<BasicBlock, string> blockLookup =
-            new Dictionary<BasicBlock, string>();
+            new Dictionary<BasicBlock, string>(new BasicBlock.Comparer());
         private readonly string labelPrefix;
 
         private StringBuilder prefixBuilder = new StringBuilder();
@@ -439,7 +439,8 @@ protected void GenerateCodeInternal()
                 blockLookup.Add(block, DeclareLabel());
 
             // Find all phi nodes, allocate target registers and setup internal mapping
-            var phiMapping = new Dictionary<BasicBlock, List<Variable>>();
+            var phiMapping = new Dictionary<BasicBlock, List<Variable>>(
+                new BasicBlock.Comparer());
             var dominators = Method.Blocks.CreateDominators();
             var phiBindings = PhiBindings.Create(
                 blocks,

Src/ILGPU/Backends/PTX/PTXCodeGenerator.cs

@@ -286,7 +286,8 @@ protected static string GetParameterName(Parameter parameter) =>
         #region Instance
 
         private int labelCounter;
-        private readonly Dictionary<BasicBlock, string> blockLookup = new();
+        private readonly Dictionary<BasicBlock, string> blockLookup = new(
+            new BasicBlock.Comparer());
 
         private readonly Dictionary<(Encoding, string), string> stringConstants = new();
         private readonly PhiBindings phiBindings;

Src/ILGPU/Backends/Velocity/VelocityCodeGenerator.cs

@@ -96,7 +96,8 @@ public readonly record struct GeneratorArgs(
         /// <summary>
         /// Maps blocks to labels.
         /// </summary>
-        private readonly Dictionary<BasicBlock, ILLabel> blockLookup = new();
+        private readonly Dictionary<BasicBlock, ILLabel> blockLookup =
+            new(new BasicBlock.Comparer());
 
         /// <summary>
         /// The masks analysis holding information about the masks being required.

Src/ILGPU/Frontend/Block.CFGBuilder.cs

@@ -71,7 +71,7 @@ public readonly void Apply(ILInstruction instruction, int offset) =>
             private readonly Dictionary<int, Block> blockMapping =
                 new Dictionary<int, Block>();
             private readonly Dictionary<BasicBlock, Block> basicBlockMapping =
-                new Dictionary<BasicBlock, Block>();
+                new Dictionary<BasicBlock, Block>(new BasicBlock.Comparer());
             private readonly Dictionary<Block, List<Block>> successorMapping =
                 new Dictionary<Block, List<Block>>();
 

Src/ILGPU/IR/Analyses/Loops.cs

@@ -757,8 +757,8 @@ private void RegisterLoop(
             // Initialize all lists and sets
             var headers = InlineList<BasicBlock>.Create(2);
             var breakers = InlineList<BasicBlock>.Create(2);
-            var entryBlocks = new HashSet<BasicBlock>();
-            var exitBlocks = new HashSet<BasicBlock>();
+            var entryBlocks = new HashSet<BasicBlock>(new BasicBlock.Comparer());
+            var exitBlocks = new HashSet<BasicBlock>(new BasicBlock.Comparer());
 
             // Gather all loop entries and exists
             foreach (var member in members)

Src/ILGPU/IR/BasicBlockCollection.cs

@@ -339,7 +339,7 @@ public readonly HashSet<BasicBlock> ToSet() =>
         public readonly HashSet<BasicBlock> ToSet<TPredicate>(TPredicate predicate)
             where TPredicate : InlineList.IPredicate<BasicBlock>
         {
-            var result = new HashSet<BasicBlock>();
+            var result = new HashSet<BasicBlock>(new BasicBlock.Comparer());
             foreach (var block in this)
             {
                 if (predicate.Apply(block))

Src/ILGPU/IR/BasicBlockMapping.cs

@@ -746,7 +746,7 @@ partial struct BasicBlockSet
 
         /// <summary cref="InitBlockSet"/>
         [MemberNotNull(nameof(blockSet))]
-        partial void InitBlockSet() => blockSet = new HashSet<BasicBlock>();
+        partial void InitBlockSet() => blockSet = new(new BasicBlock.Comparer());
 
         #endregion
 
@@ -785,7 +785,8 @@ partial struct BasicBlockMap<T>
 
         /// <summary cref="InitBlockMap"/>
         [MemberNotNull(nameof(blockMap))]
-        partial void InitBlockMap() => blockMap = new Dictionary<BasicBlock, T>();
+        partial void InitBlockMap() => blockMap =
+            new Dictionary<BasicBlock, T>(new BasicBlock.Comparer());
 
         #endregion
 

Src/ILGPU/IR/Verifier.cs

@@ -184,11 +184,11 @@ private sealed class ControlFlowVerifier : VerifierBase
             private readonly List<BasicBlock> blocksInRpo =
                 new List<BasicBlock>();
             private readonly HashSet<BasicBlock> containedBlocks =
-                new HashSet<BasicBlock>();
+                new(new BasicBlock.Comparer());
             private readonly Dictionary<BasicBlock, HashSet<BasicBlock>> predecessors =
-                new Dictionary<BasicBlock, HashSet<BasicBlock>>();
+                new(new BasicBlock.Comparer());
             private readonly Dictionary<BasicBlock, HashSet<BasicBlock>> successors =
-                new Dictionary<BasicBlock, HashSet<BasicBlock>>();
+                new(new BasicBlock.Comparer());
 
             /// <summary>
             /// Constructs a new control-flow verifier.
@@ -210,9 +210,9 @@ public ControlFlowVerifier(Method method, VerificationResult result)
             private void CreateLinkSets(BasicBlock block)
             {
                 if (!predecessors.ContainsKey(block))
-                    predecessors.Add(block, new HashSet<BasicBlock>());
+                    predecessors.Add(block, new(new BasicBlock.Comparer()));
                 if (!successors.ContainsKey(block))
-                    successors.Add(block, new HashSet<BasicBlock>());
+                    successors.Add(block, new(new BasicBlock.Comparer()));
             }
 
             /// <summary>
@@ -325,7 +325,7 @@ private sealed class ValueVerifier : VerifierBase
 
             private readonly HashSet<Value> values = new HashSet<Value>();
             private readonly Dictionary<BasicBlock, HashSet<Value>> mapping =
-                new Dictionary<BasicBlock, HashSet<Value>>();
+                new Dictionary<BasicBlock, HashSet<Value>>(new BasicBlock.Comparer());
 
             /// <summary>
             /// Constructs a new value verifier.
@@ -465,7 +465,7 @@ private void VerifyPhis() =>
                         phiValue.BasicBlock.Predecessors.Length);
 
                     // Verify nodes and sources
-                    var visited = new HashSet<BasicBlock>();
+                    var visited = new HashSet<BasicBlock>(new BasicBlock.Comparer());
                     for (int i = 0, e = phiValue.Nodes.Length; i < e; ++i)
                     {
                         Value value = phiValue.Nodes[i];