From 474f93a6f2eddc768c21748f18231d38299237bb Mon Sep 17 00:00:00 2001
From: Alexandru Fikl <alexfikl@gmail.com>
Date: Sun, 16 Jan 2022 16:28:48 -0600
Subject: [PATCH] remove stray _get_scalar_func_loopy_program
---
arraycontext/fake_numpy.py | 39 +-------------------------------------
1 file changed, 1 insertion(+), 38 deletions(-)
diff --git a/arraycontext/fake_numpy.py b/arraycontext/fake_numpy.py
index 25343df..73c9e40 100644
--- a/arraycontext/fake_numpy.py
+++ b/arraycontext/fake_numpy.py
@@ -26,47 +26,10 @@ THE SOFTWARE.
import numpy as np
from arraycontext.container import NotAnArrayContainerError, serialize_container
from arraycontext.container.traversal import rec_map_array_container
-from pytools import memoize_in
-
-
-# {{{ _get_scalar_func_loopy_program
-
-def _get_scalar_func_loopy_program(actx, c_name, nargs, naxes):
- @memoize_in(actx, _get_scalar_func_loopy_program)
- def get(c_name, nargs, naxes):
- from pymbolic import var
-
- var_names = ["i%d" % i for i in range(naxes)]
- size_names = ["n%d" % i for i in range(naxes)]
- subscript = tuple(var(vname) for vname in var_names)
- from islpy import make_zero_and_vars
- v = make_zero_and_vars(var_names, params=size_names)
- domain = v[0].domain()
- for vname, sname in zip(var_names, size_names):
- domain = domain & v[0].le_set(v[vname]) & v[vname].lt_set(v[sname])
-
- domain_bset, = domain.get_basic_sets()
-
- import loopy as lp
- from .loopy import make_loopy_program
- from arraycontext.transform_metadata import ElementwiseMapKernelTag
- return make_loopy_program(
- [domain_bset],
- [
- lp.Assignment(
- var("out")[subscript],
- var(c_name)(*[
- var("inp%d" % i)[subscript] for i in range(nargs)]))
- ],
- name="actx_special_%s" % c_name,
- tags=(ElementwiseMapKernelTag(),))
-
- return get(c_name, nargs, naxes)
-
-# }}}
# {{{ BaseFakeNumpyNamespace
+
class BaseFakeNumpyNamespace:
def __init__(self, array_context):
self._array_context = array_context
--
GitLab