Index A | B | C | D | E | F | G | H | I | J | K | L | M | N | O | P | R | S | T | U | V | W | Y A Abs (class in wave_lang.kernel.ops.wave_ops) abs() (in module wave_lang.kernel.ops.wave_ops) acc (wave_lang.kernel.ops.wave_ops.MMA attribute) (wave_lang.kernel.ops.wave_ops.ScaledMMA attribute) acc_index (wave_lang.kernel.ops.wave_ops.MMA property) (wave_lang.kernel.ops.wave_ops.ScaledMMA property) acc_type (wave_lang.kernel.ops.wave_ops.MMA property) (wave_lang.kernel.ops.wave_ops.ScaledMMA property) Add (class in wave_lang.kernel.ops.wave_ops) add_conditional_barriers_to_loop() built-in function add_to_graph() (wave_lang.kernel.ops.wave_ops.CustomOp method) (wave_lang.kernel.ops.wave_ops.Output method) (wave_lang.kernel.ops.wave_ops.Placeholder method) address_space (wave_lang.kernel.ops.wave_ops.Allocate attribute) align_index() (wave_lang.kernel.ops.wave_ops.ScaledMMA method) Allocate (class in wave_lang.kernel.ops.wave_ops) allocate() (in module wave_lang.kernel.ops.wave_ops) allocation_size (wave_lang.kernel.ops.wave_ops.Allocate property) And (class in wave_lang.kernel.ops.wave_ops) apply_expr() (in module wave_lang.kernel.ops.wave_ops) apply_scaling (wave_lang.kernel.ops.wave_ops.SoftsignOp attribute) ApplyExpr (class in wave_lang.kernel.ops.wave_ops) arg (wave_lang.kernel.ops.wave_ops.BitcastOp attribute) (wave_lang.kernel.ops.wave_ops.Broadcast attribute) (wave_lang.kernel.ops.wave_ops.CastOp attribute) (wave_lang.kernel.ops.wave_ops.Permute attribute) (wave_lang.kernel.ops.wave_ops.ReduceOp attribute) (wave_lang.kernel.ops.wave_ops.ScanOp attribute) (wave_lang.kernel.ops.wave_ops.ShuffleOp attribute) (wave_lang.kernel.ops.wave_ops.SoftsignOp attribute) (wave_lang.kernel.ops.wave_ops.TopkOp attribute) (wave_lang.kernel.ops.wave_ops.UnaryPyOp attribute) args (wave_lang.kernel.ops.wave_ops.Reshape attribute) (wave_lang.kernel.ops.wave_ops.Unknown attribute) asserts (wave_lang.support.debugging.DebugFlags attribute) Atan2 (class in wave_lang.kernel.ops.wave_ops) atan2() (in module wave_lang.kernel.ops.wave_ops) atomic_add() (in module wave_lang.kernel.ops.wave_ops) atomic_min() (in module wave_lang.kernel.ops.wave_ops) AtomicAddOp (class in wave_lang.kernel.ops.wave_ops) AtomicOp (class in wave_lang.kernel.ops.wave_ops) axis (wave_lang.kernel.ops.wave_ops.Iterate attribute) B barId (wave_lang.kernel.ops.wave_ops.SharedMemoryBarrierSignal attribute) (wave_lang.kernel.ops.wave_ops.SharedMemoryBarrierWait attribute) BinaryOpBase (class in wave_lang.kernel.ops.wave_ops) BinaryPyOp (class in wave_lang.kernel.ops.wave_ops) bind_node_proxies() (wave_lang.kernel.compiler.wave_codegen.WaveEmitter method) bind_node_proxy() (wave_lang.kernel.compiler.wave_codegen.WaveEmitter method) bitcast() (in module wave_lang.kernel.ops.wave_ops) BitcastOp (class in wave_lang.kernel.ops.wave_ops) block (wave_lang.kernel.ops.wave_ops.ReduceOp attribute) (wave_lang.kernel.ops.wave_ops.ScanOp attribute) bounds (wave_lang.kernel.ops.wave_ops.BoundsCheck attribute) (wave_lang.kernel.ops.wave_ops.Read attribute) (wave_lang.kernel.ops.wave_ops.ScatterAdd attribute) (wave_lang.kernel.ops.wave_ops.TensorLoadToLDS attribute) (wave_lang.kernel.ops.wave_ops.Write attribute) bounds_check() (in module wave_lang.kernel.ops.wave_ops) BoundsCheck (class in wave_lang.kernel.ops.wave_ops) Broadcast (class in wave_lang.kernel.ops.wave_ops) broadcast() (in module wave_lang.kernel.ops.wave_ops) built-in function add_conditional_barriers_to_loop() reorder_graph() schedule_reordering() transform_two_PP_clusters() C capture_source() (wave_lang.kernel.ops.wave_ops.NestedRegionOp static method) captured_vars() (wave_lang.kernel.ops.wave_ops.NestedRegionOp method) cast() (in module wave_lang.kernel.ops.wave_ops) CastOp (class in wave_lang.kernel.ops.wave_ops) Cbrt (class in wave_lang.kernel.ops.wave_ops) cbrt() (in module wave_lang.kernel.ops.wave_ops) clear() (wave_lang.runtime.Device method) ComparisonPyOp (class in wave_lang.kernel.ops.wave_ops) compile_target_flags (wave_lang.runtime.Device attribute) cond (wave_lang.kernel.ops.wave_ops.SelectOp attribute) condition (wave_lang.kernel.ops.wave_ops.Conditional attribute) (wave_lang.kernel.ops.wave_ops.Iterate attribute) Conditional (class in wave_lang.kernel.ops.wave_ops) conditional() (in module wave_lang.kernel.ops.wave_ops) constraints (wave_lang.kernel.compiler.wave_codegen.WaveEmitter attribute) copy() (wave_lang.kernel.ops.wave_ops.CustomOp method) copy_core_attributes() (wave_lang.kernel.ops.wave_ops.CustomOp method) Cos (class in wave_lang.kernel.ops.wave_ops) cos() (in module wave_lang.kernel.ops.wave_ops) count (wave_lang.kernel.ops.wave_ops.Iterate property) (wave_lang.kernel.ops.wave_ops.TensorCounterWait attribute) create() (wave_lang.kernel.ops.wave_ops.CustomOp class method) create_hal_module() (wave_lang.runtime.Device method) create_induction_vars() (wave_lang.kernel.wave.wave.LaunchableWave method) Cumsum (class in wave_lang.kernel.ops.wave_ops) cumsum() (in module wave_lang.kernel.ops.wave_ops) current() (wave_lang.runtime.Device static method) custom_string() (wave_lang.kernel.ops.wave_ops.CustomOp method) (wave_lang.kernel.ops.wave_ops.MMA method) (wave_lang.kernel.ops.wave_ops.Placeholder method) (wave_lang.kernel.ops.wave_ops.ScaledMMA method) (wave_lang.kernel.ops.wave_ops.Unknown method) CustomOp (class in wave_lang.kernel.ops.wave_ops) D debug_log() (in module wave_lang.kernel.ops.wave_ops) DebugFlags (class in wave_lang.support.debugging) DebugLog (class in wave_lang.kernel.ops.wave_ops) default_trace_tensor_callback() (in module wave_lang.support) DefaultFormatter (class in wave_lang.support.logging) define_interface_op() (in module wave_lang.kernel.ops.wave_ops) define_op() (in module wave_lang.kernel.ops.wave_ops) define_py_op() (in module wave_lang.kernel.ops.wave_ops) Device (class in wave_lang.runtime) device (wave_lang.runtime.DeviceState attribute) device_constraints (wave_lang.kernel.wave.wave.LaunchableWave property) DeviceState (class in wave_lang.runtime) dim (wave_lang.kernel.ops.wave_ops.ReduceOp attribute) (wave_lang.kernel.ops.wave_ops.ScanOp attribute) (wave_lang.kernel.ops.wave_ops.ScatterAdd attribute) (wave_lang.kernel.ops.wave_ops.SelfIndex attribute) (wave_lang.kernel.ops.wave_ops.TopkOp attribute) distributed_shape (wave_lang.kernel.ops.wave_ops.Allocate attribute) (wave_lang.kernel.ops.wave_ops.GetResult property) (wave_lang.kernel.ops.wave_ops.IterArg property) (wave_lang.kernel.ops.wave_ops.TensorLoadToLDS attribute) dlpack_device_type_code (wave_lang.runtime.DeviceState attribute) driver (wave_lang.runtime.DeviceState attribute) driver_id (wave_lang.runtime.Device attribute) ds (wave_lang.kernel.ops.wave_ops.MemoryCounterWait attribute) (wave_lang.kernel.ops.wave_ops.MemoryCounterWaitBarrier attribute) ds_wait (wave_lang.kernel.ops.wave_ops.SharedMemoryBarrierSignal attribute) dst (wave_lang.kernel.ops.wave_ops.GatherToLDS attribute) (wave_lang.kernel.ops.wave_ops.TensorLoadToLDS attribute) dst_index (wave_lang.kernel.ops.wave_ops.GatherToLDS attribute) dst_mapping (wave_lang.kernel.ops.wave_ops.GatherToLDS attribute) dst_mapping_dynamic_vals (wave_lang.kernel.ops.wave_ops.GatherToLDS attribute) dtype (wave_lang.kernel.ops.wave_ops.Allocate attribute) (wave_lang.kernel.ops.wave_ops.BitcastOp attribute) (wave_lang.kernel.ops.wave_ops.CastOp attribute) (wave_lang.kernel.ops.wave_ops.GatherToLDS attribute) (wave_lang.kernel.ops.wave_ops.NewRegister attribute) (wave_lang.kernel.ops.wave_ops.NewScalar attribute) (wave_lang.kernel.ops.wave_ops.Read property) (wave_lang.kernel.ops.wave_ops.SelfIndex attribute) dtype_to_element_type() (in module wave_lang.support.conversions) dump_device_info() (wave_lang.runtime.Device method) E eager_execute() (wave_lang.kernel.wave.wave.LaunchableWave method) element_type (wave_lang.kernel.ops.wave_ops.TensorLoadToLDS attribute) elements_per_thread (wave_lang.kernel.ops.wave_ops.AtomicOp attribute) (wave_lang.kernel.ops.wave_ops.GatherToLDS attribute) (wave_lang.kernel.ops.wave_ops.Read attribute) (wave_lang.kernel.ops.wave_ops.ScatterAdd attribute) (wave_lang.kernel.ops.wave_ops.SelfIndex attribute) (wave_lang.kernel.ops.wave_ops.Write attribute) else_return (wave_lang.kernel.ops.wave_ops.Conditional attribute) emit() (wave_lang.kernel.compiler.wave_codegen.WaveEmitter method) emit_func() (wave_lang.kernel.compiler.wave_codegen.WaveEmitter method) emit_host_func() (wave_lang.kernel.compiler.wave_codegen.WaveEmitter method) emit_program_invariants() (wave_lang.kernel.compiler.wave_codegen.WaveEmitter method) enumerated_device_id (wave_lang.runtime.DeviceState property) enumerated_info (wave_lang.runtime.DeviceState attribute) enumerated_name (wave_lang.runtime.DeviceState property) enumerated_path (wave_lang.runtime.DeviceState property) Eq (class in wave_lang.kernel.ops.wave_ops) eq() (in module wave_lang.kernel.ops.wave_ops) erase() (wave_lang.kernel.ops.wave_ops.CustomOp method) (wave_lang.kernel.ops.wave_ops.NestedRegionOp method) (wave_lang.kernel.ops.wave_ops.Placeholder method) Exp (class in wave_lang.kernel.ops.wave_ops) exp (wave_lang.kernel.ops.wave_ops.MemoryCounterWait attribute) (wave_lang.kernel.ops.wave_ops.MemoryCounterWaitBarrier attribute) exp() (in module wave_lang.kernel.ops.wave_ops) Exp2 (class in wave_lang.kernel.ops.wave_ops) exp2() (in module wave_lang.kernel.ops.wave_ops) expanded_dims (wave_lang.kernel.ops.wave_ops.CustomOp property) export_torch_tensor (wave_lang.runtime.Device attribute) expr (wave_lang.kernel.ops.wave_ops.ApplyExpr attribute) extra_iteration_dimensions (wave_lang.kernel.ops.wave_ops.DebugLog attribute) Extract (class in wave_lang.kernel.ops.wave_ops) extract() (in module wave_lang.kernel.ops.wave_ops) extract_slice() (in module wave_lang.kernel.ops.wave_ops) ExtractSlice (class in wave_lang.kernel.ops.wave_ops) F finalize_iree_action() (wave_lang.runtime.Device method) flags (wave_lang.kernel.ops.wave_ops.Read attribute) (wave_lang.kernel.ops.wave_ops.Write attribute) Floordiv (class in wave_lang.kernel.ops.wave_ops) from_file_cache_only() (wave_lang.runtime.Launchable static method) from_fx_node() (wave_lang.kernel.ops.wave_ops.CustomOp class method) (wave_lang.kernel.ops.wave_ops.Output class method) (wave_lang.kernel.ops.wave_ops.Placeholder class method) (wave_lang.kernel.ops.wave_ops.Unknown class method) from_uri() (wave_lang.runtime.DeviceState static method) from_vm_module() (wave_lang.runtime.Launchable static method) fx_node (wave_lang.kernel.ops.wave_ops.CustomOp attribute) G gather_to_lds() (in module wave_lang.kernel.ops.wave_ops) GatherToLDS (class in wave_lang.kernel.ops.wave_ops) Ge (class in wave_lang.kernel.ops.wave_ops) ge() (in module wave_lang.kernel.ops.wave_ops) get_capture_bindings() (wave_lang.kernel.ops.wave_ops.NestedRegionOp method) get_captured_fx_node() (wave_lang.kernel.ops.wave_ops.NestedRegionOp method) (wave_lang.kernel.ops.wave_ops.Placeholder method) get_custom() (in module wave_lang.kernel.ops.wave_ops) get_derived_indices() (wave_lang.kernel.ops.wave_ops.Read method) (wave_lang.kernel.ops.wave_ops.Write method) get_induction_vars_and_syms() (wave_lang.kernel.compiler.wave_codegen.WaveEmitter method) get_location_capture_config() (in module wave_lang.support) get_logger() (in module wave_lang.support.logging) get_node_arg_index() (wave_lang.kernel.ops.wave_ops.CustomOp method) get_result() (in module wave_lang.kernel.ops.wave_ops) get_root_graph() (wave_lang.kernel.ops.wave_ops.NestedRegionOp method) get_shape_from_bindings() (in module wave_lang.kernel.ops.wave_ops) get_type_key_hash() (wave_lang.runtime.Device method) get_vm_instance() (in module wave_lang.runtime) get_workgroup_dims() (wave_lang.kernel.wave.wave.LaunchableWave method) GetResult (class in wave_lang.kernel.ops.wave_ops) global_tile_index (wave_lang.kernel.ops.wave_ops.TensorLoadToLDS attribute) graph (wave_lang.kernel.ops.wave_ops.CustomOp attribute) grid (wave_lang.kernel.compiler.wave_codegen.WaveEmitter attribute) Gt (class in wave_lang.kernel.ops.wave_ops) gt() (in module wave_lang.kernel.ops.wave_ops) H hal_device (wave_lang.runtime.Device property) handle() (wave_lang.kernel.ops.wave_ops.CustomOp class method) (wave_lang.kernel.ops.wave_ops.NestedRegionOp class method) handler (wave_lang.kernel.ops.wave_ops.DebugLog attribute) hardware_constraint (wave_lang.kernel.compiler.wave_codegen.WaveEmitter property) hardware_constraints (wave_lang.kernel.wave.wave.LaunchableWave property) has_identity_mapping() (wave_lang.kernel.ops.wave_ops.Read method) (wave_lang.kernel.ops.wave_ops.Write method) has_same_custom_type() (in module wave_lang.kernel.ops.wave_ops) has_side_effects (wave_lang.kernel.ops.wave_ops.AtomicOp property) (wave_lang.kernel.ops.wave_ops.BoundsCheck property) (wave_lang.kernel.ops.wave_ops.CustomOp property) (wave_lang.kernel.ops.wave_ops.DebugLog property) (wave_lang.kernel.ops.wave_ops.GatherToLDS property) (wave_lang.kernel.ops.wave_ops.MemoryCounterWait property) (wave_lang.kernel.ops.wave_ops.MemoryCounterWaitBarrier property) (wave_lang.kernel.ops.wave_ops.Output property) (wave_lang.kernel.ops.wave_ops.ScatterAdd property) (wave_lang.kernel.ops.wave_ops.SetSymbol property) (wave_lang.kernel.ops.wave_ops.SetWavePrio property) (wave_lang.kernel.ops.wave_ops.SharedMemoryBarrier property) (wave_lang.kernel.ops.wave_ops.SharedMemoryBarrierSignal property) (wave_lang.kernel.ops.wave_ops.SharedMemoryBarrierWait property) (wave_lang.kernel.ops.wave_ops.TensorCounterWait property) (wave_lang.kernel.ops.wave_ops.TensorLoadToLDS property) (wave_lang.kernel.ops.wave_ops.WorkgroupBarrier property) head_dim (wave_lang.kernel.ops.wave_ops.SoftsignOp attribute) I if_false (wave_lang.kernel.ops.wave_ops.SelectOp attribute) if_true (wave_lang.kernel.ops.wave_ops.SelectOp attribute) implicit_captures (wave_lang.kernel.ops.wave_ops.Conditional attribute) (wave_lang.kernel.ops.wave_ops.Iterate attribute) import_torch_tensor (wave_lang.runtime.Device attribute) index (wave_lang.kernel.ops.wave_ops.CustomOp property) (wave_lang.kernel.ops.wave_ops.GetResult property) (wave_lang.kernel.ops.wave_ops.Iterate property) (wave_lang.kernel.ops.wave_ops.Placeholder property) index_exprs (wave_lang.kernel.ops.wave_ops.BoundsCheck attribute) indexing_dims (wave_lang.kernel.ops.wave_ops.Allocate property) (wave_lang.kernel.ops.wave_ops.ApplyExpr property) (wave_lang.kernel.ops.wave_ops.AtomicOp property) (wave_lang.kernel.ops.wave_ops.BinaryOpBase property) (wave_lang.kernel.ops.wave_ops.BitcastOp property) (wave_lang.kernel.ops.wave_ops.Broadcast property) (wave_lang.kernel.ops.wave_ops.CastOp property) (wave_lang.kernel.ops.wave_ops.Conditional property) (wave_lang.kernel.ops.wave_ops.CustomOp property) (wave_lang.kernel.ops.wave_ops.GetResult property) (wave_lang.kernel.ops.wave_ops.Iterate property) (wave_lang.kernel.ops.wave_ops.MMA property) (wave_lang.kernel.ops.wave_ops.NewRegister property) (wave_lang.kernel.ops.wave_ops.NewScalar property) (wave_lang.kernel.ops.wave_ops.Permute property) (wave_lang.kernel.ops.wave_ops.Placeholder property) (wave_lang.kernel.ops.wave_ops.Read property) (wave_lang.kernel.ops.wave_ops.ReduceOp property) (wave_lang.kernel.ops.wave_ops.Reshape property) (wave_lang.kernel.ops.wave_ops.ScaledMMA property) (wave_lang.kernel.ops.wave_ops.ScanOp property) (wave_lang.kernel.ops.wave_ops.ScatterAdd property) (wave_lang.kernel.ops.wave_ops.SelectOp property) (wave_lang.kernel.ops.wave_ops.SelfIndex property) (wave_lang.kernel.ops.wave_ops.SetSymbol property) (wave_lang.kernel.ops.wave_ops.ShuffleOp property) (wave_lang.kernel.ops.wave_ops.SoftsignOp property) (wave_lang.kernel.ops.wave_ops.TopkOp property) (wave_lang.kernel.ops.wave_ops.UnaryPyOp property) (wave_lang.kernel.ops.wave_ops.Write property) infer_broadcast_shape() (wave_lang.kernel.ops.wave_ops.SelectOp method) infer_device_layout() (wave_lang.kernel.wave.wave.LaunchableWave method) infer_grid_shape() (wave_lang.kernel.wave.wave.LaunchableWave method) infer_shape() (wave_lang.kernel.ops.wave_ops.BinaryOpBase method) infer_type() (wave_lang.kernel.ops.wave_ops.Allocate method) (wave_lang.kernel.ops.wave_ops.AtomicOp method) (wave_lang.kernel.ops.wave_ops.BinaryPyOp method) (wave_lang.kernel.ops.wave_ops.BitcastOp method) (wave_lang.kernel.ops.wave_ops.Broadcast method) (wave_lang.kernel.ops.wave_ops.CastOp method) (wave_lang.kernel.ops.wave_ops.ComparisonPyOp method) (wave_lang.kernel.ops.wave_ops.CustomOp method) (wave_lang.kernel.ops.wave_ops.DebugLog method) (wave_lang.kernel.ops.wave_ops.Extract method) (wave_lang.kernel.ops.wave_ops.GetResult method) (wave_lang.kernel.ops.wave_ops.IterArg method) (wave_lang.kernel.ops.wave_ops.MMA method) (wave_lang.kernel.ops.wave_ops.NestedRegionOp method) (wave_lang.kernel.ops.wave_ops.NewRegister method) (wave_lang.kernel.ops.wave_ops.NewScalar method) (wave_lang.kernel.ops.wave_ops.Permute method) (wave_lang.kernel.ops.wave_ops.Placeholder method) (wave_lang.kernel.ops.wave_ops.Read method) (wave_lang.kernel.ops.wave_ops.ReduceOp method) (wave_lang.kernel.ops.wave_ops.Reshape method) (wave_lang.kernel.ops.wave_ops.ScaledMMA method) (wave_lang.kernel.ops.wave_ops.ScanOp method) (wave_lang.kernel.ops.wave_ops.ScatterAdd method) (wave_lang.kernel.ops.wave_ops.SelectOp method) (wave_lang.kernel.ops.wave_ops.ShuffleOp method) (wave_lang.kernel.ops.wave_ops.SoftsignOp method) (wave_lang.kernel.ops.wave_ops.TopkOp method) (wave_lang.kernel.ops.wave_ops.UnaryPyOp method) (wave_lang.kernel.ops.wave_ops.Write method) init (wave_lang.kernel.ops.wave_ops.ReduceOp attribute) (wave_lang.kernel.ops.wave_ops.ScanOp attribute) (wave_lang.kernel.ops.wave_ops.TopkOp property) init_args (wave_lang.kernel.ops.wave_ops.Conditional property) (wave_lang.kernel.ops.wave_ops.Iterate attribute) initialize_reductions() (wave_lang.kernel.wave.wave.LaunchableWave method) initialize_symbolic_constraints() (wave_lang.kernel.wave.wave.LaunchableWave method) initialize_wave_constraints() (wave_lang.kernel.wave.wave.LaunchableWave method) initialize_workgroup_constraints() (wave_lang.kernel.wave.wave.LaunchableWave method) input_selector (wave_lang.kernel.ops.wave_ops.TensorLoadToLDS attribute) instance (wave_lang.runtime.DeviceState attribute) instance_cache_key (wave_lang.runtime.Device attribute) instructions (wave_lang.kernel.ops.wave_ops.SchedulingGroupBarrier attribute) Invert (class in wave_lang.kernel.ops.wave_ops) invoke_vm_function() (in module wave_lang.runtime) is_contiguous_vec() (wave_lang.kernel.ops.wave_ops.Read method) (wave_lang.kernel.ops.wave_ops.Write method) iter_args() (wave_lang.kernel.ops.wave_ops.NestedRegionOp method) iter_idx (wave_lang.kernel.ops.wave_ops.IterArg property) IterArg (class in wave_lang.kernel.ops.wave_ops) Iterate (class in wave_lang.kernel.ops.wave_ops) iterate() (in module wave_lang.kernel.ops.wave_ops) J jit_compile() (wave_lang.runtime.Launchable static method) K k_dim (wave_lang.kernel.ops.wave_ops.TopkOp attribute) kernel_name (wave_lang.kernel.compiler.wave_codegen.WaveEmitter attribute) kwargs (wave_lang.kernel.ops.wave_ops.Unknown attribute) L label (wave_lang.kernel.ops.wave_ops.DebugLog attribute) Launchable (class in wave_lang.runtime) LaunchableWave (class in wave_lang.kernel.wave.wave) Le (class in wave_lang.kernel.ops.wave_ops) le() (in module wave_lang.kernel.ops.wave_ops) lhs (wave_lang.kernel.ops.wave_ops.BinaryOpBase attribute) (wave_lang.kernel.ops.wave_ops.MMA attribute) (wave_lang.kernel.ops.wave_ops.ScaledMMA attribute) lhs_index (wave_lang.kernel.ops.wave_ops.MMA property) (wave_lang.kernel.ops.wave_ops.ScaledMMA property) lhs_scale (wave_lang.kernel.ops.wave_ops.ScaledMMA attribute) lhs_scale_index (wave_lang.kernel.ops.wave_ops.ScaledMMA property) lhs_scale_type (wave_lang.kernel.ops.wave_ops.ScaledMMA property) lhs_type (wave_lang.kernel.ops.wave_ops.MMA property) (wave_lang.kernel.ops.wave_ops.ScaledMMA property) load (wave_lang.kernel.ops.wave_ops.MemoryCounterWait attribute) (wave_lang.kernel.ops.wave_ops.MemoryCounterWaitBarrier attribute) location (wave_lang.kernel.ops.wave_ops.CustomOp property) location_level (wave_lang.support.debugging.DebugFlags attribute) Log10 (class in wave_lang.kernel.ops.wave_ops) log10() (in module wave_lang.kernel.ops.wave_ops) Log2 (class in wave_lang.kernel.ops.wave_ops) log2() (in module wave_lang.kernel.ops.wave_ops) log_level (wave_lang.support.debugging.DebugFlags attribute) logical_slice (wave_lang.kernel.ops.wave_ops.Reshape attribute) logit_cap (wave_lang.kernel.ops.wave_ops.SoftsignOp attribute) lookup_node_values() (wave_lang.kernel.compiler.wave_codegen.WaveEmitter method) Lt (class in wave_lang.kernel.ops.wave_ops) lt() (in module wave_lang.kernel.ops.wave_ops) M mapping (wave_lang.kernel.ops.wave_ops.AtomicOp attribute) (wave_lang.kernel.ops.wave_ops.BoundsCheck attribute) (wave_lang.kernel.ops.wave_ops.DebugLog attribute) (wave_lang.kernel.ops.wave_ops.Read attribute) (wave_lang.kernel.ops.wave_ops.ScatterAdd attribute) (wave_lang.kernel.ops.wave_ops.Write attribute) mapping_dynamic_vals (wave_lang.kernel.ops.wave_ops.AtomicOp attribute) (wave_lang.kernel.ops.wave_ops.BoundsCheck attribute) (wave_lang.kernel.ops.wave_ops.DebugLog attribute) (wave_lang.kernel.ops.wave_ops.Read attribute) (wave_lang.kernel.ops.wave_ops.Write attribute) mask_bounds (wave_lang.kernel.ops.wave_ops.BoundsCheck attribute) materialize_capture_placeholder() (wave_lang.kernel.ops.wave_ops.NestedRegionOp class method) Max (class in wave_lang.kernel.ops.wave_ops) max() (in module wave_lang.kernel.ops.wave_ops) Maximum (class in wave_lang.kernel.ops.wave_ops) maximum() (in module wave_lang.kernel.ops.wave_ops) memory (wave_lang.kernel.ops.wave_ops.DebugLog property) (wave_lang.kernel.ops.wave_ops.Read attribute) (wave_lang.kernel.ops.wave_ops.ScatterAdd attribute) (wave_lang.kernel.ops.wave_ops.Write attribute) memory_counter_wait() (in module wave_lang.kernel.ops.wave_ops) memory_counter_wait_barrier() (in module wave_lang.kernel.ops.wave_ops) memory_type (wave_lang.kernel.ops.wave_ops.AtomicOp property) (wave_lang.kernel.ops.wave_ops.Read property) (wave_lang.kernel.ops.wave_ops.ScatterAdd property) (wave_lang.kernel.ops.wave_ops.Write property) MemoryAccessFlags (class in wave_lang.kernel.ops.wave_ops) MemoryCounterWait (class in wave_lang.kernel.ops.wave_ops) MemoryCounterWaitBarrier (class in wave_lang.kernel.ops.wave_ops) Min (class in wave_lang.kernel.ops.wave_ops) min() (in module wave_lang.kernel.ops.wave_ops) Minimum (class in wave_lang.kernel.ops.wave_ops) minimum() (in module wave_lang.kernel.ops.wave_ops) MMA (class in wave_lang.kernel.ops.wave_ops) mma() (in module wave_lang.kernel.ops.wave_ops) mma_type (wave_lang.kernel.ops.wave_ops.MMA attribute) (wave_lang.kernel.ops.wave_ops.ScaledMMA attribute) MMABase (class in wave_lang.kernel.ops.wave_ops) Mod (class in wave_lang.kernel.ops.wave_ops) mod() (in module wave_lang.kernel.ops.wave_ops) mode (wave_lang.kernel.ops.wave_ops.ShuffleOp attribute) module wave.tools wave_lang.kernel.compiler.wave_codegen wave_lang.kernel.ops.wave_ops wave_lang.kernel.wave.wave wave_lang.runtime wave_lang.support wave_lang.support.conversions wave_lang.support.debugging wave_lang.support.logging wave_lang.support.tools Mul (class in wave_lang.kernel.ops.wave_ops) multicast_mask (wave_lang.kernel.ops.wave_ops.TensorLoadToLDS attribute) N name (wave_lang.kernel.ops.wave_ops.CustomOp property) (wave_lang.kernel.wave.wave.LaunchableWave property) Ne (class in wave_lang.kernel.ops.wave_ops) ne() (in module wave_lang.kernel.ops.wave_ops) Neg (class in wave_lang.kernel.ops.wave_ops) NestedRegionOp (class in wave_lang.kernel.ops.wave_ops) NewRegister (class in wave_lang.kernel.ops.wave_ops) NewScalar (class in wave_lang.kernel.ops.wave_ops) node_args (wave_lang.kernel.ops.wave_ops.CustomOp property) NONE (wave_lang.kernel.ops.wave_ops.MemoryAccessFlags attribute) NONTEMPORAL (wave_lang.kernel.ops.wave_ops.MemoryAccessFlags attribute) num_slices (wave_lang.kernel.ops.wave_ops.Reshape attribute) O offset (wave_lang.kernel.ops.wave_ops.Allocate attribute) (wave_lang.kernel.ops.wave_ops.Extract attribute) (wave_lang.kernel.ops.wave_ops.ExtractSlice attribute) (wave_lang.kernel.ops.wave_ops.ShuffleOp attribute) OP_HANDLERS (wave_lang.kernel.compiler.wave_codegen.WaveEmitter attribute) operand_index() (wave_lang.kernel.ops.wave_ops.MMA method) (wave_lang.kernel.ops.wave_ops.ScaledMMA method) operations (wave_lang.kernel.ops.wave_ops.SchedulingBarrier attribute) options (wave_lang.kernel.compiler.wave_codegen.WaveEmitter attribute) Or (class in wave_lang.kernel.ops.wave_ops) Output (class in wave_lang.kernel.ops.wave_ops) outputs() (wave_lang.kernel.ops.wave_ops.NestedRegionOp method) P padding (wave_lang.kernel.ops.wave_ops.Allocate attribute) parent (wave_lang.kernel.ops.wave_ops.Allocate attribute) parent_op() (wave_lang.kernel.ops.wave_ops.IterArg method) parse() (wave_lang.support.debugging.DebugFlags static method) parse_from_env() (wave_lang.support.debugging.DebugFlags static method) Permute (class in wave_lang.kernel.ops.wave_ops) permute() (in module wave_lang.kernel.ops.wave_ops) Placeholder (class in wave_lang.kernel.ops.wave_ops) Powf (class in wave_lang.kernel.ops.wave_ops) powf() (in module wave_lang.kernel.ops.wave_ops) pre_expansion_id (wave_lang.kernel.ops.wave_ops.CustomOp property) preload() (wave_lang.runtime.Launchable method) printer() (wave_lang.kernel.ops.wave_ops.DebugLog method) priority (wave_lang.kernel.ops.wave_ops.SetWavePrio attribute) py_operator (wave_lang.kernel.ops.wave_ops.BinaryOpBase property) (wave_lang.kernel.ops.wave_ops.UnaryPyOp property) R rank (wave_lang.kernel.ops.wave_ops.ExtractSlice property) Read (class in wave_lang.kernel.ops.wave_ops) read() (in module wave_lang.kernel.ops.wave_ops) read_meets_hw_transpose_requirements() (in module wave_lang.kernel.ops.wave_ops) Reciprocal (class in wave_lang.kernel.ops.wave_ops) reciprocal() (in module wave_lang.kernel.ops.wave_ops) ReduceOp (class in wave_lang.kernel.ops.wave_ops) reduction_dim (wave_lang.kernel.ops.wave_ops.MMA property) (wave_lang.kernel.ops.wave_ops.ReduceOp property) (wave_lang.kernel.ops.wave_ops.ScaledMMA property) (wave_lang.kernel.ops.wave_ops.TopkOp property) refresh_captures() (wave_lang.kernel.ops.wave_ops.NestedRegionOp method) register() (in module wave_lang.kernel.ops.wave_ops) register_ (wave_lang.kernel.ops.wave_ops.ApplyExpr attribute) (wave_lang.kernel.ops.wave_ops.DebugLog attribute) (wave_lang.kernel.ops.wave_ops.Extract attribute) (wave_lang.kernel.ops.wave_ops.ExtractSlice attribute) (wave_lang.kernel.ops.wave_ops.SetSymbol attribute) (wave_lang.kernel.ops.wave_ops.Write attribute) register_idx (wave_lang.kernel.ops.wave_ops.ScatterAdd attribute) register_index (wave_lang.kernel.ops.wave_ops.ScatterAdd property) (wave_lang.kernel.ops.wave_ops.Write property) register_src (wave_lang.kernel.ops.wave_ops.ScatterAdd attribute) register_type (wave_lang.kernel.ops.wave_ops.ScatterAdd property) (wave_lang.kernel.ops.wave_ops.Write property) Remf (class in wave_lang.kernel.ops.wave_ops) remf() (in module wave_lang.kernel.ops.wave_ops) reorder_graph() built-in function reordering_constraints (wave_lang.kernel.wave.wave.LaunchableWave property) replace_all_uses_with() (wave_lang.kernel.ops.wave_ops.CustomOp method) replace_all_uses_with_except() (wave_lang.kernel.ops.wave_ops.CustomOp method) replace_uses_with() (wave_lang.kernel.ops.wave_ops.CustomOp method) replacement_location_propagate() (wave_lang.kernel.ops.wave_ops.CustomOp method) res_idx (wave_lang.kernel.ops.wave_ops.GetResult attribute) Reshape (class in wave_lang.kernel.ops.wave_ops) reshape() (in module wave_lang.kernel.ops.wave_ops) return_vals (wave_lang.kernel.ops.wave_ops.Output attribute) rhs (wave_lang.kernel.ops.wave_ops.BinaryOpBase attribute) (wave_lang.kernel.ops.wave_ops.MMA attribute) (wave_lang.kernel.ops.wave_ops.ScaledMMA attribute) rhs_index (wave_lang.kernel.ops.wave_ops.MMA property) (wave_lang.kernel.ops.wave_ops.ScaledMMA property) rhs_scale (wave_lang.kernel.ops.wave_ops.ScaledMMA attribute) rhs_scale_index (wave_lang.kernel.ops.wave_ops.ScaledMMA property) rhs_scale_type (wave_lang.kernel.ops.wave_ops.ScaledMMA property) rhs_type (wave_lang.kernel.ops.wave_ops.MMA property) (wave_lang.kernel.ops.wave_ops.ScaledMMA property) root_sig (wave_lang.kernel.compiler.wave_codegen.WaveEmitter attribute) Round (class in wave_lang.kernel.ops.wave_ops) round() (in module wave_lang.kernel.ops.wave_ops) Roundeven (class in wave_lang.kernel.ops.wave_ops) roundeven() (in module wave_lang.kernel.ops.wave_ops) rrt (wave_lang.kernel.ops.wave_ops.CustomOp property) Rsqrt (class in wave_lang.kernel.ops.wave_ops) rsqrt() (in module wave_lang.kernel.ops.wave_ops) run_manual_schedule() (wave_lang.kernel.wave.wave.LaunchableWave method) runtime_trace_dir (wave_lang.support.debugging.DebugFlags attribute) S scalar() (in module wave_lang.kernel.ops.wave_ops) scale_factor (wave_lang.kernel.ops.wave_ops.BitcastOp property) scaled_mma() (in module wave_lang.kernel.ops.wave_ops) ScaledMMA (class in wave_lang.kernel.ops.wave_ops) scan_dim (wave_lang.kernel.ops.wave_ops.ScanOp property) ScanOp (class in wave_lang.kernel.ops.wave_ops) scatter_add() (in module wave_lang.kernel.ops.wave_ops) ScatterAdd (class in wave_lang.kernel.ops.wave_ops) schedule_reordering() built-in function scheduling_barrier() (in module wave_lang.kernel.ops.wave_ops) scheduling_group_barrier() (in module wave_lang.kernel.ops.wave_ops) scheduling_parameters (wave_lang.kernel.ops.wave_ops.CustomOp property) SchedulingBarrier (class in wave_lang.kernel.ops.wave_ops) SchedulingGroupBarrier (class in wave_lang.kernel.ops.wave_ops) select() (in module wave_lang.kernel.ops.wave_ops) SelectOp (class in wave_lang.kernel.ops.wave_ops) self_index() (in module wave_lang.kernel.ops.wave_ops) SelfIndex (class in wave_lang.kernel.ops.wave_ops) set() (wave_lang.runtime.Device method) (wave_lang.support.debugging.DebugFlags method) set_symbol() (in module wave_lang.kernel.ops.wave_ops) set_wave_prio() (in module wave_lang.kernel.ops.wave_ops) SetSymbol (class in wave_lang.kernel.ops.wave_ops) setup_iree_action() (wave_lang.runtime.Device method) SetWavePrio (class in wave_lang.kernel.ops.wave_ops) shape (wave_lang.kernel.ops.wave_ops.Allocate attribute) (wave_lang.kernel.ops.wave_ops.NewRegister attribute) shared_memory_barrier() (in module wave_lang.kernel.ops.wave_ops) shared_memory_barrier_signal() (in module wave_lang.kernel.ops.wave_ops) shared_memory_barrier_wait() (in module wave_lang.kernel.ops.wave_ops) shared_tile_index (wave_lang.kernel.ops.wave_ops.TensorLoadToLDS attribute) SharedMemoryBarrier (class in wave_lang.kernel.ops.wave_ops) SharedMemoryBarrierSignal (class in wave_lang.kernel.ops.wave_ops) SharedMemoryBarrierWait (class in wave_lang.kernel.ops.wave_ops) shuffle() (in module wave_lang.kernel.ops.wave_ops) ShuffleOp (class in wave_lang.kernel.ops.wave_ops) Sin (class in wave_lang.kernel.ops.wave_ops) sin() (in module wave_lang.kernel.ops.wave_ops) Sinh (class in wave_lang.kernel.ops.wave_ops) sinh() (in module wave_lang.kernel.ops.wave_ops) size (wave_lang.kernel.ops.wave_ops.ExtractSlice attribute) Softsign (class in wave_lang.kernel.ops.wave_ops) softsign() (in module wave_lang.kernel.ops.wave_ops) SoftsignOp (class in wave_lang.kernel.ops.wave_ops) source (wave_lang.kernel.ops.wave_ops.Read attribute) (wave_lang.kernel.ops.wave_ops.Write attribute) Sqrt (class in wave_lang.kernel.ops.wave_ops) sqrt() (in module wave_lang.kernel.ops.wave_ops) src (wave_lang.kernel.ops.wave_ops.GatherToLDS attribute) (wave_lang.kernel.ops.wave_ops.TensorLoadToLDS attribute) src_bounds (wave_lang.kernel.ops.wave_ops.GatherToLDS attribute) src_index (wave_lang.kernel.ops.wave_ops.GatherToLDS attribute) src_mapping (wave_lang.kernel.ops.wave_ops.GatherToLDS attribute) src_mapping_dynamic_vals (wave_lang.kernel.ops.wave_ops.GatherToLDS attribute) start (wave_lang.kernel.ops.wave_ops.Iterate attribute) step (wave_lang.kernel.ops.wave_ops.Iterate attribute) store (wave_lang.kernel.ops.wave_ops.MemoryCounterWait attribute) (wave_lang.kernel.ops.wave_ops.MemoryCounterWaitBarrier attribute) stride (wave_lang.kernel.ops.wave_ops.ExtractSlice attribute) Sub (class in wave_lang.kernel.ops.wave_ops) subgraph_name (wave_lang.kernel.ops.wave_ops.Conditional attribute) (wave_lang.kernel.ops.wave_ops.Iterate attribute) Sum (class in wave_lang.kernel.ops.wave_ops) sum() (in module wave_lang.kernel.ops.wave_ops) symbol (wave_lang.kernel.ops.wave_ops.SetSymbol attribute) symbolic_constraints (wave_lang.kernel.wave.wave.LaunchableWave property) sync (wave_lang.runtime.Device property) sync_id (wave_lang.kernel.ops.wave_ops.SchedulingGroupBarrier attribute) T tag (wave_lang.kernel.ops.wave_ops.CustomOp property) tag() (in module wave_lang.kernel.ops.wave_ops) tail_padding (wave_lang.kernel.ops.wave_ops.Allocate attribute) Tanh (class in wave_lang.kernel.ops.wave_ops) tanh() (in module wave_lang.kernel.ops.wave_ops) tanh_approx() (in module wave_lang.kernel.ops.wave_ops) TanhApprox (class in wave_lang.kernel.ops.wave_ops) target (wave_lang.kernel.ops.wave_ops.Read attribute) (wave_lang.kernel.ops.wave_ops.Write attribute) target_shape (wave_lang.kernel.ops.wave_ops.Broadcast attribute) (wave_lang.kernel.ops.wave_ops.Permute attribute) target_vector_shape (wave_lang.kernel.ops.wave_ops.Reshape attribute) tensor_counter_wait() (in module wave_lang.kernel.ops.wave_ops) tensor_load_to_lds() (in module wave_lang.kernel.ops.wave_ops) tensor_wait (wave_lang.kernel.ops.wave_ops.SharedMemoryBarrier attribute) (wave_lang.kernel.ops.wave_ops.SharedMemoryBarrierSignal attribute) TensorCounterWait (class in wave_lang.kernel.ops.wave_ops) TensorLoadToLDS (class in wave_lang.kernel.ops.wave_ops) tiling_constraints (wave_lang.kernel.wave.wave.LaunchableWave property) tkw_op_name (wave_lang.kernel.ops.wave_ops.Abs attribute) (wave_lang.kernel.ops.wave_ops.Add attribute) (wave_lang.kernel.ops.wave_ops.Allocate attribute) (wave_lang.kernel.ops.wave_ops.And attribute) (wave_lang.kernel.ops.wave_ops.ApplyExpr attribute) (wave_lang.kernel.ops.wave_ops.Atan2 attribute) (wave_lang.kernel.ops.wave_ops.AtomicAddOp attribute) (wave_lang.kernel.ops.wave_ops.AtomicOp attribute) (wave_lang.kernel.ops.wave_ops.BitcastOp attribute) (wave_lang.kernel.ops.wave_ops.BoundsCheck attribute) (wave_lang.kernel.ops.wave_ops.Broadcast attribute) (wave_lang.kernel.ops.wave_ops.CastOp attribute) (wave_lang.kernel.ops.wave_ops.Cbrt attribute) (wave_lang.kernel.ops.wave_ops.Conditional attribute) (wave_lang.kernel.ops.wave_ops.Cos attribute) (wave_lang.kernel.ops.wave_ops.Cumsum attribute) (wave_lang.kernel.ops.wave_ops.CustomOp attribute) (wave_lang.kernel.ops.wave_ops.DebugLog attribute) (wave_lang.kernel.ops.wave_ops.Eq attribute) (wave_lang.kernel.ops.wave_ops.Exp attribute) (wave_lang.kernel.ops.wave_ops.Exp2 attribute) (wave_lang.kernel.ops.wave_ops.Extract attribute) (wave_lang.kernel.ops.wave_ops.ExtractSlice attribute) (wave_lang.kernel.ops.wave_ops.Floordiv attribute) (wave_lang.kernel.ops.wave_ops.GatherToLDS attribute) (wave_lang.kernel.ops.wave_ops.Ge attribute) (wave_lang.kernel.ops.wave_ops.GetResult attribute) (wave_lang.kernel.ops.wave_ops.Gt attribute) (wave_lang.kernel.ops.wave_ops.Invert attribute) (wave_lang.kernel.ops.wave_ops.Iterate attribute) (wave_lang.kernel.ops.wave_ops.Le attribute) (wave_lang.kernel.ops.wave_ops.Log10 attribute) (wave_lang.kernel.ops.wave_ops.Log2 attribute) (wave_lang.kernel.ops.wave_ops.Lt attribute) (wave_lang.kernel.ops.wave_ops.Max attribute) (wave_lang.kernel.ops.wave_ops.Maximum attribute) (wave_lang.kernel.ops.wave_ops.MemoryCounterWait attribute) (wave_lang.kernel.ops.wave_ops.MemoryCounterWaitBarrier attribute) (wave_lang.kernel.ops.wave_ops.Min attribute) (wave_lang.kernel.ops.wave_ops.Minimum attribute) (wave_lang.kernel.ops.wave_ops.MMA attribute) (wave_lang.kernel.ops.wave_ops.Mod attribute) (wave_lang.kernel.ops.wave_ops.Mul attribute) (wave_lang.kernel.ops.wave_ops.Ne attribute) (wave_lang.kernel.ops.wave_ops.Neg attribute) (wave_lang.kernel.ops.wave_ops.NewRegister attribute) (wave_lang.kernel.ops.wave_ops.NewScalar attribute) (wave_lang.kernel.ops.wave_ops.Or attribute) (wave_lang.kernel.ops.wave_ops.Output attribute) (wave_lang.kernel.ops.wave_ops.Permute attribute) (wave_lang.kernel.ops.wave_ops.Placeholder attribute) (wave_lang.kernel.ops.wave_ops.Powf attribute) (wave_lang.kernel.ops.wave_ops.Read attribute) (wave_lang.kernel.ops.wave_ops.Reciprocal attribute) (wave_lang.kernel.ops.wave_ops.Remf attribute) (wave_lang.kernel.ops.wave_ops.Reshape attribute) (wave_lang.kernel.ops.wave_ops.Round attribute) (wave_lang.kernel.ops.wave_ops.Roundeven attribute) (wave_lang.kernel.ops.wave_ops.Rsqrt attribute) (wave_lang.kernel.ops.wave_ops.ScaledMMA attribute) (wave_lang.kernel.ops.wave_ops.ScatterAdd attribute) (wave_lang.kernel.ops.wave_ops.SchedulingBarrier attribute) (wave_lang.kernel.ops.wave_ops.SchedulingGroupBarrier attribute) (wave_lang.kernel.ops.wave_ops.SelectOp attribute) (wave_lang.kernel.ops.wave_ops.SelfIndex attribute) (wave_lang.kernel.ops.wave_ops.SetSymbol attribute) (wave_lang.kernel.ops.wave_ops.SetWavePrio attribute) (wave_lang.kernel.ops.wave_ops.SharedMemoryBarrier attribute) (wave_lang.kernel.ops.wave_ops.SharedMemoryBarrierSignal attribute) (wave_lang.kernel.ops.wave_ops.SharedMemoryBarrierWait attribute) (wave_lang.kernel.ops.wave_ops.ShuffleOp attribute) (wave_lang.kernel.ops.wave_ops.Sin attribute) (wave_lang.kernel.ops.wave_ops.Sinh attribute) (wave_lang.kernel.ops.wave_ops.Softsign attribute) (wave_lang.kernel.ops.wave_ops.Sqrt attribute) (wave_lang.kernel.ops.wave_ops.Sub attribute) (wave_lang.kernel.ops.wave_ops.Sum attribute) (wave_lang.kernel.ops.wave_ops.Tanh attribute) (wave_lang.kernel.ops.wave_ops.TanhApprox attribute) (wave_lang.kernel.ops.wave_ops.TensorCounterWait attribute) (wave_lang.kernel.ops.wave_ops.TensorLoadToLDS attribute) (wave_lang.kernel.ops.wave_ops.TopkOp attribute) (wave_lang.kernel.ops.wave_ops.Truediv attribute) (wave_lang.kernel.ops.wave_ops.WorkgroupBarrier attribute) (wave_lang.kernel.ops.wave_ops.Write attribute) topk() (in module wave_lang.kernel.ops.wave_ops) TopkOp (class in wave_lang.kernel.ops.wave_ops) torch_device (wave_lang.runtime.DeviceState attribute) torch_dtype_to_numpy() (in module wave_lang.support.conversions) torch_dtyped_shape_to_iree_format() (in module wave_lang.support.conversions) torch_stream (wave_lang.runtime.DeviceState attribute) trace (wave_lang.kernel.compiler.wave_codegen.WaveEmitter attribute) trace_tensor_callback() (in module wave_lang.support) trace_tensor_to_npy() (in module wave_lang.support) transform_index() (wave_lang.kernel.ops.wave_ops.CustomOp method) (wave_lang.kernel.ops.wave_ops.Permute method) transform_index_backwards() (wave_lang.kernel.ops.wave_ops.CustomOp method) (wave_lang.kernel.ops.wave_ops.Read method) (wave_lang.kernel.ops.wave_ops.Write method) transform_two_PP_clusters() built-in function Truediv (class in wave_lang.kernel.ops.wave_ops) type (wave_lang.kernel.ops.wave_ops.Allocate property) (wave_lang.kernel.ops.wave_ops.ApplyExpr property) (wave_lang.kernel.ops.wave_ops.CustomOp property) (wave_lang.kernel.ops.wave_ops.ExtractSlice property) (wave_lang.kernel.ops.wave_ops.SelfIndex property) (wave_lang.kernel.ops.wave_ops.SetSymbol property) type_cache_key (wave_lang.runtime.Device attribute) U UnaryPyOp (class in wave_lang.kernel.ops.wave_ops) Unknown (class in wave_lang.kernel.ops.wave_ops) unpadded_dims (wave_lang.kernel.ops.wave_ops.Allocate property) unpadded_shape (wave_lang.kernel.ops.wave_ops.Allocate property) unroll_iteration (wave_lang.kernel.ops.wave_ops.CustomOp property) update_aliased_workgroup_constraints() (wave_lang.kernel.wave.wave.LaunchableWave method) update_arg() (wave_lang.kernel.ops.wave_ops.CustomOp method) users (wave_lang.kernel.ops.wave_ops.CustomOp property) V value (wave_lang.kernel.ops.wave_ops.GetResult attribute) (wave_lang.kernel.ops.wave_ops.NewRegister attribute) (wave_lang.kernel.ops.wave_ops.NewScalar attribute) vector_shapes (wave_lang.kernel.ops.wave_ops.CustomOp property) vm_instance (wave_lang.runtime.Device property) VOLATILE (wave_lang.kernel.ops.wave_ops.MemoryAccessFlags attribute) W wait_async_ops (wave_lang.kernel.ops.wave_ops.SharedMemoryBarrier attribute) wave.tools module wave_constraints (wave_lang.kernel.wave.wave.LaunchableWave property) wave_lang.kernel.compiler.wave_codegen module wave_lang.kernel.ops.wave_ops module wave_lang.kernel.wave.wave module wave_lang.runtime module wave_lang.support module wave_lang.support.conversions module wave_lang.support.debugging module wave_lang.support.logging module wave_lang.support.tools module WaveEmitter (class in wave_lang.kernel.compiler.wave_codegen) width (wave_lang.kernel.ops.wave_ops.ShuffleOp attribute) workgroup_barrier() (in module wave_lang.kernel.ops.wave_ops) workgroup_constraints (wave_lang.kernel.wave.wave.LaunchableWave property) WorkgroupBarrier (class in wave_lang.kernel.ops.wave_ops) Write (class in wave_lang.kernel.ops.wave_ops) write() (in module wave_lang.kernel.ops.wave_ops) write_dependency (wave_lang.kernel.ops.wave_ops.Read property) Y yielded_values (wave_lang.kernel.ops.wave_ops.Output property)