pub fn hc_pre(
gpu: &dyn GpuBackend,
kernel: KernelHandle,
streams: DevicePtr,
hc_fn: DevicePtr,
hc_scale: DevicePtr,
hc_base: DevicePtr,
y_out: DevicePtr,
post_out: DevicePtr,
comb_out: DevicePtr,
num_tokens: u32,
hidden_size: u32,
hc_mult: u32,
sinkhorn_iters: u32,
norm_eps: f32,
hc_eps: f32,
stream: u64,
) -> Result<()>Expand description
Collapse hc_mult streams to one (RMS-rescaled mix → sigmoid pre
weighted sum) and emit post / comb (Sinkhorn) for the matching
hc_post. One block per token.