From 29bd9565f4865f3a639ab142f246f370f0adaec7 Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Wed, 12 Aug 2015 14:40:59 -0500
Subject: [PATCH] Fix assumption merging in fusion
---
loopy/fusion.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/loopy/fusion.py b/loopy/fusion.py
index 21b6a7033..8845951ea 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])
--
GitLab