From 2ba4b536320420ba4d1b6513d3d727b37692f972 Mon Sep 17 00:00:00 2001 From: James Stevens Date: Thu, 25 Jun 2015 16:48:43 -0500 Subject: [PATCH] checking for consecutive global mem access (in progress) --- loopy/statistics.py | 16 ++++++++++++++-- 1 file changed, 14 insertions(+), 2 deletions(-) diff --git a/loopy/statistics.py b/loopy/statistics.py index ab161a77b..7775c8a8f 100755 --- a/loopy/statistics.py +++ b/loopy/statistics.py @@ -239,12 +239,24 @@ class SubscriptCounter(CombineMapper): #print("my_inames: ", my_inames) #print("iname_to_tag: ", self.knl.iname_to_tag) + from loopy.kernel.data import LocalIndexTag + local_id0 = None # TODO can there be two? for iname in my_inames: # find local id0 through self.knl.index_to_tag - print("iname: ", iname, "; tag: ", self.knl.iname_to_tag.get(iname)) - pass + tag = self.knl.iname_to_tag.get(iname) + if isinstance(tag, LocalIndexTag): + local_id0 = iname + + if local_id0 is None: + print("TESTING: no local id found, assume sequential access") + # TODO assume sequential for now? + else: + print("TESTING: local id found: ", local_id0) """ + # If you don't have a local id0 + # -> not stride1 (for now) + for dim_tag, axis_index in zip(index, array.dim_tags): # check if he contains the lid 0 guy -- GitLab