Skip to main content
Back to top
Ctrl
+
K
ejkernel documentation
Getting Started:
Installation
Architecture & Design:
ejKernel Project Overview
Kernel Registry System Analysis
Ops System Architecture Analysis
MaskInfo Guide: Comprehensive Attention Mask Management
Kernel Implementations Analysis
Module Operations Layer Analysis
Utilities and Helper Functions Analysis
Test Suite and Examples Analysis
ejKernel Comprehensive Architecture Report
User Guides:
ejkernel API Documentation
ejkernel.modules - High-Level Operations API
ejkernel.types - Masking and Type Utilities
ejkernel.ops - Kernel Framework and Configuration Management
API Reference:
ejkernel API Reference
ejkernel.callib package
ejkernel.callib._cute_call
ejkernel.callib._cute_ffi
ejkernel.callib._ejit
ejkernel.callib._pallas_call
ejkernel.callib._triton_call
ejkernel.callib._utils
ejkernel.kernels package
ejkernel.kernels._cuda package
ejkernel.kernels._cuda.blocksparse_attention package
ejkernel.kernels._cuda.flash_attention package
ejkernel.kernels._cuda.quantized_matmul package
ejkernel.kernels._cuda.ragged_page_attention_v3 package
ejkernel.kernels._cuda.unified_attention package
ejkernel.kernels._cute package
ejkernel.kernels._cute.chunked_prefill_paged_decode package
ejkernel.kernels._cute.flash_attention package
ejkernel.kernels._cute.quantized_matmul package
ejkernel.kernels._cute.ragged_page_attention_v3 package
ejkernel.kernels._cute.unified_attention package
ejkernel.kernels._pallas package
ejkernel.kernels._pallas.gpu package
ejkernel.kernels._pallas.tpu package
ejkernel.kernels._triton package
ejkernel.kernels._triton.blocksparse_attention package
ejkernel.kernels._triton.chunked_prefill_paged_decode package
ejkernel.kernels._triton.decode_attention package
ejkernel.kernels._triton.flash_attention package
ejkernel.kernels._triton.flash_mla package
ejkernel.kernels._triton.gla package
ejkernel.kernels._triton.lightning_attn package
ejkernel.kernels._triton.mean_pooling package
ejkernel.kernels._triton.native_sparse_attention package
ejkernel.kernels._triton.page_attention package
ejkernel.kernels._triton.quantized_matmul package
ejkernel.kernels._triton.ragged_decode_attention package
ejkernel.kernels._triton.ragged_page_attention_v2 package
ejkernel.kernels._triton.ragged_page_attention_v3 package
ejkernel.kernels._triton.recurrent package
ejkernel.kernels._triton.ring_attention package
ejkernel.kernels._triton.rwkv4 package
ejkernel.kernels._triton.rwkv6 package
ejkernel.kernels._triton.rwkv7 package
ejkernel.kernels._triton.unified_attention package
ejkernel.kernels._xla package
ejkernel.kernels._xla.attention package
ejkernel.kernels._xla.blocksparse_attention package
ejkernel.kernels._xla.chunked_prefill_paged_decode package
ejkernel.kernels._xla.decode_attention package
ejkernel.kernels._xla.flash_attention package
ejkernel.kernels._xla.flash_mla package
ejkernel.kernels._xla.gla package
ejkernel.kernels._xla.grouped_matmul package
ejkernel.kernels._xla.kernel_delta_attention package
ejkernel.kernels._xla.lightning_attn package
ejkernel.kernels._xla.mean_pooling package
ejkernel.kernels._xla.native_sparse_attention package
ejkernel.kernels._xla.page_attention package
ejkernel.kernels._xla.prefill_page_attention package
ejkernel.kernels._xla.quantized_matmul package
ejkernel.kernels._xla.ragged_decode_attention package
ejkernel.kernels._xla.ragged_page_attention_v2 package
ejkernel.kernels._xla.ragged_page_attention_v3 package
ejkernel.kernels._xla.recurrent package
ejkernel.kernels._xla.ring_attention package
ejkernel.kernels._xla.rwkv4 package
ejkernel.kernels._xla.rwkv6 package
ejkernel.kernels._xla.rwkv7 package
ejkernel.kernels._xla.scaled_dot_product_attention package
ejkernel.kernels._xla.state_space_v1 package
ejkernel.kernels._xla.state_space_v2 package
ejkernel.kernels._xla.unified_attention package
ejkernel.kernels._registry
ejkernel.modules package
ejkernel.modules.operations package
ejkernel.modules.operations.attention
ejkernel.modules.operations.blocksparse_attention
ejkernel.modules.operations.chunked_prefill_paged_decode
ejkernel.modules.operations.configs
ejkernel.modules.operations.decode_attention
ejkernel.modules.operations.flash_attention
ejkernel.modules.operations.gated_linear_attention
ejkernel.modules.operations.grouped_matmul
ejkernel.modules.operations.kernel_delta_attention
ejkernel.modules.operations.lightning_attention
ejkernel.modules.operations.multi_head_latent_attention
ejkernel.modules.operations.native_sparse_attention
ejkernel.modules.operations.page_attention
ejkernel.modules.operations.pooling
ejkernel.modules.operations.prefill_page_attention
ejkernel.modules.operations.quantized_matmul
ejkernel.modules.operations.ragged_decode_attention
ejkernel.modules.operations.ragged_page_attention_v2
ejkernel.modules.operations.ragged_page_attention_v3
ejkernel.modules.operations.recurrent
ejkernel.modules.operations.ring_attention
ejkernel.modules.operations.rwkv4
ejkernel.modules.operations.rwkv6
ejkernel.modules.operations.rwkv7
ejkernel.modules.operations.scaled_dot_product_attention
ejkernel.modules.operations.state_space_v1
ejkernel.modules.operations.state_space_v2
ejkernel.modules.operations.unified_attention
ejkernel.modules.base
ejkernel.ops package
ejkernel.ops.config package
ejkernel.ops.config.cache
ejkernel.ops.config.persistent
ejkernel.ops.config.selection
ejkernel.ops.core package
ejkernel.ops.core.kernel
ejkernel.ops.core.types
ejkernel.ops.execution package
ejkernel.ops.execution.batch
ejkernel.ops.execution.executor
ejkernel.ops.execution.offline
ejkernel.ops.execution.profiler
ejkernel.ops.execution.tuning
ejkernel.ops.utils package
ejkernel.ops.utils.datacarrier
ejkernel.ops.utils.fingerprint
ejkernel.ops.utils.meta
ejkernel.ops.utils.serialize
ejkernel.ops.registry
ejkernel.quantization package
ejkernel.quantization._quants package
ejkernel.quantization._quants.quantizations
ejkernel.quantization._utils package
ejkernel.quantization._utils.bitpack
ejkernel.quantization._utils.fp_tables
ejkernel.quantization._utils.grouping
ejkernel.quantization._utils.qparams
ejkernel.types package
ejkernel.types.mask
ejkernel.xla_utils package
ejkernel.xla_utils.cumsum
ejkernel.xla_utils.shardings
ejkernel.xla_utils.utils
ejkernel.benchmarks
ejkernel.build_cudalib
ejkernel.errors
ejkernel.loggings
ejkernel.utils
.rst
.pdf
ejkernel.kernels._pallas.tpu.flash_attention._pallas_impl_bwd
ejkernel.kernels._pallas.tpu.flash_attention._pallas_impl_bwd
#