From e844a1066b33e23c96e31708ec9f3b1a4a91c4d5 Mon Sep 17 00:00:00 2001 From: Kaushik Kulkarni Date: Sun, 14 Nov 2021 15:33:18 -0600 Subject: [PATCH] [precompute] Makes the compute insn's deps. precise Depend on an instruction only if it writes to the variables that the compute instruction reads. --- loopy/transform/precompute.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/loopy/transform/precompute.py b/loopy/transform/precompute.py index 7c20d7a01..31f0057c8 100644 --- a/loopy/transform/precompute.py +++ b/loopy/transform/precompute.py @@ -248,8 +248,7 @@ def map_kernel(self, kernel): dep_insn = kernel.id_to_insn[dep] if (frozenset(dep_insn.assignee_var_names()) & self.compute_read_variables): - self.compute_insn_depends_on.update( - insn.depends_on - excluded_insn_ids) + self.compute_insn_depends_on.add(dep) new_insns.append(insn)