From 78dc92213c02daf01e515fb0702c0886fdcf5440 Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Wed, 11 Aug 2021 23:03:21 -0500
Subject: [PATCH] Pass no_numpy option for pytato loopy generation
---
arraycontext/impl/pytato/compile.py | 5 ++++-
1 file changed, 4 insertions(+), 1 deletion(-)
diff --git a/arraycontext/impl/pytato/compile.py b/arraycontext/impl/pytato/compile.py
index 2c1026c..5fad96c 100644
--- a/arraycontext/impl/pytato/compile.py
+++ b/arraycontext/impl/pytato/compile.py
@@ -225,8 +225,11 @@ class LazilyCompilingFunctionCaller:
rec_keyed_map_array_container(_as_dict_of_named_arrays,
outputs)
+ import loopy as lp
pytato_program = pt.generate_loopy(dict_of_named_arrays,
- options={"return_dict": True},
+ options=lp.Options(
+ return_dict=True,
+ no_numpy=True),
cl_device=self.actx.queue.device)
assert isinstance(pytato_program, BoundPyOpenCLProgram)
--
GitLab