gitadmin/foxhunt
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: correct CUDA kernel names for VSN/GLU in constructor

Browse Source 
vsn_bottleneck_fwd → variable_select_bottleneck
glu_gate_combine → glu_combine
glu_gate_backward → glu_backward

19/19 smoke tests pass with full CPBI stack.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
This commit is contained in:
jgrusewski
2026-04-15 00:36:34 +02:00
parent f867064786
commit 3a6cfe9a9b
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
crates/ml/src/cuda_pipeline/gpu_dqn_trainer.rs
View File

@@ -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) ──────────────────────