diff --git a/tests/transformations/tasklet_fusion_test.py b/tests/transformations/tasklet_fusion_test.py index 1e10759753..c7fd6802d5 100644 --- 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)