Index

_

__call__() (arraycontext.impl.pytato.compile.BaseLazilyCompilingFunctionCaller method) (arraycontext.impl.pytato.compile.CompiledFunction method) (arraycontext.PytestPyOpenCLArrayContextFactory method)

__init__() (arraycontext.PytatoPyOpenCLArrayContext method) (arraycontext.PytestPyOpenCLArrayContextFactory method)

A

allocator (arraycontext.PyOpenCLArrayContext attribute) (arraycontext.PytatoPyOpenCLArrayContext attribute) Array (class in arraycontext) array_types (arraycontext.ArrayContext attribute) ArrayContainer (class in arraycontext) ArrayContainerT (class in arraycontext) (class in arraycontext.container) arraycontext module ArrayContext (class in arraycontext) arraycontext.container module arraycontext.container.arithmetic module arraycontext.container.dataclass module arraycontext.container.traversal module arraycontext.context module arraycontext.impl.jax module arraycontext.impl.pyopencl module

arraycontext.impl.pyopencl.taggable_cl_array module arraycontext.impl.pytato module arraycontext.impl.pytato.compile module arraycontext.loopy module arraycontext.metadata module arraycontext.pytest module ArrayOrContainer (class in arraycontext) ArrayOrContainerOrScalar (class in arraycontext) ArrayOrContainerOrScalarT (class in arraycontext) (class in arraycontext.context) ArrayOrContainerT (class in arraycontext) (class in arraycontext.container) (class in arraycontext.context) ArrayT (class in arraycontext) (class in arraycontext.context) axes (arraycontext.impl.pyopencl.taggable_cl_array.TaggableCLArray attribute) Axis (class in arraycontext.impl.pyopencl.taggable_cl_array)

B

BaseLazilyCompilingFunctionCaller (class in arraycontext.impl.pytato.compile)

C

call_loopy() (arraycontext.ArrayContext method) compile() (arraycontext.ArrayContext method) (arraycontext.PytatoPyOpenCLArrayContext method)

CompiledFunction (class in arraycontext.impl.pytato.compile) context (arraycontext.PyOpenCLArrayContext attribute)

D

dataclass_array_container() (in module arraycontext)

deserialize_container() (in module arraycontext) dtype (arraycontext.Array attribute)

E

EagerJAXArrayContext (class in arraycontext) einsum() (arraycontext.ArrayContext method)

empty() (arraycontext.ArrayContext method) empty_like() (arraycontext.ArrayContext method)

F

f (arraycontext.impl.pytato.compile.BaseLazilyCompilingFunctionCaller attribute) flat_size_and_dtype() (in module arraycontext) flatten() (in module arraycontext) freeze() (arraycontext.ArrayContext method) (in module arraycontext)

freeze_thaw() (arraycontext.ArrayContext method) from_numpy() (arraycontext.ArrayContext method) (in module arraycontext) FromArrayContextCompile (class in arraycontext.impl.pytato.compile)

G

get_container_context_opt() (in module arraycontext)

get_container_context_recursively() (in module arraycontext) get_container_context_recursively_opt() (in module arraycontext)

I

input_id_to_name_in_program (arraycontext.impl.pytato.compile.CompiledFunction attribute)

is_array_container_type() (in module arraycontext)

L

LazilyJAXCompilingFunctionCaller (class in arraycontext.impl.pytato.compile)

LazilyPyOpenCLCompilingFunctionCaller (class in arraycontext.impl.pytato.compile)

M

make_loopy_program() (in module arraycontext) map_array_container() (in module arraycontext) map_reduce_array_container() (in module arraycontext) mapped_over_array_containers() (in module arraycontext) module arraycontext arraycontext.container arraycontext.container.arithmetic arraycontext.container.dataclass arraycontext.container.traversal arraycontext.context arraycontext.impl.jax arraycontext.impl.pyopencl arraycontext.impl.pyopencl.taggable_cl_array arraycontext.impl.pytato arraycontext.impl.pytato.compile arraycontext.loopy arraycontext.metadata arraycontext.pytest

multimap_array_container() (in module arraycontext) multimap_reduce_array_container() (in module arraycontext) multimapped_over_array_containers() (in module arraycontext)

N

name (arraycontext.metadata.NameHint attribute) NameHint (class in arraycontext.metadata)

NotAnArrayContainerError np (arraycontext.ArrayContext attribute)

O

outer() (in module arraycontext)

P

PyOpenCLArrayContext (class in arraycontext) pytato_program (arraycontext.impl.pytato.compile.CompiledFunction attribute) PytatoJAXArrayContext (class in arraycontext) PytatoPyOpenCLArrayContext (class in arraycontext)

pytest_generate_tests_for_array_contexts() (in module arraycontext) pytest_generate_tests_for_pyopencl_array_context() (in module arraycontext) PytestArrayContextFactory (class in arraycontext) PytestPyOpenCLArrayContextFactory (class in arraycontext)

Q

queue (arraycontext.PyOpenCLArrayContext attribute) (arraycontext.PytatoPyOpenCLArrayContext attribute)

R

rec_map_array_container() (in module arraycontext) rec_map_reduce_array_container() (in module arraycontext)

rec_multimap_array_container() (in module arraycontext) rec_multimap_reduce_array_container() (in module arraycontext) register_multivector_as_array_container() (in module arraycontext)

S

ScalarLike (class in arraycontext) SelfType (class in arraycontext.context)

serialize_container() (in module arraycontext) shape (arraycontext.Array attribute) size (arraycontext.Array attribute)

T

tag() (arraycontext.ArrayContext method) tag_axes() (in module arraycontext) tag_axis() (arraycontext.ArrayContext method) TaggableCLArray (class in arraycontext.impl.pyopencl.taggable_cl_array) tags (arraycontext.impl.pyopencl.taggable_cl_array.TaggableCLArray attribute) thaw() (arraycontext.ArrayContext method) (in module arraycontext)

to_numpy() (arraycontext.ArrayContext method) (in module arraycontext) to_tagged_cl_array() (in module arraycontext.impl.pyopencl.taggable_cl_array) transform_dag() (arraycontext.PytatoPyOpenCLArrayContext method) transform_loopy_program() (arraycontext.PyOpenCLArrayContext method)

U

unflatten() (in module arraycontext)

W

with_container_arithmetic() (in module arraycontext)

Z

zeros() (arraycontext.ArrayContext method)

zeros_like() (arraycontext.ArrayContext method)