From 681b4133929c3904a83efa3ecad7c832c3d1df8d Mon Sep 17 00:00:00 2001
From: tj-sun <tianjiao.sun.2010@gmail.com>
Date: Wed, 28 Feb 2018 08:51:26 +0000
Subject: [PATCH] invalidate inst_ids_to_try when active group changes
---
loopy/schedule/__init__.py | 20 ++++++++++++--------
1 file changed, 12 insertions(+), 8 deletions(-)
diff --git a/loopy/schedule/__init__.py b/loopy/schedule/__init__.py
index 975671fa4..771ba7378 100644
--- a/loopy/schedule/__init__.py
+++ b/loopy/schedule/__init__.py
@@ -884,11 +884,6 @@ def generate_loop_schedules_internal(
if is_ready and debug_mode:
print("ready to schedule '%s'" % format_insn(kernel, insn.id))
- # Remove the current instruction from the sorted list of instructions
- # to try
- new_insn_ids_to_try = list(insn_ids_to_try)
- new_insn_ids_to_try.remove(insn.id)
-
if is_ready and not debug_mode:
iid_set = frozenset([insn.id])
@@ -906,14 +901,23 @@ def generate_loop_schedules_internal(
else:
new_active_group_counts[grp] = (
sched_state.group_insn_counts[grp] - 1)
- # invalidate instruction to try when active group changes
- new_insn_ids_to_try = None
-
else:
new_active_group_counts = sched_state.active_group_counts
# }}}
+ # {{{ update instruction_ids_to_try
+
+ new_insn_ids_to_try = list(insn_ids_to_try)
+ new_insn_ids_to_try.remove(insn.id)
+
+ # invalidate instruction_ids_to_try when active group changes
+ if set(new_active_group_counts.keys()) != set(
+ sched_state.active_group_counts.keys()):
+ new_insn_ids_to_try = None
+
+ # }}}
+
new_uses_of_boostability = []
if allow_boost:
if orig_have & insn.boostable_into:
--
GitLab