dispatch.h

/**
 *  @brief Common Definitions for Dispatch Files.
 *  @file c/dispatch.h
 *  @author Ash Vardanian
 *  @date February 3, 2026
 */
#ifndef NK_DISPATCH_H
#define NK_DISPATCH_H

#define NK_DYNAMIC_DISPATCH 1
#define NK_NATIVE_F16       0
#define NK_NATIVE_BF16      0

/*  NK_TARGET_* defines are set by the build system:
 *  - Python: setup.py
 *  - Rust: build.rs
 *  - Node.js: binding.gyp
 *  - CMake: CMakeLists.txt
 *
 *  For header-only usage without a build system, types.h provides
 *  compiler-intrinsic-based fallback detection.
 *
 *  OS/compiler capabilities summary:
 *  - Linux: everything available in GCC 12+ and Clang 16+.
 *  - FreeBSD: same as Linux, except AMX (no kernel tile permission support).
 *  - Windows - MSVC: Haswell/Skylake/Icelake, plus Sapphire FP16 (MSVC 2022 17.2+).
 *  - macOS - Apple Clang: only Arm NEON and x86 AVX2 Haswell extensions.
 */

#include <numkong/numkong.h>

#ifdef __cplusplus
extern "C" {
#endif

// Forward declaration of dispatch table type (same structure as in numkong.c)
typedef struct {
    // Dot products
    nk_metric_dense_punned_t dot_f64c;
    nk_metric_dense_punned_t dot_f32c;
    nk_metric_dense_punned_t dot_bf16c;
    nk_metric_dense_punned_t dot_f16c;
    nk_metric_dense_punned_t dot_f64;
    nk_metric_dense_punned_t dot_f32;
    nk_metric_dense_punned_t dot_bf16;
    nk_metric_dense_punned_t dot_f16;
    nk_metric_dense_punned_t dot_e5m2;
    nk_metric_dense_punned_t dot_e4m3;
    nk_metric_dense_punned_t dot_e3m2;
    nk_metric_dense_punned_t dot_e2m3;
    nk_metric_dense_punned_t dot_i8;
    nk_metric_dense_punned_t dot_u8;
    nk_metric_dense_punned_t dot_i4;
    nk_metric_dense_punned_t dot_u4;
    nk_metric_dense_punned_t dot_u1;
    nk_metric_dense_punned_t vdot_f64c;
    nk_metric_dense_punned_t vdot_f32c;
    nk_metric_dense_punned_t vdot_bf16c;
    nk_metric_dense_punned_t vdot_f16c;
    // Angular distances
    nk_metric_dense_punned_t angular_f64;
    nk_metric_dense_punned_t angular_f32;
    nk_metric_dense_punned_t angular_bf16;
    nk_metric_dense_punned_t angular_f16;
    nk_metric_dense_punned_t angular_e5m2;
    nk_metric_dense_punned_t angular_e4m3;
    nk_metric_dense_punned_t angular_e3m2;
    nk_metric_dense_punned_t angular_e2m3;
    nk_metric_dense_punned_t angular_i8;
    nk_metric_dense_punned_t angular_i4;
    nk_metric_dense_punned_t angular_u8;
    nk_metric_dense_punned_t angular_u4;
    // Euclidean distances
    nk_metric_dense_punned_t euclidean_f64;
    nk_metric_dense_punned_t euclidean_f32;
    nk_metric_dense_punned_t euclidean_bf16;
    nk_metric_dense_punned_t euclidean_f16;
    nk_metric_dense_punned_t euclidean_e5m2;
    nk_metric_dense_punned_t euclidean_e4m3;
    nk_metric_dense_punned_t euclidean_e3m2;
    nk_metric_dense_punned_t euclidean_e2m3;
    nk_metric_dense_punned_t euclidean_i8;
    nk_metric_dense_punned_t euclidean_i4;
    nk_metric_dense_punned_t euclidean_u8;
    nk_metric_dense_punned_t euclidean_u4;
    // Squared Euclidean distances
    nk_metric_dense_punned_t sqeuclidean_f64;
    nk_metric_dense_punned_t sqeuclidean_f32;
    nk_metric_dense_punned_t sqeuclidean_bf16;
    nk_metric_dense_punned_t sqeuclidean_f16;
    nk_metric_dense_punned_t sqeuclidean_e5m2;
    nk_metric_dense_punned_t sqeuclidean_e4m3;
    nk_metric_dense_punned_t sqeuclidean_e3m2;
    nk_metric_dense_punned_t sqeuclidean_e2m3;
    nk_metric_dense_punned_t sqeuclidean_i8;
    nk_metric_dense_punned_t sqeuclidean_i4;
    nk_metric_dense_punned_t sqeuclidean_u8;
    nk_metric_dense_punned_t sqeuclidean_u4;
    // Binary distances
    nk_metric_dense_punned_t hamming_u8;
    nk_metric_dense_punned_t hamming_u1;
    nk_metric_dense_punned_t jaccard_u32;
    nk_metric_dense_punned_t jaccard_u16;
    nk_metric_dense_punned_t jaccard_u1;
    // Curved spaces
    nk_metric_curved_punned_t bilinear_f64c;
    nk_metric_curved_punned_t bilinear_f32c;
    nk_metric_curved_punned_t bilinear_bf16c;
    nk_metric_curved_punned_t bilinear_f16c;
    nk_metric_curved_punned_t bilinear_f64;
    nk_metric_curved_punned_t bilinear_f32;
    nk_metric_curved_punned_t bilinear_bf16;
    nk_metric_curved_punned_t bilinear_f16;
    nk_metric_curved_punned_t mahalanobis_f64;
    nk_metric_curved_punned_t mahalanobis_f32;
    nk_metric_curved_punned_t mahalanobis_bf16;
    nk_metric_curved_punned_t mahalanobis_f16;
    // Geospatial distances
    nk_metric_geospatial_punned_t haversine_f64;
    nk_metric_geospatial_punned_t haversine_f32;
    nk_metric_geospatial_punned_t vincenty_f64;
    nk_metric_geospatial_punned_t vincenty_f32;
    // Probability distributions
    nk_metric_dense_punned_t kld_f64;
    nk_metric_dense_punned_t kld_f32;
    nk_metric_dense_punned_t kld_bf16;
    nk_metric_dense_punned_t kld_f16;
    nk_metric_dense_punned_t jsd_f64;
    nk_metric_dense_punned_t jsd_f32;
    nk_metric_dense_punned_t jsd_bf16;
    nk_metric_dense_punned_t jsd_f16;
    // Mesh alignment
    nk_metric_mesh_punned_t rmsd_f64;
    nk_metric_mesh_punned_t rmsd_f32;
    nk_metric_mesh_punned_t rmsd_bf16;
    nk_metric_mesh_punned_t rmsd_f16;
    nk_metric_mesh_punned_t kabsch_f64;
    nk_metric_mesh_punned_t kabsch_f32;
    nk_metric_mesh_punned_t kabsch_bf16;
    nk_metric_mesh_punned_t kabsch_f16;
    nk_metric_mesh_punned_t umeyama_f64;
    nk_metric_mesh_punned_t umeyama_f32;
    nk_metric_mesh_punned_t umeyama_bf16;
    nk_metric_mesh_punned_t umeyama_f16;
    // Sparse intersections
    nk_sparse_intersect_punned_t sparse_intersect_u64;
    nk_sparse_intersect_punned_t sparse_intersect_u32;
    nk_sparse_intersect_punned_t sparse_intersect_u16;
    // Sparse dot products
    nk_sparse_dot_punned_t sparse_dot_u32f32;
    nk_sparse_dot_punned_t sparse_dot_u16bf16;
    // Element-wise scale
    nk_each_scale_punned_t each_scale_f64c;
    nk_each_scale_punned_t each_scale_f32c;
    nk_each_scale_punned_t each_scale_f64;
    nk_each_scale_punned_t each_scale_f32;
    nk_each_scale_punned_t each_scale_bf16;
    nk_each_scale_punned_t each_scale_f16;
    nk_each_scale_punned_t each_scale_e5m2;
    nk_each_scale_punned_t each_scale_e4m3;
    nk_each_scale_punned_t each_scale_e3m2;
    nk_each_scale_punned_t each_scale_e2m3;
    nk_each_scale_punned_t each_scale_i64;
    nk_each_scale_punned_t each_scale_i32;
    nk_each_scale_punned_t each_scale_i16;
    nk_each_scale_punned_t each_scale_i8;
    nk_each_scale_punned_t each_scale_u64;
    nk_each_scale_punned_t each_scale_u32;
    nk_each_scale_punned_t each_scale_u16;
    nk_each_scale_punned_t each_scale_u8;
    // Element-wise sum
    nk_each_sum_punned_t each_sum_f64c;
    nk_each_sum_punned_t each_sum_f32c;
    nk_each_sum_punned_t each_sum_f64;
    nk_each_sum_punned_t each_sum_f32;
    nk_each_sum_punned_t each_sum_bf16;
    nk_each_sum_punned_t each_sum_f16;
    nk_each_sum_punned_t each_sum_e5m2;
    nk_each_sum_punned_t each_sum_e4m3;
    nk_each_sum_punned_t each_sum_e3m2;
    nk_each_sum_punned_t each_sum_e2m3;
    nk_each_sum_punned_t each_sum_i64;
    nk_each_sum_punned_t each_sum_i32;
    nk_each_sum_punned_t each_sum_i16;
    nk_each_sum_punned_t each_sum_i8;
    nk_each_sum_punned_t each_sum_u64;
    nk_each_sum_punned_t each_sum_u32;
    nk_each_sum_punned_t each_sum_u16;
    nk_each_sum_punned_t each_sum_u8;
    // Element-wise blend
    nk_each_blend_punned_t each_blend_f64c;
    nk_each_blend_punned_t each_blend_f32c;
    nk_each_blend_punned_t each_blend_f64;
    nk_each_blend_punned_t each_blend_f32;
    nk_each_blend_punned_t each_blend_bf16;
    nk_each_blend_punned_t each_blend_f16;
    nk_each_blend_punned_t each_blend_e5m2;
    nk_each_blend_punned_t each_blend_e4m3;
    nk_each_blend_punned_t each_blend_e3m2;
    nk_each_blend_punned_t each_blend_e2m3;
    nk_each_blend_punned_t each_blend_i64;
    nk_each_blend_punned_t each_blend_i32;
    nk_each_blend_punned_t each_blend_i16;
    nk_each_blend_punned_t each_blend_i8;
    nk_each_blend_punned_t each_blend_u64;
    nk_each_blend_punned_t each_blend_u32;
    nk_each_blend_punned_t each_blend_u16;
    nk_each_blend_punned_t each_blend_u8;
    // Element-wise FMA
    nk_each_fma_punned_t each_fma_f64c;
    nk_each_fma_punned_t each_fma_f32c;
    nk_each_fma_punned_t each_fma_f64;
    nk_each_fma_punned_t each_fma_f32;
    nk_each_fma_punned_t each_fma_bf16;
    nk_each_fma_punned_t each_fma_f16;
    nk_each_fma_punned_t each_fma_e5m2;
    nk_each_fma_punned_t each_fma_e4m3;
    nk_each_fma_punned_t each_fma_e3m2;
    nk_each_fma_punned_t each_fma_e2m3;
    nk_each_fma_punned_t each_fma_i64;
    nk_each_fma_punned_t each_fma_i32;
    nk_each_fma_punned_t each_fma_i16;
    nk_each_fma_punned_t each_fma_i8;
    nk_each_fma_punned_t each_fma_u64;
    nk_each_fma_punned_t each_fma_u32;
    nk_each_fma_punned_t each_fma_u16;
    nk_each_fma_punned_t each_fma_u8;
    // Trigonometry
    nk_kernel_trigonometry_punned_t each_sin_f64;
    nk_kernel_trigonometry_punned_t each_sin_f32;
    nk_kernel_trigonometry_punned_t each_sin_f16;
    nk_kernel_trigonometry_punned_t each_cos_f64;
    nk_kernel_trigonometry_punned_t each_cos_f32;
    nk_kernel_trigonometry_punned_t each_cos_f16;
    nk_kernel_trigonometry_punned_t each_atan_f64;
    nk_kernel_trigonometry_punned_t each_atan_f32;
    nk_kernel_trigonometry_punned_t each_atan_f16;
    // Reduce moments (sum + sum-of-squares)
    nk_kernel_reduce_moments_punned_t reduce_moments_f64;
    nk_kernel_reduce_moments_punned_t reduce_moments_f32;
    nk_kernel_reduce_moments_punned_t reduce_moments_bf16;
    nk_kernel_reduce_moments_punned_t reduce_moments_f16;
    nk_kernel_reduce_moments_punned_t reduce_moments_e5m2;
    nk_kernel_reduce_moments_punned_t reduce_moments_e4m3;
    nk_kernel_reduce_moments_punned_t reduce_moments_e3m2;
    nk_kernel_reduce_moments_punned_t reduce_moments_e2m3;
    nk_kernel_reduce_moments_punned_t reduce_moments_i64;
    nk_kernel_reduce_moments_punned_t reduce_moments_i32;
    nk_kernel_reduce_moments_punned_t reduce_moments_i16;
    nk_kernel_reduce_moments_punned_t reduce_moments_i8;
    nk_kernel_reduce_moments_punned_t reduce_moments_i4;
    nk_kernel_reduce_moments_punned_t reduce_moments_u64;
    nk_kernel_reduce_moments_punned_t reduce_moments_u32;
    nk_kernel_reduce_moments_punned_t reduce_moments_u16;
    nk_kernel_reduce_moments_punned_t reduce_moments_u8;
    nk_kernel_reduce_moments_punned_t reduce_moments_u4;
    nk_kernel_reduce_moments_punned_t reduce_moments_u1;
    // Reduce minmax (min + argmin + max + argmax)
    nk_kernel_reduce_minmax_punned_t reduce_minmax_f64;
    nk_kernel_reduce_minmax_punned_t reduce_minmax_f32;
    nk_kernel_reduce_minmax_punned_t reduce_minmax_bf16;
    nk_kernel_reduce_minmax_punned_t reduce_minmax_f16;
    nk_kernel_reduce_minmax_punned_t reduce_minmax_e5m2;
    nk_kernel_reduce_minmax_punned_t reduce_minmax_e4m3;
    nk_kernel_reduce_minmax_punned_t reduce_minmax_e3m2;
    nk_kernel_reduce_minmax_punned_t reduce_minmax_e2m3;
    nk_kernel_reduce_minmax_punned_t reduce_minmax_i64;
    nk_kernel_reduce_minmax_punned_t reduce_minmax_i32;
    nk_kernel_reduce_minmax_punned_t reduce_minmax_i16;
    nk_kernel_reduce_minmax_punned_t reduce_minmax_i8;
    nk_kernel_reduce_minmax_punned_t reduce_minmax_i4;
    nk_kernel_reduce_minmax_punned_t reduce_minmax_u64;
    nk_kernel_reduce_minmax_punned_t reduce_minmax_u32;
    nk_kernel_reduce_minmax_punned_t reduce_minmax_u16;
    nk_kernel_reduce_minmax_punned_t reduce_minmax_u8;
    nk_kernel_reduce_minmax_punned_t reduce_minmax_u4;
    nk_kernel_reduce_minmax_punned_t reduce_minmax_u1;
    // Dots packed size
    nk_dots_packed_size_punned_t dots_packed_size_f64;
    nk_dots_packed_size_punned_t dots_packed_size_f32;
    nk_dots_packed_size_punned_t dots_packed_size_bf16;
    nk_dots_packed_size_punned_t dots_packed_size_f16;
    nk_dots_packed_size_punned_t dots_packed_size_e5m2;
    nk_dots_packed_size_punned_t dots_packed_size_e4m3;
    nk_dots_packed_size_punned_t dots_packed_size_e3m2;
    nk_dots_packed_size_punned_t dots_packed_size_e2m3;
    nk_dots_packed_size_punned_t dots_packed_size_i8;
    nk_dots_packed_size_punned_t dots_packed_size_i4;
    nk_dots_packed_size_punned_t dots_packed_size_u8;
    nk_dots_packed_size_punned_t dots_packed_size_u4;
    nk_dots_packed_size_punned_t dots_packed_size_u1;
    // Dots pack
    nk_dots_pack_punned_t dots_pack_f64;
    nk_dots_pack_punned_t dots_pack_f32;
    nk_dots_pack_punned_t dots_pack_bf16;
    nk_dots_pack_punned_t dots_pack_f16;
    nk_dots_pack_punned_t dots_pack_e5m2;
    nk_dots_pack_punned_t dots_pack_e4m3;
    nk_dots_pack_punned_t dots_pack_e3m2;
    nk_dots_pack_punned_t dots_pack_e2m3;
    nk_dots_pack_punned_t dots_pack_i8;
    nk_dots_pack_punned_t dots_pack_i4;
    nk_dots_pack_punned_t dots_pack_u8;
    nk_dots_pack_punned_t dots_pack_u4;
    nk_dots_pack_punned_t dots_pack_u1;
    // Dots packed
    nk_dots_packed_punned_t dots_packed_f64;
    nk_dots_packed_punned_t dots_packed_f32;
    nk_dots_packed_punned_t dots_packed_bf16;
    nk_dots_packed_punned_t dots_packed_f16;
    nk_dots_packed_punned_t dots_packed_e5m2;
    nk_dots_packed_punned_t dots_packed_e4m3;
    nk_dots_packed_punned_t dots_packed_e3m2;
    nk_dots_packed_punned_t dots_packed_e2m3;
    nk_dots_packed_punned_t dots_packed_i8;
    nk_dots_packed_punned_t dots_packed_i4;
    nk_dots_packed_punned_t dots_packed_u8;
    nk_dots_packed_punned_t dots_packed_u4;
    nk_dots_packed_punned_t dots_packed_u1;
    // Sets packed
    nk_hammings_packed_punned_t hammings_packed_u1;
    nk_jaccards_packed_punned_t jaccards_packed_u1;
    // Dots symmetric
    nk_dots_symmetric_punned_t dots_symmetric_f64;
    nk_dots_symmetric_punned_t dots_symmetric_f32;
    nk_dots_symmetric_punned_t dots_symmetric_bf16;
    nk_dots_symmetric_punned_t dots_symmetric_f16;
    nk_dots_symmetric_punned_t dots_symmetric_e5m2;
    nk_dots_symmetric_punned_t dots_symmetric_e4m3;
    nk_dots_symmetric_punned_t dots_symmetric_e3m2;
    nk_dots_symmetric_punned_t dots_symmetric_e2m3;
    nk_dots_symmetric_punned_t dots_symmetric_i8;
    nk_dots_symmetric_punned_t dots_symmetric_i4;
    nk_dots_symmetric_punned_t dots_symmetric_u8;
    nk_dots_symmetric_punned_t dots_symmetric_u4;
    nk_dots_symmetric_punned_t dots_symmetric_u1;
    // Sets symmetric
    nk_hammings_symmetric_punned_t hammings_symmetric_u1;
    nk_jaccards_symmetric_punned_t jaccards_symmetric_u1;
    // Angulars packed
    nk_angulars_packed_punned_t angulars_packed_f64;
    nk_angulars_packed_punned_t angulars_packed_f32;
    nk_angulars_packed_punned_t angulars_packed_bf16;
    nk_angulars_packed_punned_t angulars_packed_f16;
    nk_angulars_packed_punned_t angulars_packed_e5m2;
    nk_angulars_packed_punned_t angulars_packed_e4m3;
    nk_angulars_packed_punned_t angulars_packed_e3m2;
    nk_angulars_packed_punned_t angulars_packed_e2m3;
    nk_angulars_packed_punned_t angulars_packed_i8;
    nk_angulars_packed_punned_t angulars_packed_i4;
    nk_angulars_packed_punned_t angulars_packed_u8;
    nk_angulars_packed_punned_t angulars_packed_u4;
    // Angulars symmetric
    nk_angulars_symmetric_punned_t angulars_symmetric_f64;
    nk_angulars_symmetric_punned_t angulars_symmetric_f32;
    nk_angulars_symmetric_punned_t angulars_symmetric_bf16;
    nk_angulars_symmetric_punned_t angulars_symmetric_f16;
    nk_angulars_symmetric_punned_t angulars_symmetric_e5m2;
    nk_angulars_symmetric_punned_t angulars_symmetric_e4m3;
    nk_angulars_symmetric_punned_t angulars_symmetric_e3m2;
    nk_angulars_symmetric_punned_t angulars_symmetric_e2m3;
    nk_angulars_symmetric_punned_t angulars_symmetric_i8;
    nk_angulars_symmetric_punned_t angulars_symmetric_i4;
    nk_angulars_symmetric_punned_t angulars_symmetric_u8;
    nk_angulars_symmetric_punned_t angulars_symmetric_u4;
    // Euclideans packed
    nk_euclideans_packed_punned_t euclideans_packed_f64;
    nk_euclideans_packed_punned_t euclideans_packed_f32;
    nk_euclideans_packed_punned_t euclideans_packed_bf16;
    nk_euclideans_packed_punned_t euclideans_packed_f16;
    nk_euclideans_packed_punned_t euclideans_packed_e5m2;
    nk_euclideans_packed_punned_t euclideans_packed_e4m3;
    nk_euclideans_packed_punned_t euclideans_packed_e3m2;
    nk_euclideans_packed_punned_t euclideans_packed_e2m3;
    nk_euclideans_packed_punned_t euclideans_packed_i8;
    nk_euclideans_packed_punned_t euclideans_packed_i4;
    nk_euclideans_packed_punned_t euclideans_packed_u8;
    nk_euclideans_packed_punned_t euclideans_packed_u4;
    // Euclideans symmetric
    nk_euclideans_symmetric_punned_t euclideans_symmetric_f64;
    nk_euclideans_symmetric_punned_t euclideans_symmetric_f32;
    nk_euclideans_symmetric_punned_t euclideans_symmetric_bf16;
    nk_euclideans_symmetric_punned_t euclideans_symmetric_f16;
    nk_euclideans_symmetric_punned_t euclideans_symmetric_e5m2;
    nk_euclideans_symmetric_punned_t euclideans_symmetric_e4m3;
    nk_euclideans_symmetric_punned_t euclideans_symmetric_e3m2;
    nk_euclideans_symmetric_punned_t euclideans_symmetric_e2m3;
    nk_euclideans_symmetric_punned_t euclideans_symmetric_i8;
    nk_euclideans_symmetric_punned_t euclideans_symmetric_i4;
    nk_euclideans_symmetric_punned_t euclideans_symmetric_u8;
    nk_euclideans_symmetric_punned_t euclideans_symmetric_u4;
    // MaxSim packed size
    nk_dots_packed_size_punned_t maxsim_packed_size_f32;
    nk_dots_packed_size_punned_t maxsim_packed_size_bf16;
    nk_dots_packed_size_punned_t maxsim_packed_size_f16;
    // MaxSim pack
    nk_dots_pack_punned_t maxsim_pack_f32;
    nk_dots_pack_punned_t maxsim_pack_bf16;
    nk_dots_pack_punned_t maxsim_pack_f16;
    // MaxSim packed
    nk_maxsim_packed_punned_t maxsim_packed_f32;
    nk_maxsim_packed_punned_t maxsim_packed_bf16;
    nk_maxsim_packed_punned_t maxsim_packed_f16;
    // Type casting
    nk_kernel_cast_punned_t cast;
    // Scalar conversions
    void (*bf16_to_f32)(nk_bf16_t const *, nk_f32_t *);
    void (*f32_to_bf16)(nk_f32_t const *, nk_bf16_t *);
    void (*f16_to_f32)(nk_f16_t const *, nk_f32_t *);
    void (*f32_to_f16)(nk_f32_t const *, nk_f16_t *);
    void (*e5m2_to_f32)(nk_e5m2_t const *, nk_f32_t *);
    void (*f32_to_e5m2)(nk_f32_t const *, nk_e5m2_t *);
    void (*e4m3_to_f32)(nk_e4m3_t const *, nk_f32_t *);
    void (*f32_to_e4m3)(nk_f32_t const *, nk_e4m3_t *);
    void (*e3m2_to_f32)(nk_e3m2_t const *, nk_f32_t *);
    void (*f32_to_e3m2)(nk_f32_t const *, nk_e3m2_t *);
    void (*e2m3_to_f32)(nk_e2m3_t const *, nk_f32_t *);
    void (*f32_to_e2m3)(nk_f32_t const *, nk_e2m3_t *);
    // Scalar math
    nk_f64_t (*f64_sqrt)(nk_f64_t);
    nk_f64_t (*f64_rsqrt)(nk_f64_t);
    nk_f64_t (*f64_fma)(nk_f64_t, nk_f64_t, nk_f64_t);
    nk_f32_t (*f32_sqrt)(nk_f32_t);
    nk_f32_t (*f32_rsqrt)(nk_f32_t);
    nk_f32_t (*f32_fma)(nk_f32_t, nk_f32_t, nk_f32_t);
    nk_f16_t (*f16_sqrt)(nk_f16_t);
    nk_f16_t (*f16_rsqrt)(nk_f16_t);
    nk_f16_t (*f16_fma)(nk_f16_t, nk_f16_t, nk_f16_t);
    // Scalar saturating arithmetic
    nk_i64_t (*i64_saturating_add)(nk_i64_t, nk_i64_t);
    nk_i64_t (*i64_saturating_mul)(nk_i64_t, nk_i64_t);
    nk_i32_t (*i32_saturating_add)(nk_i32_t, nk_i32_t);
    nk_i32_t (*i32_saturating_mul)(nk_i32_t, nk_i32_t);
    nk_i16_t (*i16_saturating_add)(nk_i16_t, nk_i16_t);
    nk_i16_t (*i16_saturating_mul)(nk_i16_t, nk_i16_t);
    nk_i8_t (*i8_saturating_add)(nk_i8_t, nk_i8_t);
    nk_i8_t (*i8_saturating_mul)(nk_i8_t, nk_i8_t);
    nk_i4x2_t (*i4x2_saturating_add)(nk_i4x2_t, nk_i4x2_t);
    nk_i4x2_t (*i4x2_saturating_mul)(nk_i4x2_t, nk_i4x2_t);
    nk_u64_t (*u64_saturating_add)(nk_u64_t, nk_u64_t);
    nk_u64_t (*u64_saturating_mul)(nk_u64_t, nk_u64_t);
    nk_u32_t (*u32_saturating_add)(nk_u32_t, nk_u32_t);
    nk_u32_t (*u32_saturating_mul)(nk_u32_t, nk_u32_t);
    nk_u16_t (*u16_saturating_add)(nk_u16_t, nk_u16_t);
    nk_u16_t (*u16_saturating_mul)(nk_u16_t, nk_u16_t);
    nk_u8_t (*u8_saturating_add)(nk_u8_t, nk_u8_t);
    nk_u8_t (*u8_saturating_mul)(nk_u8_t, nk_u8_t);
    nk_u4x2_t (*u4x2_saturating_add)(nk_u4x2_t, nk_u4x2_t);
    nk_u4x2_t (*u4x2_saturating_mul)(nk_u4x2_t, nk_u4x2_t);
    // Scalar ordering
    int (*bf16_order)(nk_bf16_t, nk_bf16_t);
    int (*f16_order)(nk_f16_t, nk_f16_t);
    int (*e5m2_order)(nk_e5m2_t, nk_e5m2_t);
    int (*e4m3_order)(nk_e4m3_t, nk_e4m3_t);
    int (*e3m2_order)(nk_e3m2_t, nk_e3m2_t);
    int (*e2m3_order)(nk_e2m3_t, nk_e2m3_t);
} nk_implementations_t;

// Global dispatch table - defined in numkong.c
extern nk_implementations_t nk_dispatch_table;

// Error handlers - defined in numkong.c
extern void nk_error_dense_(void const *, void const *, nk_size_t, void *);
extern void nk_error_sparse_intersect_(void const *, void const *, nk_size_t, nk_size_t, void *, nk_size_t *);
extern void nk_error_sparse_dot_(void const *, void const *, void const *, void const *, nk_size_t, nk_size_t, void *);
extern void nk_error_curved_(void const *, void const *, void const *, nk_size_t, void *);
extern void nk_error_geospatial_(void const *, void const *, void const *, void const *, nk_size_t, void *);
extern void nk_error_each_fma_(void const *, void const *, void const *, nk_size_t, void const *, void const *, void *);
extern void nk_error_each_blend_(void const *, void const *, nk_size_t, void const *, void const *, void *);
extern void nk_error_each_scale_(void const *, nk_size_t, void const *, void const *, void *);
extern void nk_error_each_sum_(void const *, void const *, nk_size_t, void *);
extern void nk_error_trigonometry_(void const *, nk_size_t, void *);
extern void nk_error_mesh_(void const *, void const *, nk_size_t, void *, void *, void *, void *, void *);
extern void nk_error_reduce_moments_(void const *, nk_size_t, nk_size_t, void *, void *);
extern void nk_error_reduce_minmax_(void const *, nk_size_t, nk_size_t, void *, nk_size_t *, void *, nk_size_t *);
extern nk_size_t nk_error_packed_size_(nk_size_t, nk_size_t);
extern void nk_error_pack_(void const *, nk_size_t, nk_size_t, nk_size_t, void *);
extern void nk_error_dots_(void const *, void const *, void *, nk_size_t, nk_size_t, nk_size_t, nk_size_t, nk_size_t);
extern void nk_error_dots_symmetric_(void const *, nk_size_t, nk_size_t, nk_size_t, void *, nk_size_t, nk_size_t,
                                     nk_size_t);

// Dtype-specific kernel lookup functions
extern void nk_dispatch_f64c_find_(nk_capability_t, nk_kernel_kind_t, nk_kernel_punned_t *, nk_capability_t *);
extern void nk_dispatch_f32c_find_(nk_capability_t, nk_kernel_kind_t, nk_kernel_punned_t *, nk_capability_t *);
extern void nk_dispatch_bf16c_find_(nk_capability_t, nk_kernel_kind_t, nk_kernel_punned_t *, nk_capability_t *);
extern void nk_dispatch_f16c_find_(nk_capability_t, nk_kernel_kind_t, nk_kernel_punned_t *, nk_capability_t *);
extern void nk_dispatch_f64_find_(nk_capability_t, nk_kernel_kind_t, nk_kernel_punned_t *, nk_capability_t *);
extern void nk_dispatch_f32_find_(nk_capability_t, nk_kernel_kind_t, nk_kernel_punned_t *, nk_capability_t *);
extern void nk_dispatch_bf16_find_(nk_capability_t, nk_kernel_kind_t, nk_kernel_punned_t *, nk_capability_t *);
extern void nk_dispatch_f16_find_(nk_capability_t, nk_kernel_kind_t, nk_kernel_punned_t *, nk_capability_t *);
extern void nk_dispatch_e5m2_find_(nk_capability_t, nk_kernel_kind_t, nk_kernel_punned_t *, nk_capability_t *);
extern void nk_dispatch_e4m3_find_(nk_capability_t, nk_kernel_kind_t, nk_kernel_punned_t *, nk_capability_t *);
extern void nk_dispatch_e3m2_find_(nk_capability_t, nk_kernel_kind_t, nk_kernel_punned_t *, nk_capability_t *);
extern void nk_dispatch_e2m3_find_(nk_capability_t, nk_kernel_kind_t, nk_kernel_punned_t *, nk_capability_t *);
extern void nk_dispatch_i64_find_(nk_capability_t, nk_kernel_kind_t, nk_kernel_punned_t *, nk_capability_t *);
extern void nk_dispatch_i32_find_(nk_capability_t, nk_kernel_kind_t, nk_kernel_punned_t *, nk_capability_t *);
extern void nk_dispatch_i16_find_(nk_capability_t, nk_kernel_kind_t, nk_kernel_punned_t *, nk_capability_t *);
extern void nk_dispatch_i8_find_(nk_capability_t, nk_kernel_kind_t, nk_kernel_punned_t *, nk_capability_t *);
extern void nk_dispatch_i4_find_(nk_capability_t, nk_kernel_kind_t, nk_kernel_punned_t *, nk_capability_t *);
extern void nk_dispatch_u64_find_(nk_capability_t, nk_kernel_kind_t, nk_kernel_punned_t *, nk_capability_t *);
extern void nk_dispatch_u32_find_(nk_capability_t, nk_kernel_kind_t, nk_kernel_punned_t *, nk_capability_t *);
extern void nk_dispatch_u16_find_(nk_capability_t, nk_kernel_kind_t, nk_kernel_punned_t *, nk_capability_t *);
extern void nk_dispatch_u8_find_(nk_capability_t, nk_kernel_kind_t, nk_kernel_punned_t *, nk_capability_t *);
extern void nk_dispatch_u4_find_(nk_capability_t, nk_kernel_kind_t, nk_kernel_punned_t *, nk_capability_t *);
extern void nk_dispatch_u1_find_(nk_capability_t, nk_kernel_kind_t, nk_kernel_punned_t *, nk_capability_t *);
extern void nk_dispatch_cast_find_(nk_capability_t, nk_kernel_kind_t, nk_kernel_punned_t *, nk_capability_t *);

#ifdef __cplusplus
}
#endif

#endif // NK_DISPATCH_H