fix to_sdfg in test

spcl · Oct 29, 2024 · 109eeba · 109eeba
1 parent 3216513
commit 109eeba
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/transformations/raise_and_duplicate_test.py b/tests/transformations/raise_and_duplicate_test.py
@@ -22,7 +22,7 @@ def program(flag: dace.bool, in_arr: dace.float64[N], arr: dace.float64[N]):
         for i in dace.map[0:N]:
                 arr[i] = tmp2[i]
 
-    sdfg = program.to_sdfg()
+    sdfg = program.to_sdfg(simplify=True)
     sdfg.apply_transformations([IfRaising, StateReplication])
     sdfg.simplify()
     sdfg.apply_transformations_repeated([OTFMapFusion])
@@ -49,7 +49,7 @@ def program(arr: dace.float64[N]):
         else:
             return 0
 
-    sdfg = program.to_sdfg()
+    sdfg = program.to_sdfg(simplify=True)
 
     transform = IfRaising()
     transform.if_guard = sdfg.start_state