From 04e25ed32bdd402c494f65d2b17f91019e8325c3 Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Sat, 26 Jun 2021 23:59:40 -0500
Subject: [PATCH] make_loopy_program: allow passing tags
---
arraycontext/loopy.py | 5 +++--
1 file changed, 3 insertions(+), 2 deletions(-)
diff --git a/arraycontext/loopy.py b/arraycontext/loopy.py
index 8f2816d..f4c9775 100644
--- a/arraycontext/loopy.py
+++ b/arraycontext/loopy.py
@@ -39,7 +39,7 @@ _DEFAULT_LOOPY_OPTIONS = lp.Options(
def make_loopy_program(domains, statements, kernel_data=None,
- name="mm_actx_kernel"):
+ name="mm_actx_kernel", tags=None):
"""Return a :class:`loopy.LoopKernel` suitable for use with
:meth:`ArrayContext.call_loopy`.
"""
@@ -53,7 +53,8 @@ def make_loopy_program(domains, statements, kernel_data=None,
options=_DEFAULT_LOOPY_OPTIONS,
default_offset=lp.auto,
name=name,
- lang_version=MOST_RECENT_LANGUAGE_VERSION)
+ lang_version=MOST_RECENT_LANGUAGE_VERSION,
+ tags=tags)
def get_default_entrypoint(t_unit):
--
GitLab