Merge pull request #486 from stkaplan/index-compare

Don't check tensor when comparing IndexSets
tensor-compiler · Aug 5, 2021 · 0ede002 · 0ede002
2 parents ef99f8d + 4e01734
commit 0ede002
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/src/index_notation/index_notation.cpp b/src/index_notation/index_notation.cpp
@@ -189,11 +189,20 @@ struct Isomorphic : public IndexNotationVisitorStrict {
       }
     }
     if (anode->isAccessingStructure != bnode->isAccessingStructure ||
-        anode->windowedModes != bnode->windowedModes ||
-        anode->indexSetModes != bnode->indexSetModes) {
+        anode->windowedModes != bnode->windowedModes) {
+      eq = false;
+      return;
+    }
+    if (anode->indexSetModes.size() != bnode->indexSetModes.size()) {
       eq = false;
       return;
     }
+    for (auto aset = anode->indexSetModes.begin(), bset = bnode->indexSetModes.begin(); aset != anode->indexSetModes.end(); ++aset, ++bset) {
+      if (aset->first != bset->first || *aset->second.set != *bset->second.set) {
+        eq = false;
+        return;
+      }
+    }
     eq = true;
   }