Index

_

__call__() (helion.Kernel method) __init__() (helion.autotuner.differential_evolution.DifferentialEvolutionSearch method) (helion.autotuner.finite_search.FiniteSearch method) (helion.autotuner.random_search.RandomSearch method) (helion.Config method), [1], [2] (helion.exc.RankMismatch method) (helion.Kernel method) (helion.language.Tile method) (helion.runtime.config.Config method) (helion.Settings method), [1], [2]

A

allow_warp_specialize (helion.Settings attribute), [1], [2] arange() (in module helion.language) args (helion.autotuner.random_search.RandomSearch attribute) AssignmentMultipleTargets (class in helion.exc) associative_scan() (in module helion.language)

atomic_add() (in module helion.language), [1] autotune() (helion.Kernel method) autotune_compile_timeout (helion.Settings attribute), [1], [2] autotune_log_level (helion.Settings attribute), [1], [2] autotune_precompile (helion.Settings attribute), [1], [2] AutotuningDisallowedInEnvironment (class in helion.exc)

B

Base (class in helion.exc) BaseError (class in helion.exc) BaseWarning (class in helion.exc) begin (helion.language.Tile property)

bind() (helion.Kernel method) block_size (helion.language.Tile property) block_sizes (helion.Config attribute) (helion.Config property), [1] (helion.runtime.config.Config property)

C

CannotModifyHostVariableOnDevice (class in helion.exc) CannotReadDeviceVariableOnHost (class in helion.exc) CantCombineTypesInControlFlow (class in helion.exc) CantReadOnDevice (class in helion.exc) check_autotuning_disabled() (helion.Settings method), [1] ClosureMutation (class in helion.exc) ClosuresNotSupported (class in helion.exc)

Config (class in helion), [1] (class in helion.runtime.config) config (helion.Config attribute), [1] (helion.runtime.config.Config attribute) ConfigSpecFragmentWithSymInt (class in helion.exc) constexpr (in module helion.language) count (helion.autotuner.random_search.RandomSearch attribute)

D

DecoratorAfterHelionKernelDecorator (class in helion.exc) default() (helion.Settings static method), [1] DeviceAPIOnHost (class in helion.exc)

DeviceLoopElseBlock (class in helion.exc) DeviceTensorSubscriptAssignmentNotAllowed (class in helion.exc) DifferentialEvolutionSearch (class in helion.autotuner.differential_evolution) dot_precision (helion.Settings attribute), [1], [2]

E

end (helion.language.Tile property)

ErrorCompilingKernel (class in helion.exc) evolve_population() (helion.autotuner.differential_evolution.DifferentialEvolutionSearch method)

F

FailedToUnpackTile (class in helion.exc) FailedToUnpackTupleAssign (class in helion.exc) FiniteSearch (class in helion.autotuner.finite_search) flatten_loops (helion.Config attribute) (helion.Config property), [1] (helion.runtime.config.Config property)

force_autotune (helion.Settings attribute), [1], [2] from_json() (helion.Config class method), [1] (helion.runtime.config.Config class method) full() (in module helion.language), [1]

G

get_num_sm() (in module helion.runtime)

GlobalMutation (class in helion.exc) grid() (in module helion.language), [1]

H

helion.autotuner module helion.autotuner.differential_evolution module

helion.autotuner.finite_search module helion.autotuner.random_search module HostTensorDirectUsage (class in helion.exc)

I

id (helion.language.Tile property) ignore_warnings (helion.Settings attribute), [1], [2] IncorrectTileUsage (class in helion.exc) index (helion.language.Tile property) index_dtype (helion.Settings attribute), [1], [2] indexing (helion.Config attribute) (helion.Config property), [1] (helion.runtime.config.Config property)

InductorLoweringError (class in helion.exc) initial_two_generations() (helion.autotuner.differential_evolution.DifferentialEvolutionSearch method) InternalError (class in helion.exc) InvalidAssignment (class in helion.exc) InvalidConfig (class in helion.exc) InvalidDeviceForLoop (class in helion.exc) InvalidIndexingType (class in helion.exc) iter_candidates() (helion.autotuner.differential_evolution.DifferentialEvolutionSearch method)

K

Kernel (class in helion)

kernel (helion.autotuner.random_search.RandomSearch attribute) kernel() (in module helion)

L

l2_groupings (helion.Config attribute) (helion.Config property), [1] (helion.runtime.config.Config property) load() (helion.Config class method), [1] (helion.runtime.config.Config class method) (in module helion.language), [1]

loop_orders (helion.Config attribute) (helion.Config property), [1] (helion.runtime.config.Config property) LoopDependencyError (class in helion.exc) LoopFunctionNotInFor (class in helion.exc)

M

message (helion.exc.AssignmentMultipleTargets attribute) (helion.exc.AutotuningDisallowedInEnvironment attribute) (helion.exc.BaseError attribute) (helion.exc.BaseWarning attribute) (helion.exc.CannotModifyHostVariableOnDevice attribute) (helion.exc.CannotReadDeviceVariableOnHost attribute) (helion.exc.CantCombineTypesInControlFlow attribute) (helion.exc.CantReadOnDevice attribute) (helion.exc.ClosureMutation attribute) (helion.exc.ClosuresNotSupported attribute) (helion.exc.ConfigSpecFragmentWithSymInt attribute) (helion.exc.DecoratorAfterHelionKernelDecorator attribute) (helion.exc.DeviceAPIOnHost attribute) (helion.exc.DeviceLoopElseBlock attribute) (helion.exc.DeviceTensorSubscriptAssignmentNotAllowed attribute) (helion.exc.ErrorCompilingKernel attribute) (helion.exc.FailedToUnpackTile attribute) (helion.exc.FailedToUnpackTupleAssign attribute) (helion.exc.GlobalMutation attribute) (helion.exc.HostTensorDirectUsage attribute) (helion.exc.IncorrectTileUsage attribute) (helion.exc.InductorLoweringError attribute) (helion.exc.InvalidAssignment attribute) (helion.exc.InvalidConfig attribute) (helion.exc.InvalidDeviceForLoop attribute) (helion.exc.InvalidIndexingType attribute) (helion.exc.LoopDependencyError attribute) (helion.exc.LoopFunctionNotInFor attribute) (helion.exc.NamingConflict attribute) (helion.exc.NestedDeviceLoopsConflict attribute) (helion.exc.NestedGridLoop attribute) (helion.exc.NonTensorSubscriptAssign attribute) (helion.exc.NotAllowedInHelperFunction attribute) (helion.exc.NotAllowedOnDevice attribute) (helion.exc.NotEnoughConfigs attribute) (helion.exc.NoTensorArgs attribute) (helion.exc.NotInsideKernel attribute) (helion.exc.OverpackedTile attribute) (helion.exc.RankMismatch attribute) (helion.exc.RegisterTunableArgTypes attribute) (helion.exc.RequiresTensorInAssignment attribute) (helion.exc.ShapeMismatch attribute) (helion.exc.ShapeSpecializingAllocation attribute) (helion.exc.ShapeSpecializingCall attribute) (helion.exc.SpecializeArgType attribute) (helion.exc.SpecializeOnDevice attribute) (helion.exc.StarredArgsNotSupportedOnDevice attribute) (helion.exc.StatementNotSupported attribute) (helion.exc.TensorOperationInWrapper attribute) (helion.exc.TensorOperationsInHostCall attribute) (helion.exc.TopLevelStatementBetweenLoops attribute) (helion.exc.TracedArgNotSupported attribute) (helion.exc.TritonError attribute) (helion.exc.TunableNameConflict attribute) (helion.exc.TunableTypeNotSupported attribute) (helion.exc.TypeInferenceError attribute) (helion.exc.UndefinedVariable attribute) (helion.exc.UnsupportedPythonType attribute) (helion.exc.WrongDevice attribute)

module helion.autotuner helion.autotuner.differential_evolution helion.autotuner.finite_search helion.autotuner.random_search mutate() (helion.autotuner.differential_evolution.DifferentialEvolutionSearch method)

N

NamingConflict (class in helion.exc) NestedDeviceLoopsConflict (class in helion.exc) NestedGridLoop (class in helion.exc) NonTensorSubscriptAssign (class in helion.exc) normalize_args() (helion.Kernel method) NotAllowedInHelperFunction (class in helion.exc) NotAllowedOnDevice (class in helion.exc) NotEnoughConfigs (class in helion.exc)

NoTensorArgs (class in helion.exc) NotInsideKernel (class in helion.exc) num_stages (helion.Config attribute) (helion.Config property), [1] (helion.runtime.config.Config property) num_warps (helion.Config attribute) (helion.Config property), [1] (helion.runtime.config.Config property)

O

OverpackedTile (class in helion.exc)

P

pid_type (helion.Config attribute) (helion.Config property), [1] (helion.runtime.config.Config property)

print_output_code (helion.Settings attribute), [1], [2]

R

RandomSearch (class in helion.autotuner.random_search) range_flattens (helion.Config attribute) (helion.Config property), [1] (helion.runtime.config.Config property) range_multi_buffers (helion.Config attribute) (helion.Config property), [1] (helion.runtime.config.Config property) range_num_stages (helion.Config attribute) (helion.Config property), [1] (helion.runtime.config.Config property) range_unroll_factors (helion.Config attribute) (helion.Config property), [1] (helion.runtime.config.Config property) range_warp_specializes (helion.Config attribute) (helion.Config property), [1] (helion.runtime.config.Config property)

RankMismatch (class in helion.exc) reduce() (in module helion.language) reduction_loops (helion.Config attribute) (helion.Config property), [1] (helion.runtime.config.Config property) register_block_size() (in module helion.language), [1] register_reduction_dim() (in module helion.language) register_tunable() (in module helion.language), [1] RegisterTunableArgTypes (class in helion.exc) report() (helion.exc.Base method) (helion.exc.BaseError method) (helion.exc.BaseWarning method) RequiresTensorInAssignment (class in helion.exc) reset() (helion.Kernel method)

S

save() (helion.Config method), [1] (helion.runtime.config.Config method) set_default_settings() (in module helion) set_triton_allocator() (in module helion.runtime) Settings (class in helion), [1] ShapeMismatch (class in helion.exc) ShapeSpecializingAllocation (class in helion.exc) ShapeSpecializingCall (class in helion.exc) specialization_key() (helion.Kernel method) specialize() (in module helion.language)

SpecializeArgType (class in helion.exc) SpecializeOnDevice (class in helion.exc) StarredArgsNotSupportedOnDevice (class in helion.exc) StatementNotSupported (class in helion.exc) static_ranges (helion.Config attribute) (helion.Config property), [1] (helion.runtime.config.Config property) static_shapes (helion.Settings attribute), [1], [2] store() (in module helion.language), [1] subscript() (in module helion.language)

T

TensorOperationInWrapper (class in helion.exc) TensorOperationsInHostCall (class in helion.exc) Tile (class in helion.language) tile() (in module helion.language), [1] tile_begin() (in module helion.language) tile_block_size() (in module helion.language) tile_end() (in module helion.language) tile_id() (in module helion.language) tile_index() (in module helion.language) to_dict() (helion.Settings method), [1]

to_json() (helion.Config method), [1] (helion.runtime.config.Config method) TopLevelStatementBetweenLoops (class in helion.exc) TorchOpTracingError (class in helion.exc) TracedArgNotSupported (class in helion.exc) triton_wait_signal() (in module helion.runtime) TritonError (class in helion.exc) TunableNameConflict (class in helion.exc) TunableTypeNotSupported (class in helion.exc) TypeInferenceError (class in helion.exc)

U

UndefinedVariable (class in helion.exc)

UnsupportedPythonType (class in helion.exc) use_default_config (helion.Settings attribute), [1], [2]

W

WrongDevice (class in helion.exc)

Z

zeros() (in module helion.language), [1]