From deb2a81323af055ca677b6f33dd6a85357d2acd0 Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Wed, 30 Jun 2021 12:37:12 -0500
Subject: [PATCH] fake_numpy: use loopy to allocate elementwise function
outputs
---
arraycontext/fake_numpy.py | 6 ++----
1 file changed, 2 insertions(+), 4 deletions(-)
diff --git a/arraycontext/fake_numpy.py b/arraycontext/fake_numpy.py
index 0c1309c..cdb9534 100644
--- a/arraycontext/fake_numpy.py
+++ b/arraycontext/fake_numpy.py
@@ -148,13 +148,11 @@ class BaseFakeNumpyNamespace:
def __getattr__(self, name):
def loopy_implemented_elwise_func(*args):
actx = self._array_context
- # FIXME: Maybe involve loopy type inference?
- result = actx.empty(args[0].shape, args[0].dtype)
prg = _get_scalar_func_loopy_program(actx,
c_name, nargs=len(args), naxes=len(args[0].shape))
- actx.call_loopy(prg, out=result,
+ outputs = actx.call_loopy(prg,
**{"inp%d" % i: arg for i, arg in enumerate(args)})
- return result
+ return outputs["out"]
if name in self._c_to_numpy_arc_functions:
from warnings import warn
--
GitLab