From 5b9b1aba33c987bd2e8de250b18c241b362f204f Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Mon, 15 Jul 2013 10:53:00 -0400
Subject: [PATCH] Handle the unlikely case of no output arguments in
loopy.compiled
---
loopy/compiled.py | 12 ++++++++----
1 file changed, 8 insertions(+), 4 deletions(-)
diff --git a/loopy/compiled.py b/loopy/compiled.py
index e73127828..2a1f5fb81 100644
--- a/loopy/compiled.py
+++ b/loopy/compiled.py
@@ -622,10 +622,14 @@ def generate_invoker(kernel, impl_arg_info, flags):
for arg in impl_arg_info
if arg.base_name in kernel.get_written_variables()))
else:
- gen("return _lpy_evt, (%s,)"
- % ", ".join(arg.name
- for arg in impl_arg_info
- if arg.base_name in kernel.get_written_variables()))
+ out_args = [arg
+ for arg in impl_arg_info
+ if arg.base_name in kernel.get_written_variables()]
+ if not out_args:
+ gen("return _lpy_evt, (%s)"
+ % ", ".join(arg.name for arg in out_args))
+ else:
+ gen("return _lpy_evt, ()")
# }}}
--
GitLab