diff --git a/crates/ml/src/cuda_pipeline/gpu_dqn_trainer.rs b/crates/ml/src/cuda_pipeline/gpu_dqn_trainer.rs index 453ff8054..9dcd08375 100644 --- a/crates/ml/src/cuda_pipeline/gpu_dqn_trainer.rs +++ b/crates/ml/src/cuda_pipeline/gpu_dqn_trainer.rs @@ -403,9 +403,9 @@ impl GpuDqnTrainer { let b = config.batch_size; let total_params = compute_total_params(&config); - // Disable event tracking — required for CUDA Graph capture (no CudaEvents - // inside captured regions). Safe: single stream, explicit ordering. - unsafe { stream.context().disable_event_tracking(); } + // Event tracking: kept ENABLED during normal ops (buffer allocation, DtoD). + // DISABLED only during CUDA Graph capture (in capture_training_graph). + // Global disable pollutes the context for other tests/models. // Stack size for the training kernel: ~46KB/thread with NUM_ATOMS=51, // ~12KB/thread with NUM_ATOMS=11. DIST_SIZE(SCRATCH1_DIM=256) * 4 = 11KB