Struct JitFamilySpec 

pub struct JitFamilySpec {
    pub spec_id: u64,
    pub body: String,
}

Stage 6 custom-family JIT specification.

Two levels per the charter:

• Level A (JitFamilySpec::glm): provide a (family, link) enum value plus an optional shape constant. The generator emits the matching built-in row body, identical to the cached built-in kernel — useful for end-to-end JIT path validation against the built-in cache.
• Level B (JitFamilySpec::raw): provide raw CUDA source for the row body. The body must define the same per-row locals the kernel shell expects: mu, grad_eta, w_fisher, w_hessian, w_solver, z_f, z_h, dev, and update flags. The shell wraps it in the canonical extern "C" __global__ void pirls_row_jit_{spec_id}(...) signature that launch_row_reweight_on_stream expects.

Fields

spec_id: u64

Process-unique identifier for this spec; the module cache uses it as a key so callers must reuse the same spec_id for the same body and pick a new one whenever the body changes.

body: String

CUDA body source. Must read from eta_c, y_i, wp, set flags, and assign to mu, grad_eta, w_fisher, w_hessian, w_solver, z_f, z_h, dev. See [COMMON_DEVICE_PROLOG] for the available helpers.

Implementations

impl JitFamilySpec

pub fn glm( spec_id: u64, family: PirlsRowFamily, curvature: CurvatureMode, ) -> Self

Level A: build a spec from a built-in (family, curvature) pair. The generator reuses the same per-family body as the built-in cached kernel — useful to validate the JIT pipeline end-to-end against the built-in numerical reference.

pub fn raw(spec_id: u64, body: impl Into<String>) -> Self

Level B: build a spec from caller-supplied body source. The kernel shell wraps it; the body must define the required locals listed on JitFamilySpec.

pub fn kernel_name(&self) -> String

The extern "C" kernel symbol the JIT-compiled module exposes.

pub fn cuda_source(&self, curvature: CurvatureMode) -> String

Build the full CUDA source ready for NVRTC compilation. The shell + prolog match the built-in cuda_source_for so the JIT kernel ABI is bit-identical to the cached built-ins; launch_row_reweight_on_stream cannot tell the difference.

Trait Implementations

impl Clone for JitFamilySpec

fn clone(&self) -> JitFamilySpec

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for JitFamilySpec

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.