List of all items
Structs
- BranchBuilder
- GradMap
- Graph
- GraphId
- Node
- NodeHandle
- NodeId
- NodeRemap
- Run
- ScanPredicate
- grad::AddRule
- grad::MulRule
- grad::NoGradientBinaryRule
- grad::ReluRule
- grad::WhereRule
- jit::OpAttrs
- opt::CastFusionRule
- opt::FusionRule
- opt::LoweringRule
- opt::OrderingEdges
- opt::RuleRegistry
- registry::FusedOpEntry
- registry::FusedOpId
- registry::FusedOpParamsKey
- registry::FusedOpRegistry
- registry::FusedOps
- registry::PatternHash
- registry::PatternMatch
- registry::PatternTree
- run::Run
- runtime_fused::RuntimeFusedOpEntry
- scan::ScanLayout
- scan::ScanStep
Enums
- Op
- PadMode
- QuantType
- ScanEmit
- ScanRole
- StorageClass
- jit::OpTag
- jit::PatternNode
- opt::RuleFamily
- registry::BackwardKind
- registry::FusedOpFamily
- registry::FusedOpParams
- registry::Reduction
- registry::SubgraphPattern
- runtime_fused::RelAttrError
- runtime_fused::RuntimeFusedError
Traits
Functions
- branch_density
- branch_density_multi
- branches_in_topo_order
- build_rope_tables
- build_rope_tables_with_inv_freq
- compact
- derive_view_output_layout
- device_alternating_order
- effective_roots
- extract_runs
- extract_runs_multi
- grad::dispatch_gradient
- infer_storage_class
- jit::match_region
- jit::match_region_extract
- jit::op_to_tag
- lower_picked_route
- lower_picked_route_streaming
- lower_run
- lower_runs_arm0
- opt::base_map_hash
- opt::derive_ordering
- opt::execution_plan
- opt::fuse_linear
- opt::insert_cast_fixups
- opt::insert_copies
- opt::insert_cross_device_copies
- opt::insert_evict_reload
- opt::insert_layout_fixups
- opt::insert_safety_copies
- opt::lower_const_placement
- opt::lower_to_base_map
- opt::optimize
- opt::promote_views_for_liveness
- passes_fewness_gate
- primitive_shape
- registry::awq_matmul::canonical_pattern
- registry::awq_matmul::decompose
- registry::awq_matmul::entry
- registry::causal_conv1d::canonical_pattern
- registry::causal_conv1d::decompose
- registry::causal_conv1d::entry
- registry::conv2d::canonical_pattern
- registry::conv2d::decompose
- registry::conv2d::entry
- registry::conv_transpose_2d::canonical_pattern
- registry::conv_transpose_2d::decompose
- registry::conv_transpose_2d::entry
- registry::decompose_via_recipe
- registry::default_registry
- registry::flash_attn::canonical_pattern
- registry::flash_attn::decompose
- registry::flash_attn::entry
- registry::flash_attn_backward::entry_k
- registry::flash_attn_backward::entry_q
- registry::flash_attn_backward::entry_v
- registry::fused_linear::canonical_pattern
- registry::fused_linear::decompose
- registry::fused_linear::entry
- registry::fused_softmax_cross_entropy::canonical_pattern
- registry::fused_softmax_cross_entropy::decompose
- registry::fused_softmax_cross_entropy::entry
- registry::inplace_affine::canonical_pattern
- registry::inplace_affine::decompose
- registry::inplace_affine::entry
- registry::layer_norm_last_dim::canonical_pattern
- registry::layer_norm_last_dim::decompose
- registry::layer_norm_last_dim::entry
- registry::layer_norm_last_dim_backward::canonical_pattern
- registry::layer_norm_last_dim_backward::decompose
- registry::layer_norm_last_dim_backward::entry
- registry::nf4_matmul::canonical_pattern
- registry::nf4_matmul::decompose
- registry::nf4_matmul::entry
- registry::paged_attn::canonical_pattern
- registry::paged_attn::decompose
- registry::paged_attn::entry
- registry::paged_attn::recipe_for
- registry::powi_backward::canonical_pattern
- registry::powi_backward::decompose
- registry::powi_backward::entry
- registry::qmatmul::canonical_pattern
- registry::qmatmul::decompose
- registry::qmatmul::entry
- registry::recipe_for
- registry::reduce_max_to_backward::canonical_pattern
- registry::reduce_max_to_backward::decompose
- registry::reduce_max_to_backward::entry
- registry::rms_norm_last_dim::canonical_pattern
- registry::rms_norm_last_dim::decompose
- registry::rms_norm_last_dim::entry
- registry::rms_norm_last_dim_backward::canonical_pattern
- registry::rms_norm_last_dim_backward::decompose
- registry::rms_norm_last_dim_backward::entry
- registry::rope::canonical_pattern
- registry::rope::decompose
- registry::rope::entry
- registry::selective_scan::canonical_pattern
- registry::selective_scan::decompose
- registry::selective_scan::entry
- registry::softmax_last_dim::canonical_pattern
- registry::softmax_last_dim::decompose
- registry::softmax_last_dim::entry
- registry::softmax_last_dim_backward::canonical_pattern
- registry::softmax_last_dim_backward::decompose
- registry::softmax_last_dim_backward::entry
- registry::ssd_chunk_scan::canonical_pattern
- registry::ssd_chunk_scan::decompose
- registry::ssd_chunk_scan::entry
- run::branch_density
- run::branch_density_multi
- run::branches_in_topo_order
- run::device_alternating_order
- run::effective_roots
- run::extract_runs
- run::extract_runs_multi
- run::lower_picked_route
- run::lower_picked_route_streaming
- run::lower_run
- run::lower_runs_arm0
- run::passes_fewness_gate
- run::walk_picked_route_streaming
- runtime_fused::count_scalar_slots
- runtime_fused::decompose_region
- runtime_fused::emit_region
- runtime_fused::register_runtime_fused
- runtime_fused::resolve_rel_attrs
- runtime_fused::runtime_entries
- runtime_fused::runtime_lowering_decompose
- runtime_fused::runtime_name
- runtime_fused::runtime_region
- runtime_fused::try_emit_region
- scan::build_scan_step
- scan::parse_scan_layout
- scan::unroll_scan
- topo_order
- topo_order_multi
- walk_picked_route_streaming