test: remove plan_noise_inject from graph_mega — isolating hang cause

plan_noise_inject is the ONLY new kernel inside graph_mega vs the
working 11b1a1ca9. Everything else is identical (submit_forward_ops_main,
submit_aux_ops, capture_graph_mega). If this fixes the hang,
plan_noise_inject is incompatible with graph_mega on sm_90.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
This commit is contained in:
jgrusewski
2026-04-17 11:58:08 +02:00
parent f264ba00cb
commit 88307b8e3b

View File

@@ -7884,7 +7884,7 @@ impl GpuDqnTrainer {
self.launch_isv_feature_gate(batch_size)?;
self.launch_recursive_confidence_forward(batch_size)?;
self.launch_trade_plan_forward(batch_size)?;
self.launch_plan_noise_inject(batch_size)?;
// plan_noise_inject runs outside graph (pre-replay) — see fused_training.rs
}
// ── 2+3. Loss + gradient (blended MSE + C51 via c51_alpha ramp) ─