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:
@@ -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) ─
|
||||
|
||||
Reference in New Issue
Block a user