brainevent.binary_csrmv_indexed_p

brainevent.binary_csrmv_indexed_p#

brainevent.binary_csrmv_indexed_p = <brainevent.XLACustomKernel object>#

Low-level XLA custom-kernel primitive for binary_csrmv_indexed.

Event-driven CSR sparse matrix-vector multiplication where heterogeneous weights are read through a permutation perm (structural slot j uses weights[perm[j]]). Homogeneous weights ignore perm. Registered backends: numba (CPU), jax_raw (CPU/GPU/TPU), and cuda_raw (CUDA warp kernels in binary_csrmv.cu / binary_indexed_csrmv.cu).