From 11fa1a90635f5715ac296e4a361bfcb802fce739 Mon Sep 17 00:00:00 2001 From: Alex Fikl Date: Wed, 18 Apr 2018 16:47:12 -0500 Subject: [PATCH] tools: remove some globalargs that loopy can infer --- sumpy/tools.py | 11 ++++------- 1 file changed, 4 insertions(+), 7 deletions(-) diff --git a/sumpy/tools.py b/sumpy/tools.py index b891da0e..9d73d1a1 100644 --- a/sumpy/tools.py +++ b/sumpy/tools.py @@ -347,6 +347,7 @@ class MatrixBlockIndex(object): ], name="block_cumsum_knl", default_offset=lp.auto, + assumptions="nranges>=1", lang_version=MOST_RECENT_LANGUAGE_VERSION) loopy_knl = lp.realize_reduction(loopy_knl, force_scan=True, @@ -397,19 +398,15 @@ class MatrixBlockIndex(object): end """, [ - lp.GlobalArg("srcindices", None), - lp.GlobalArg("tgtindices", None), - lp.GlobalArg("srcranges", None, shape="nranges + 1"), - lp.GlobalArg("tgtranges", None, shape="nranges + 1"), - lp.GlobalArg("blkranges", None, shape="nranges + 1"), + lp.GlobalArg('blkranges', None, shape="nranges + 1"), lp.GlobalArg("rowindices", None, shape="nresults"), lp.GlobalArg("colindices", None, shape="nresults"), lp.ValueArg("nresults", None), - lp.ValueArg("nranges", None), - "..." + '...' ], name="block_index_knl", default_offset=lp.auto, + assumptions='nranges>=1', silenced_warnings="write_race(write_index*)", lang_version=MOST_RECENT_LANGUAGE_VERSION) loopy_knl = lp.split_iname(loopy_knl, "irange", 128, outer_tag="g.0") -- GitLab