Corrected taskletfusion test case

spcl · Jun 6, 2023 · 642bf2a · 642bf2a
1 parent c3f5548
commit 642bf2a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/transformations/tasklet_fusion_test.py b/tests/transformations/tasklet_fusion_test.py
@@ -185,8 +185,8 @@ def map_uses_param(A: dace.float32[10], B: dace.float32[10], C: dace.float32[10]
 
     sdfg = map_uses_param.to_sdfg(simplify=True)
 
-    num_tasklet_fusions = sdfg.apply_transformations(TaskletFusion)
-    assert (num_tasklet_fusions == 1)
+    num_tasklet_fusions = sdfg.apply_transformations_repeated(TaskletFusion)
+    assert (num_tasklet_fusions == 3)
 
     A = np.zeros([10], dtype=np.float32)
     B = np.ones([10], dtype=np.float32)