Name update fixes

spcl · Dec 3, 2024 · 21dd0c3 · 21dd0c3
1 parent c597e24
commit 21dd0c3
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/dace/sdfg/sdfg.py b/dace/sdfg/sdfg.py
@@ -773,9 +773,11 @@ def replace_dict(self,
 
         # Update size descriptors
         # Return_size break things delete it from the arrays
+        # Changes symbol names might not related to arrays
         for arr_name, size_desc_name in size_desc_map.items():
-            arr = self.arrays[arr_name]
-            arr.size_desc_name = size_desc_name if size_desc_name != "__return_size" else None
+            arr = self.arrays[arr_name] if arr_name in self.arrays else None
+            if arr is not None:
+                arr.size_desc_name = size_desc_name if size_desc_name != "__return_size" else None
 
         # Replace inside data descriptors
         for array in self.arrays.values():