Reduce number of calls to symbols_defined_at

spcl · Sep 18, 2024 · 98655d6 · 98655d6
1 parent 696f5bc
commit 98655d6
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/dace/codegen/targets/cpu.py b/dace/codegen/targets/cpu.py
@@ -1930,8 +1930,9 @@ def _generate_MapEntry(
                 if var in map_param_types.keys():
                     var_type = map_param_types[var]
                 else:
-                    var_type = dtypes.result_type_of(infer_expr_type(r[0], state_dfg.symbols_defined_at(node)), 
-                                                     infer_expr_type(r[1], state_dfg.symbols_defined_at(node)))
+                    defined_symbols = state_dfg.symbols_defined_at(node)
+                    var_type = dtypes.result_type_of(infer_expr_type(r[0], defined_symbols),
+                                                     infer_expr_type(r[1], defined_symbols))
 
                 begin, end, skip = r