Replace dots only in the for-loop header.

spcl · Sep 27, 2023 · 2eda9da · 2eda9da
1 parent 6490b2f
commit 2eda9da
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/dace/codegen/control_flow.py b/dace/codegen/control_flow.py
@@ -391,10 +391,10 @@ def as_cpp(self, codegen, symbols) -> str:
             update = f'{self.itervar} = {self.update}'
 
         expr = f'{preinit}\nfor ({init}; {cond}; {update}) {{\n'
-        expr += _clean_loop_body(self.body.as_cpp(codegen, symbols))
-        expr += '\n}\n'
         # TODO: Check that the dot is used to access struct members
         expr = expr.replace('.', '->')
+        expr += _clean_loop_body(self.body.as_cpp(codegen, symbols))
+        expr += '\n}\n'
         return expr
 
     @property