Pool inst worklists and hashsets to avoid rehashing. (#2982)

Co-authored-by: Yong He <yhe@nvidia.com>

1 files changed, 3 insertions, 3 deletions

diff --git a/source/slang/slang-ir-check-differentiability.cpp b/source/slang/slang-ir-check-differentiability.cpp
index 6c9280fca..d4b93be5e 100644
--- a/source/slang/slang-ir-check-differentiability.cpp
+++ b/source/slang/slang-ir-check-differentiability.cpp
@@ -261,9 +261,9 @@ public:
         // `produceDiffSet`.
         // `expectDiffSet` is a set of insts that expects their operands to produce a diff. It is an error
         // if they don't.
-        HashSet<IRInst*> produceDiffSet;
-        HashSet<IRInst*> expectDiffSet;
-        HashSet<IRInst*> carryNonTrivialDiffSet;
+        InstHashSet produceDiffSet(funcInst->getModule());
+        InstHashSet expectDiffSet(funcInst->getModule());
+        InstHashSet carryNonTrivialDiffSet(funcInst->getModule());
 
         int differentiableOutputs = 0;
         bool isDifferentiableReturnType = false;