Skip to content
Snippets Groups Projects
Commit 474f93a6 authored by Alexandru Fikl's avatar Alexandru Fikl Committed by Andreas Klöckner
Browse files

remove stray _get_scalar_func_loopy_program

parent 3b02d065
No related branches found
No related tags found
No related merge requests found
Pipeline #254383 passed
......@@ -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
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment