From 94467bbe095f72426e9a587285b1e48fa99f28f2 Mon Sep 17 00:00:00 2001
From: Kaushik Kulkarni <kaushikcfd@gmail.com>
Date: Wed, 16 Jun 2021 14:15:25 -0500
Subject: [PATCH] fix scalar input to pytato program
---
arraycontext/impl/pytato.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/arraycontext/impl/pytato.py b/arraycontext/impl/pytato.py
index 7d4f09f..739a922 100644
--- a/arraycontext/impl/pytato.py
+++ b/arraycontext/impl/pytato.py
@@ -308,8 +308,9 @@ class PytatoExecutable:
for pos, arg in enumerate(args):
if isinstance(arg, np.number):
+
input_kwargs_to_loopy[self.input_id_to_name_in_program[(pos,)]] = (
- arg)
+ cla.to_device(self.actx.queue, np.array(arg)))
elif is_array_container(arg):
def _extract_lpy_kwargs(keys, ary):
if isinstance(ary, pt.array.DataWrapper):
--
GitLab