Add insn_count_subgroups_upper_bound warning in all our Loo.py kernel generation functions.

Author: sv2518

firedrake/extrusion_utils.py

@@ -219,7 +219,8 @@ def _get_lp_domains(_inames, _extents):
         raise NotImplementedError('Unsupported extrusion type "%s"' % extrusion_type)
 
     ast = lp.make_function(domains, instructions, data, name=name, target=target,
-                           seq_dependencies=True, silenced_warnings=["summing_if_branches_ops"])
+                           seq_dependencies=True, silenced_warnings=["summing_if_branches_ops",
+                                                                     "insn_count_subgroups_upper_bound"])
     kernel = op2.Kernel(ast, name)
     op2.ParLoop(kernel,
                 ext_coords.cell_set,

firedrake/parloops.py

@@ -129,7 +129,8 @@ def _form_loopy_kernel(kernel_domains, instructions, measure, args, **kwargs):
     except KeyError:
         kargs.append(...)
         knl = loopy.make_function(kernel_domains, instructions, kargs, name="par_loop_kernel", target=target,
-                                  seq_dependencies=True, silenced_warnings=["summing_if_branches_ops"])
+                                  seq_dependencies=True, silenced_warnings=["summing_if_branches_ops",
+                                                                            "insn_count_subgroups_upper_bound"])
         knl = op2.Kernel(knl, "par_loop_kernel", **kwargs)
         if kernel_cache is not None:
             return kernel_cache.setdefault(key, knl)

firedrake/slate/slac/utils.py

@@ -354,7 +354,8 @@ def merge_loopy(slate_loopy, output_arg, builder, var2terminal, name):
     preamble = preamble_init+preamble if preamble else []
     slate_wrapper = lp.make_function(domains, insns, loopy_args, name=name,
                                      seq_dependencies=True, target=target,
-                                     lang_version=(2018, 2), preambles=preamble)
+                                     lang_version=(2018, 2), preambles=preamble,
+                                     silenced_warnings=["insn_count_subgroups_upper_bound"])
 
     # Generate program from kernel, so that one can register kernels
     from pyop2.codegen.loopycompat import _match_caller_callee_argument_dimension_