From 24fde8e43a3880c7fec1841f894f09be40c8d9c8 Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Wed, 29 May 2013 08:11:12 -0400
Subject: [PATCH] Support Ellipsis instead of "..." string as guessed argument
marker.
---
loopy/kernel/creation.py | 7 ++++---
1 file changed, 4 insertions(+), 3 deletions(-)
diff --git a/loopy/kernel/creation.py b/loopy/kernel/creation.py
index ed0faf8ea..3050c9b5b 100644
--- a/loopy/kernel/creation.py
+++ b/loopy/kernel/creation.py
@@ -368,11 +368,12 @@ class IndexRankFinder(WalkMapper):
def guess_kernel_args_if_requested(domains, instructions, temporary_variables,
subst_rules, kernel_args, default_offset):
- if "..." not in kernel_args:
+ # Ellipsis is syntactically allowed in Py3.
+ if "..." not in kernel_args and Ellipsis not in kernel_args:
return kernel_args
- kernel_args = kernel_args[:]
- kernel_args.remove("...")
+ kernel_args = [arg for arg in kernel_args
+ if arg is not Ellipsis and arg != "..."]
from loopy.symbolic import SubstitutionRuleExpander
submap = SubstitutionRuleExpander(subst_rules)
--
GitLab