diff --git a/loopy/fusion.py b/loopy/fusion.py index 21b6a703383fed8b2014276d104c1e83d9aaaae4..8845951ea293d5a0e66d457a4bcb8680db57623c 100644 --- a/loopy/fusion.py +++ b/loopy/fusion.py @@ -218,9 +218,9 @@ def _fuse_two_kernels(knla, knlb): assump_a, assump_b = isl.align_two(assump_a, assump_b) shared_param_names = list( - set(dom_a.get_var_dict(dim_type.set)) + set(assump_a.get_var_dict(dim_type.set)) & - set(dom_b.get_var_dict(dim_type.set))) + set(assump_b.get_var_dict(dim_type.set))) assump_a_s = assump_a.project_out_except(shared_param_names, [dim_type.param]) assump_b_s = assump_a.project_out_except(shared_param_names, [dim_type.param])