scirs2-special 0.5.0

Special functions module for SciRS2 (scirs2-special)
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34

//! GPU kernel implementations for batch special-function evaluation.
//!
//! This module houses compute-shader sources and host-side dispatch stubs for
//! two GPU backends:
//!
//! | Sub-module | Backend | Feature flag | Default |
//! |------------|---------|--------------|---------|
//! | [`wgsl`]   | WebGPU (WGSL shaders) | `gpu` via `scirs2-core/gpu` | off |
//! | [`cuda`]   | CUDA / ROCm           | `cuda_kernels`              | off |
//!
//! Both backends currently provide stub implementations that return
//! "not available" errors.  The actual dispatch logic lives in
//! [`crate::gpu_dispatch`] which calls these stubs under
//! `DispatchTarget::Gpu` and falls back to the CPU rayon path automatically.
//!
//! # Pure Rust policy
//!
//! The `cuda_kernels` feature is off by default.  CUDA depends on C dynamic
//! libraries (`libcuda.so`, `libcudart.so`) which violate the Pure Rust
//! Policy unless feature-gated.  WebGPU is Pure Rust (via `wgpu`) but the
//! host-side integration with `scirs2-core/gpu` is not yet stabilised.

pub mod cuda;
pub mod wgsl;

pub use cuda::{
    bessel_j0_batch_cuda, erf_batch_cuda, gamma_batch_cuda, CudaDispatchError, BESSEL_J0_PTX_STUB,
    ERF_PTX_STUB, GAMMA_PTX_STUB,
};
pub use wgsl::{
    bessel_j0_batch_wgpu, erf_batch_wgpu, erfc_batch_wgpu, erfinv_batch_wgpu, gamma_batch_wgpu,
    lgamma_batch_wgpu, WgslDispatchError, BESSEL_J0_WGSL, ERFC_WGSL, ERFINV_WGSL, ERF_WGSL,
    GAMMA_WGSL, LGAMMA_WGSL,
};