List of all items

Structs

• backend::WgpuExecutable
• buffer::Arena
• buffer::ReadbackLayout
• buffer::ReadbackStaging
• buffer::TinyReadbackStaging
• calibrate::Calibration
• device::WgpuDevice
• fft_dispatch::FftGpuResources
• kernels::ArgmaxParams
• kernels::AttentionBwdParams
• kernels::AttentionParams
• kernels::BatchElementwiseRegionParams
• kernels::BinaryParams
• kernels::Conv1dParams
• kernels::Conv2dParams
• kernels::Conv3dParams
• kernels::CopyParams
• kernels::CumsumBwdParams
• kernels::CumsumParams
• kernels::DequantMatmulParams
• kernels::ElementwiseRegionParams
• kernels::ExpandParams
• kernels::FftGpuParams
• kernels::FftParams
• kernels::FusedResidualLnParams
• kernels::FusedResidualLnTeeParams
• kernels::GatherAxisParams
• kernels::GatherBwdParams
• kernels::GatherParams
• kernels::GroupedMatmulParams
• kernels::Kernel
• kernels::LayerNormBwdParams
• kernels::LayerNormParams
• kernels::MatmulParams
• kernels::MatmulQkvParams
• kernels::NarrowConcatParams
• kernels::Pool1dParams
• kernels::Pool2dParams
• kernels::Pool3dParams
• kernels::ReduceParams
• kernels::RmsNormBwdParams
• kernels::RopeBwdParams
• kernels::RopeParams
• kernels::SampleParams
• kernels::ScatterAddParams
• kernels::SelectiveScanParams
• kernels::SoftmaxParams
• kernels::TopKParams
• kernels::TransposeParams
• kernels::UmapKnnParams
• kernels::UnaryParams
• kernels::WelchPeaksGpuParams
• kernels::WhereParams

Functions

• buffer::decode_mapped_readback_f32
• buffer::decode_tiny_mapped_f32
• buffer::encode_readback_copies
• buffer::map_readback_f32
• buffer::plan_f32_uniform
• buffer::read_f32_many_pooled
• buffer::read_f32_pooled
• buffer::read_tiny_f32_after_submit
• buffer::schedule_readback_map
• buffer::use_tiny_readback
• buffer::wait_readback_map
• coop_f16_vk::refresh_wide_b_flag
• coop_f16_vk::use_wide_matmul
• coop_f16_vk::weight_oscillation_score
• device::adapter_name
• device::coop_discrete_backend
• device::coop_f16_16x16_f32_acc_supported
• device::coop_f16_16x16_supported
• device::coop_f32_8x8_supported
• device::is_vulkan_available
• device::select_vulkan_backend
• device::wgpu_device
• fft_dispatch::dispatch_fft_gpu_in_pass
• fft_dispatch::run_fft_gpu
• fft_dispatch::run_fft_gpu_cached
• fft_host::run_fft1d
• gdn_host::run_gated_delta_net
• gguf_host::gguf_scheme_id
• gguf_host::run_dequant_grouped_matmul_gguf
• gguf_host::run_dequant_matmul_gguf
• gguf_host::scheme_from_id
• im2col_host::run_im2col
• is_available
• kernels::argmax_kernel
• kernels::attention_bwd_kernel
• kernels::attention_kernel
• kernels::batch_elementwise_region_kernel
• kernels::binary_kernel
• kernels::cast_f32_to_f16_kernel
• kernels::compare_kernel
• kernels::concat_kernel
• kernels::conv1d_kernel
• kernels::conv2d_kernel
• kernels::conv3d_kernel
• kernels::coop_f16_vk_f32acc_available
• kernels::coop_f16_vk_widen_b_load
• kernels::copy_kernel
• kernels::cumsum_backward_kernel
• kernels::cumsum_kernel
• kernels::dequant_matmul_kernel
• kernels::elementwise_region_kernel
• kernels::elementwise_region_spatial_kernel
• kernels::expand_kernel
• kernels::fft_gpu_bit_reverse_kernel
• kernels::fft_gpu_inner_kernel
• kernels::fft_gpu_outer_r2_kernel
• kernels::fft_gpu_outer_r4_kernel
• kernels::fft_gpu_radix2_full_kernel
• kernels::fused_residual_ln_kernel
• kernels::fused_residual_ln_tee_kernel
• kernels::fused_residual_rms_norm_kernel
• kernels::gather_axis_kernel
• kernels::gather_backward_acc_kernel
• kernels::gather_backward_zero_kernel
• kernels::gather_kernel
• kernels::grouped_matmul_kernel
• kernels::layer_norm_backward_gamma_partial_kernel
• kernels::layer_norm_backward_gamma_reduce_kernel
• kernels::layer_norm_backward_input_kernel
• kernels::layernorm_kernel
• kernels::matmul_coop16_kernel
• kernels::matmul_coop_f16_vulkan_active_kernel
• kernels::matmul_coop_f16_vulkan_f32acc_kernel
• kernels::matmul_coop_f16_vulkan_kernel
• kernels::matmul_coop_f16_vulkan_widen_f32acc_kernel
• kernels::matmul_coop_f16_vulkan_widen_kernel
• kernels::matmul_coop_f32_active_kernel
• kernels::matmul_coop_f32_kernel
• kernels::matmul_coop_f32_portable_kernel
• kernels::matmul_f16_compute_kernel
• kernels::matmul_f16w_kernel
• kernels::matmul_kernel
• kernels::matmul_qkv_coop_f16_vk_active_kernel
• kernels::matmul_qkv_coop_f16_vk_f32acc_kernel
• kernels::matmul_qkv_coop_f16_vk_kernel
• kernels::matmul_qkv_coop_f16_vk_widen_f32acc_kernel
• kernels::matmul_qkv_coop_f16_vk_widen_kernel
• kernels::matmul_qkv_coop_f32_kernel
• kernels::matmul_qkv_kernel
• kernels::matmul_wide_active_kernel
• kernels::matmul_wide_kernel
• kernels::matmul_wide_nv_kernel
• kernels::narrow_kernel
• kernels::pool1d_kernel
• kernels::pool2d_kernel
• kernels::pool3d_kernel
• kernels::reduce_kernel
• kernels::rms_norm_backward_kernel
• kernels::rms_norm_backward_param_kernel
• kernels::rope_backward_kernel
• kernels::rope_kernel
• kernels::sample_kernel
• kernels::scatter_add_kernel
• kernels::selective_scan_kernel
• kernels::softmax_kernel
• kernels::topk_kernel
• kernels::transpose_kernel
• kernels::umap_knn_kernel
• kernels::unary_f16_mirror_kernel
• kernels::unary_kernel
• kernels::welch_peaks_gpu_kernel
• kernels::where_kernel
• llada2_gate_host::run_llada2_group_limited_gate
• log_mel_host::run_log_mel
• log_mel_host::run_log_mel_backward
• training_bwd_host::run_cumsum_backward
• training_bwd_host::run_gather_backward
• training_bwd_host::run_rms_norm_backward_beta
• training_bwd_host::run_rms_norm_backward_gamma
• training_bwd_host::run_rms_norm_backward_input
• training_bwd_host::run_rope_backward
• umap_knn_host::run_umap_knn
• unfuse::unfuse
• welch_peaks_dispatch::dispatch_welch_peaks_gpu_in_pass
• welch_peaks_dispatch::run_welch_peaks_gpu
• welch_peaks_dispatch::welch_peaks_gpu_params
• welch_peaks_host::run_welch_peaks

Type Aliases

• kernels::FusedResidualRmsNormParams

Constants

• kernels::ARGMAX_WGSL
• kernels::ATTENTION_BWD_WGSL
• kernels::ATTENTION_WGSL
• kernels::BINARY_WGSL
• kernels::CAST_F32_TO_F16_WGSL
• kernels::COMPARE_WGSL
• kernels::CONCAT_WGSL
• kernels::CONV1D_WGSL
• kernels::CONV2D_WGSL
• kernels::CONV3D_WGSL
• kernels::COOP_F16_VK_WIDEN_N
• kernels::COPY_WGSL
• kernels::CUMSUM_BWD_WGSL
• kernels::CUMSUM_WGSL
• kernels::DEQUANT_MATMUL_WGSL
• kernels::ELEMENTWISE_REGION_WGSL
• kernels::EXPAND_WGSL
• kernels::FFT_GPU_WGSL
• kernels::FUSED_RESIDUAL_LN_TEE_WGSL
• kernels::FUSED_RESIDUAL_LN_WGSL
• kernels::FUSED_RESIDUAL_RMS_NORM_WGSL
• kernels::GATHER_AXIS_WGSL
• kernels::GATHER_BWD_WGSL
• kernels::GATHER_WGSL
• kernels::GROUPED_MATMUL_WGSL
• kernels::LAYERNORM_WGSL
• kernels::LAYER_NORM_BWD_WGSL
• kernels::MATMUL_COOP16_WGSL
• kernels::MATMUL_COOP_F16_VULKAN_F32ACC_WGSL
• kernels::MATMUL_COOP_F16_VULKAN_WGSL
• kernels::MATMUL_COOP_F16_VULKAN_WIDEN_F32ACC_WGSL
• kernels::MATMUL_COOP_F16_VULKAN_WIDEN_WGSL
• kernels::MATMUL_COOP_F32_PORTABLE_WGSL
• kernels::MATMUL_COOP_F32_WGSL
• kernels::MATMUL_F16W_WGSL
• kernels::MATMUL_F16_COMPUTE_WGSL
• kernels::MATMUL_QKV_COOP_F16_VK_F32ACC_WGSL
• kernels::MATMUL_QKV_COOP_F16_VK_WGSL
• kernels::MATMUL_QKV_COOP_F16_VK_WIDEN_F32ACC_WGSL
• kernels::MATMUL_QKV_COOP_F16_VK_WIDEN_WGSL
• kernels::MATMUL_QKV_COOP_F32_WGSL
• kernels::MATMUL_QKV_WGSL
• kernels::MATMUL_WGSL
• kernels::MATMUL_WIDE_NV_WGSL
• kernels::MATMUL_WIDE_WGSL
• kernels::NARROW_WGSL
• kernels::POOL1D_WGSL
• kernels::POOL2D_WGSL
• kernels::POOL3D_WGSL
• kernels::REDUCE_WGSL
• kernels::RMS_NORM_BWD_WGSL
• kernels::ROPE_BWD_WGSL
• kernels::ROPE_WGSL
• kernels::SAMPLE_WGSL
• kernels::SCATTER_ADD_WGSL
• kernels::SELECTIVE_SCAN_WGSL
• kernels::SOFTMAX_WGSL
• kernels::TOPK_WGSL
• kernels::TRANSPOSE_WGSL
• kernels::UMAP_KNN_WGSL
• kernels::UNARY_F16_MIRROR_WGSL
• kernels::UNARY_WGSL
• kernels::WELCH_PEAKS_GPU_WGSL
• kernels::WHERE_WGSL
• umap_knn_host::UMAP_KNN_GPU_MIN_N