Index

_

__abs__() (pytato.Array method) __add__() (pytato.Array method) __and__() (pytato.Array method) __call__() (pytato.partition.GraphPartitioner method) (pytato.target.BoundProgram 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.partition.GraphPartitioner 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) _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.partition.GraphPart 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) 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 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) bindings (pytato.array.IndexLambda attribute) bound_arguments (pytato.target.BoundProgram attribute)

BoundJAXPythonProgram (class in pytato.target.python) BoundProgram (class in pytato.target) 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_loopy() (in module pytato.loopy) CannotBeLoweredToIndexLambda (class in pytato.diagnostic) CannotBroadcastError (class in pytato.diagnostic) 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.DistributedRecv attribute) (pytato.DistributedSend attribute)

CommTagType (class in pytato) 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.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) distributed_sends (pytato.DistributedGraphPart attribute) DistributedGraphPart (class in pytato) DistributedGraphPartition (class in pytato) DistributedRecv (class in pytato) DistributedSend (class in pytato) DistributedSendRefHolder (class in pytato) domain_for_shape() (in module pytato.target.loopy.codegen) dot() (in module pytato) dtype (pytato.Array attribute) (pytato.array.DataInterface attribute) (pytato.DistributedRecv attribute)

E

EdgeCachedMapper (class in pytato.transform) Einsum (class in pytato.array) einsum() (in module pytato) EinsumAxisDescriptor (class in pytato.array) 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) execute_partition() (in module pytato.partition) 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)

find_partition() (in module pytato.partition) full() (in module pytato)

G

generate_jax() (in module pytato) generate_loopy() (in module pytato) get_ascii_graph() (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_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_nodes() (in module pytato.analysis) get_nusers() (in module pytato.analysis) get_shape_after_broadcasting() (in module pytato.utils) GraphPart (class in pytato.partition) GraphPartition (class in pytato.partition) GraphPartitioner (class in pytato.partition) greater() (in module pytato) greater_equal() (in module pytato)

H

handle_edge() (pytato.transform.EdgeCachedMapper method)

handle_unsupported_array() (pytato.transform.Mapper method) HighLevelOp (class in pytato.raising)

I

identifier 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) InlinedExpressionGenMapper (class in pytato.target.loopy.codegen) InlinedResult (class in pytato.target.loopy.codegen) input_name_to_recv_node (pytato.DistributedGraphPart attribute) 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_partition() (pytato.partition.GraphPartitioner method) 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) 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.loopy pytato.partition 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.lower_to_index_lambda pytato.transform.remove_broadcasts_einsum pytato.utils pytato.visualization

N

name (pytato.array.DataWrapper attribute) (pytato.array.Placeholder attribute) (pytato.array.SizeParam attribute) (pytato.tags.CountNamed attribute) (pytato.tags.Named attribute) NameClashError (class in pytato.diagnostic) Named (class in pytato.tags) NamedArray (class in pytato) ndim (pytato.Array attribute)

needed_pids (pytato.partition.GraphPart 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_name_to_send_node (pytato.DistributedGraphPart attribute) output_names (pytato.partition.GraphPart attribute)

P

parse() (in module pytato.scalar_expr) partition_input_names (pytato.partition.GraphPart attribute) PartitionInducedCycleError parts (pytato.partition.GraphPartition attribute) passthrough_data (pytato.DistributedSendRefHolder attribute) PersistentExpressionContext (class in pytato.target.loopy.codegen) pid (pytato.partition.GraphPart 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.loopy module pytato.partition 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.lower_to_index_lambda module pytato.transform.remove_broadcasts_einsum module pytato.utils module pytato.visualization 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) 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) 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_ascii_graph() (in module pytato) show_dot_graph() (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.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.partition) (pytato.Array attribute) Tag (class in pytato) (class in pytato.loopy) 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) toposorted_part_ids (pytato.partition.GraphPartition attribute) TopoSortMapper (class in pytato.transform) 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.partition.GraphPart attribute) UsersCollector (class in pytato.transform)

V

var_name_gen (pytato.target.loopy.codegen.CodeGenState attribute) var_name_to_result (pytato.partition.GraphPartition 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) without_tags() (pytato.Array method)

Z

zeros() (in module pytato)