Index

_

__abs__() (pytato.Array method) __add__() (pytato.Array method) __and__() (pytato.Array method) __call__() (pytato.target.BoundProgram method) (pytato.target.loopy.BoundPyOpenCLExecutable method) (pytato.target.loopy.BoundPyOpenCLProgram method) (pytato.target.python.BoundPythonProgram method) (pytato.transform.Mapper method) __contains__() (pytato.AbstractResultWithNamedArrays method) __eq__() (pytato.reductions.ReductionOperation method) __getitem__() (pytato.AbstractResultWithNamedArrays method) (pytato.Array method) __hash__() (pytato.reductions.ReductionOperation method) __init__() (pytato.AbstractResultWithNamedArrays method) (pytato.array.Placeholder method) (pytato.DictOfNamedArrays method) (pytato.NamedArray method) (pytato.transform.UsersCollector method)

__len__() (pytato.AbstractResultWithNamedArrays method) __mul__() (pytato.Array method) __neg__() (pytato.Array method) __or__() (pytato.Array method) __pos__() (pytato.Array method) __radd__() (pytato.Array method) __rand__() (pytato.Array method) __rmul__() (pytato.Array method) __ror__() (pytato.Array method) __rsub__() (pytato.Array method) __rtruediv__() (pytato.Array method) __rxor__() (pytato.Array method) __sub__() (pytato.Array method) __truediv__() (pytato.Array method) __xor__() (pytato.Array method) _SelfMapper (class in pytato.transform) _t_unit (pytato.target.loopy.codegen.CodeGenState attribute)

A

abs() (in module pytato) AbstractResultWithNamedArrays (class in pytato) access_descr_to_index (pytato.array.Einsum attribute) access_descriptors (pytato.array.Einsum attribute) add_store() (in module pytato.target.loopy.codegen) AdvancedIndexInContiguousAxes (class in pytato.array) AdvancedIndexInNoncontiguousAxes (class in pytato.array) all() (in module pytato) (pytato.Array method) all_input_names() (pytato.DistributedGraphPart method) AllReductionOperation (class in pytato.reductions) amax() (in module pytato) amin() (in module pytato) any() (in module pytato) (pytato.Array method) AnyReductionOperation (class in pytato.reductions) apply_distributive_property_to_einsums() (in module pytato.transform.einsum_distributive_law) arange() (in module pytato) arccos() (in module pytato) arcsin() (in module pytato) arctan() (in module pytato) arctan2() (in module pytato)

are_shape_components_equal() (in module pytato.utils) are_shapes_equal() (in module pytato.utils) args (pytato.array.Einsum attribute) Array (class in pytato) array (pytato.array.AxisPermutation attribute) (pytato.array.IndexRemappingBase attribute) (pytato.array.Reshape attribute) array expression array result ArrayOrScalar (class in pytato.loopy) arrays (pytato.array.Concatenate attribute) (pytato.array.Stack attribute) AssumeNonNegative (class in pytato.tags) axes (pytato.Array attribute) AxesT (class in pytato) (class in pytato.array) (class in pytato.loopy) Axis (class in pytato) axis (pytato.array.Concatenate attribute) (pytato.array.Roll attribute) (pytato.array.Stack attribute) axis_permutation (pytato.array.AxisPermutation attribute) AxisPermutation (class in pytato.array)

B

BasicIndex (class in pytato.array) BasicSet (class in islpy._isl) bind_program() (pytato.target.loopy.LoopyTarget method) bind_to_context() (pytato.target.loopy.BoundPyOpenCLProgram method) bindings (pytato.array.IndexLambda attribute) (pytato.function.Call attribute)

bound_arguments (pytato.target.BoundProgram attribute) BoundJAXPythonProgram (class in pytato.target.python) BoundProgram (class in pytato.target) BoundPyOpenCLExecutable (class in pytato.target.loopy) BoundPyOpenCLProgram (class in pytato.target.loopy) BoundPythonProgram (class in pytato.target.python) broadcast_to() (in module pytato)

C

CachedMapAndCopyMapper (class in pytato.transform) CachedMapper (class in pytato.transform) CachedWalkMapper (class in pytato.transform) Call (class in pytato.function) call (pytato.function.NamedCallResult attribute) call_loopy() (in module pytato.loopy) CallImplementationTag (class in pytato.tags) CannotBeLoweredToIndexLambda (class in pytato.diagnostic) CannotBroadcastError (class in pytato.diagnostic) clone_for_callee() (pytato.transform.CopyMapper method) CodeGenMapper (class in pytato.target.loopy.codegen) CodeGenPreprocessor (class in pytato.codegen) CodeGenState (class in pytato.target.loopy.codegen) combine() (pytato.transform.CombineMapper method) CombineMapper (class in pytato.transform) CombineT (class in pytato.transform) comm_tag (pytato.distributed.partition.CommunicationOpIdentifier attribute) (pytato.DistributedRecv attribute) (pytato.DistributedSend attribute)

CommTagType (class in pytato) CommunicationDepGraph (class in pytato.distributed.partition) CommunicationOpIdentifier (class in pytato.distributed.partition) Concatenate (class in pytato.array) concatenate() (in module pytato) conj() (in module pytato) (pytato.Array method) ConvertibleToShape (class in pytato.array) copy() (pytato.target.loopy.BoundPyOpenCLProgram method) (pytato.target.python.BoundPythonProgram method) copy_dict_of_named_arrays() (in module pytato.transform) CopyMapper (class in pytato.transform) CopyMapperWithExtraArgs (class in pytato.transform) cos() (in module pytato) cosh() (in module pytato) CountNamed (class in pytato.tags)

D

data (pytato.array.DataWrapper attribute) (pytato.DistributedSend attribute) DataInterface (class in pytato.array) DataWrapper (class in pytato.array) deduplicate_data_wrappers() (in module pytato.transform) DependencyMapper (class in pytato.transform) depends_on (pytato.target.loopy.codegen.PersistentExpressionContext attribute) dest_rank (pytato.distributed.partition.CommunicationOpIdentifier attribute) (pytato.DistributedSend attribute) device (pytato.target.loopy.LoopyPyOpenCLTarget attribute) DictOfNamedArrays (class in pytato) dim_to_index_lambda_components() (in module pytato.utils) DirectPredecessorsGetter (class in pytato.analysis)

DistributedGraphPart (class in pytato) DistributedGraphPartition (class in pytato) DistributedRecv (class in pytato) DistributedSend (class in pytato) DistributedSendRefHolder (class in pytato) DoDistribute (class in pytato.transform.einsum_distributive_law) domain_for_shape() (in module pytato.target.loopy.codegen) DoNotDistribute (class in pytato.transform.einsum_distributive_law) dot() (in module pytato) dtype (class in np) (pytato.Array attribute) (pytato.array.DataInterface attribute) (pytato.DistributedRecv attribute)

E

Einsum (class in pytato.array) einsum() (in module pytato) EinsumAxisDescriptor (class in pytato.array) EinsumDistributiveLawDescriptor (class in pytato.transform.einsum_distributive_law) EinsumElementwiseAxis (class in pytato.array) EinsumReductionAxis (class in pytato.array) ellipsis (built-in class)

equal() (in module pytato) EqualityComparer (class in pytato.equality) execute_distributed_partition() (in module pytato) exp() (in module pytato) expand_dims() (in module pytato) ExpandedDimsReshape (class in pytato.tags) expr (pytato.array.IndexLambda attribute) eye() (in module pytato)

F

find_distributed_partition() (in module pytato) full() (in module pytato)

function (pytato.function.Call attribute) FunctionDefinition (class in pytato.function) FunctionIdentifier (class in pytato.tags)

G

generate_jax() (in module pytato) generate_loopy() (in module pytato) get_common_dtype_of_ary_or_scalars() (in module pytato.utils) get_dependencies() (in module pytato.scalar_expr) (in module pytato.transform) get_dot_graph() (in module pytato) get_dot_graph_from_partition() (in module pytato) get_einsum_subscript_str() (in module pytato.utils) get_initial_codegen_state() (in module pytato.target.loopy.codegen)

get_loopy_target() (pytato.target.loopy.LoopyTarget method) get_loopy_temporary() (in module pytato.target.loopy.codegen) get_num_call_sites() (in module pytato.analysis) get_num_nodes() (in module pytato.analysis) get_nusers() (in module pytato.analysis) get_placeholder() (pytato.function.FunctionDefinition method) get_shape_after_broadcasting() (in module pytato.utils) greater() (in module pytato) greater_equal() (in module pytato)

H

handle_unsupported_array() (pytato.transform.Mapper method)

HighLevelOp (class in pytato.raising)

I

identifier (pytato.tags.FunctionIdentifier attribute) imag (pytato.Array attribute) imag() (in module pytato) ImplementationStrategy (class in pytato.tags) ImplementedResult (class in pytato.target.loopy.codegen) ImplInlined (class in pytato.tags) ImplStored (class in pytato.tags) ImplSubstitution (class in pytato.target.loopy) index_lambda_to_high_level_op() (in module pytato.raising) IndexBase (class in pytato.array) IndexLambda (class in pytato.array)

IndexRemappingBase (class in pytato.array) indices (pytato.array.IndexBase attribute) inline_calls() (in module pytato.transform.calls) InlineCallTag (class in pytato.tags) InlinedExpressionGenMapper (class in pytato.target.loopy.codegen) InlinedResult (class in pytato.target.loopy.codegen) InputArgumentBase (class in pytato.array) InputGatherer (class in pytato.transform) insn_id_gen (pytato.target.loopy.codegen.CodeGenState attribute) IntegralT (class in pytato.array) is_einsum_similar_to_subscript() (in module pytato.analysis) isnan() (in module pytato)

J

JAXPythonTarget (class in pytato.target.python)

L

less() (in module pytato) less_equal() (in module pytato) local_namespace (pytato.target.loopy.codegen.LocalExpressionContext attribute) LocalExpressionContext (class in pytato.target.loopy.codegen) log() (in module pytato) log10() (in module pytato) logical_and() (in module pytato)

logical_not() (in module pytato) logical_or() (in module pytato) lookup() (pytato.target.loopy.codegen.LocalExpressionContext method) LoopKernel (class in loopy.kernel) LoopyCall (class in pytato.loopy) LoopyCallResult (class in pytato.loopy) LoopyPyOpenCLTarget (class in pytato.target.loopy) LoopyTarget (class in pytato.target.loopy)

M

make_data_wrapper() (in module pytato.array) make_dict_of_named_arrays() (in module pytato.array) make_distributed_recv() (in module pytato) make_distributed_send() (in module pytato) make_placeholder() (in module pytato.array) make_size_param() (in module pytato.array) map_and_copy() (in module pytato.transform) map_foreign() (pytato.transform.Mapper method) MappedT (class in pytato.transform) Mapper (class in pytato.transform) materialize_with_mpms() (in module pytato.transform) matmul() (in module pytato) maximum() (in module pytato) MaxReductionOperation (class in pytato.reductions) MemoryObject (class in pyopencl._cl) minimum() (in module pytato) MinReductionOperation (class in pytato.reductions) module pytato pytato.analysis pytato.array pytato.cmath pytato.codegen pytato.diagnostic pytato.distributed pytato.distributed.execute pytato.distributed.nodes pytato.distributed.partition pytato.distributed.verify pytato.equality pytato.function pytato.loopy pytato.pad pytato.raising pytato.reductions pytato.scalar_expr pytato.stringifier pytato.tags pytato.target pytato.target.loopy pytato.target.loopy.codegen pytato.target.python pytato.transform pytato.transform.calls pytato.transform.einsum_distributive_law pytato.transform.lower_to_index_lambda pytato.transform.remove_broadcasts_einsum pytato.utils pytato.visualization pytato.visualization.dot pytato.visualization.fancy_placeholder_data_flow

N

name (pytato.array.DataWrapper attribute) (pytato.array.Placeholder attribute) (pytato.array.SizeParam attribute) (pytato.function.NamedCallResult attribute) (pytato.tags.CountNamed attribute) (pytato.tags.Named attribute) name_to_output (pytato.DistributedGraphPartition attribute) name_to_recv_node (pytato.DistributedGraphPart attribute) name_to_send_nodes (pytato.DistributedGraphPart attribute) NameClashError (class in pytato.diagnostic) Named (class in pytato.tags)

NamedArray (class in pytato) NamedCallResult (class in pytato.function) ndim (pytato.Array attribute) needed_pids (pytato.DistributedGraphPart attribute) neutral_element() (pytato.reductions.ReductionOperation method) newshape (pytato.array.Reshape attribute) node_to_users (pytato.transform.UsersCollector attribute) normalize_outputs() (in module pytato.codegen) (in module pytato.target.loopy.codegen) NormalizedSlice (class in pytato.array) not_equal() (in module pytato) num_indices (pytato.target.loopy.codegen.LocalExpressionContext attribute)

O

ones() (in module pytato) Options (class in loopy.options)

order (pytato.array.Reshape attribute) output_names (pytato.DistributedGraphPart attribute) overall_output_names (pytato.DistributedGraphPartition attribute)

P

pad() (in module pytato.pad) parameters (pytato.function.FunctionDefinition attribute) parse() (in module pytato.scalar_expr) partition_input_names (pytato.DistributedGraphPart attribute) PartitionInducedCycleError parts (pytato.DistributedGraphPartition attribute) passthrough_data (pytato.DistributedSendRefHolder attribute) PersistentExpressionContext (class in pytato.target.loopy.codegen) pid (pytato.DistributedGraphPart attribute) Placeholder (class in pytato.array) placeholder name post_visit() (pytato.transform.WalkMapper method) prefix (pytato.tags.PrefixNamed attribute) PrefixNamed (class in pytato.tags) preprocess() (in module pytato.codegen) PreprocessResult (class in pytato.codegen) prod() (in module pytato) ProductReductionOperation (class in pytato.reductions) program (pytato.target.BoundProgram attribute) pytato module pytato.analysis module pytato.array module pytato.cmath module pytato.codegen module pytato.diagnostic module pytato.distributed module pytato.distributed.execute module pytato.distributed.nodes module pytato.distributed.partition module pytato.distributed.verify module pytato.equality module

pytato.function module pytato.loopy module pytato.pad module pytato.raising module pytato.reductions module pytato.scalar_expr module pytato.stringifier module pytato.tags module pytato.target module pytato.target.loopy module pytato.target.loopy.codegen module pytato.target.python module pytato.transform module pytato.transform.calls module pytato.transform.einsum_distributive_law module pytato.transform.lower_to_index_lambda module pytato.transform.remove_broadcasts_einsum module pytato.utils module pytato.visualization module pytato.visualization.dot module pytato.visualization.fancy_placeholder_data_flow module PythonTarget (class in pytato.target.python)

R

real (pytato.Array attribute) real() (in module pytato) rec() (pytato.transform.Mapper method) rec_get_user_nodes() (in module pytato.transform) ReductionBounds (class in pytato.target.loopy.codegen) ReductionDescriptor (class in pytato) ReductionOperation (class in pytato.reductions) Reprifier (class in pytato.stringifier)

Reshape (class in pytato.array) results (pytato.target.loopy.codegen.CodeGenState attribute) return_type (pytato.function.FunctionDefinition attribute) returns (pytato.function.FunctionDefinition attribute) ReturnT (class in pytato.function) ReturnType (class in pytato.function) rewrite_einsums_with_no_broadcasts() (in module pytato.transform.remove_broadcasts_einsum) Roll (class in pytato.array) roll() (in module pytato)

S

ScalarExpression (in module pytato.scalar_expr) send (pytato.DistributedSendRefHolder attribute) shape (pytato.Array attribute) (pytato.array.DataInterface attribute) (pytato.array.DataWrapper attribute) (pytato.DistributedRecv attribute) ShapeComponent (class in pytato.array) ShapeType (class in pytato) shift (pytato.array.Roll attribute) show_dot_graph() (in module pytato) show_fancy_placeholder_data_flow() (in module pytato) sin() (in module pytato) sinh() (in module pytato) SizeParam (class in pytato.array) SizeParamGatherer (class in pytato.transform) sqrt() (in module pytato)

squeeze() (in module pytato) src_rank (pytato.distributed.partition.CommunicationOpIdentifier attribute) (pytato.DistributedRecv attribute) Stack (class in pytato.array) stack() (in module pytato) staple_distributed_send() (in module pytato) start (pytato.array.NormalizedSlice attribute) state (pytato.target.loopy.codegen.PersistentExpressionContext attribute) step (pytato.array.NormalizedSlice attribute) stop (pytato.array.NormalizedSlice attribute) StoredResult (class in pytato.target.loopy.codegen) SubsetDependencyMapper (class in pytato.transform) substitute() (in module pytato.scalar_expr) SubstitutionRuleResult (class in pytato.target.loopy.codegen) sum() (in module pytato) SumReductionOperation (class in pytato.reductions)

T

T (class in pytato.distributed.partition) (pytato.Array attribute) Tag (class in pytato) (class in pytato.loopy) tag_all_calls_to_be_inlined() (in module pytato.transform.calls) tag_user_nodes() (in module pytato.transform) tagged() (pytato.Array method) tags (pytato.Array attribute) tan() (in module pytato)

tanh() (in module pytato) Target (class in pytato.target) target (pytato.target.BoundProgram attribute) TemporaryVariable (class in loopy.kernel.data) to_index_lambda() (in module pytato.transform.lower_to_index_lambda) to_loopy_expression() (pytato.target.loopy.codegen.ImplementedResult method) TopoSortMapper (class in pytato.transform) trace_call() (in module pytato) TranslationUnit (class in lp) transpose() (in module pytato)

U

UnknownIndexLambdaExpr (class in pytato.diagnostic) update_depends_on() (pytato.target.loopy.codegen.PersistentExpressionContext method)

update_kernel() (pytato.target.loopy.codegen.CodeGenState method) user_input_names (pytato.DistributedGraphPart attribute) UsersCollector (class in pytato.transform)

V

var_name_gen (pytato.target.loopy.codegen.CodeGenState attribute) var_to_reduction_descr (pytato.array.IndexLambda attribute)

vdot() (in module pytato) verify_distributed_partition() (in module pytato) visit() (pytato.transform.WalkMapper method)

W

WalkMapper (class in pytato.transform) where() (in module pytato) with_tagged_axis() (pytato.Array method) with_tagged_reduction() (pytato.array.Einsum method) (pytato.array.IndexLambda method)

with_transformed_program() (pytato.target.loopy.BoundPyOpenCLProgram method) (pytato.target.python.BoundPythonProgram method) with_transformed_translation_unit() (pytato.target.loopy.BoundPyOpenCLExecutable method) without_tags() (pytato.Array method)

Z

zeros() (in module pytato)