diff --git a/crates/ml/src/cuda_pipeline/gpu_dqn_trainer.rs b/crates/ml/src/cuda_pipeline/gpu_dqn_trainer.rs index eb6a876ea..8dc152265 100644 --- a/crates/ml/src/cuda_pipeline/gpu_dqn_trainer.rs +++ b/crates/ml/src/cuda_pipeline/gpu_dqn_trainer.rs @@ -2769,12 +2769,12 @@ impl GpuDqnTrainer { .map_err(|e| MLError::ModelError(format!("selectivity_forward load: {e}")))?; let sel_bwd_kernel = cpbi_module.load_function("selectivity_backward") .map_err(|e| MLError::ModelError(format!("selectivity_backward load: {e}")))?; - let vsn_kernel = cpbi_module.load_function("vsn_bottleneck_fwd") - .map_err(|e| MLError::ModelError(format!("vsn_bottleneck_fwd load: {e}")))?; - let glu_combine_kernel = cpbi_module.load_function("glu_gate_combine") - .map_err(|e| MLError::ModelError(format!("glu_gate_combine load: {e}")))?; - let glu_backward_kernel = cpbi_module.load_function("glu_gate_backward") - .map_err(|e| MLError::ModelError(format!("glu_gate_backward load: {e}")))?; + let vsn_kernel = cpbi_module.load_function("variable_select_bottleneck") + .map_err(|e| MLError::ModelError(format!("variable_select_bottleneck load: {e}")))?; + let glu_combine_kernel = cpbi_module.load_function("glu_combine") + .map_err(|e| MLError::ModelError(format!("glu_combine load: {e}")))?; + let glu_backward_kernel = cpbi_module.load_function("glu_backward") + .map_err(|e| MLError::ModelError(format!("glu_backward load: {e}")))?; info!("GpuDqnTrainer: Q-attn + selectivity + VSN + GLU kernels loaded"); // ── Compile CQL penalty kernel (if enabled) ──────────────────────