Files
foxhunt/crates
jgrusewski 967c964dd8 fix(4branch): add branch 3 args to all 6 loss/grad kernel launches — fixes CUDA_ERROR_INVALID_VALUE
The CUDA kernels (.cu files) were updated to accept 4 branch pointers and
b3_size, but the Rust launch functions still only passed 3 branches. This
caused CUDA_ERROR_INVALID_VALUE on H100 due to argument count mismatch.

Fixed 7 launch sites in gpu_dqn_trainer.rs:
- launch_c51_loss: added on_b3/tg_b3/on_next_b3 pointers + b3_i32
- launch_c51_mixup: added branch_3_size arg
- launch_c51_grad: added b3_i32, fixed thread count 3*na → 4*na
- launch_mse_loss: added on_b3/tg_b3/on_next_b3 pointers + b3_i32
- launch_mse_grad_inner: added b3_i32, fixed thread count 3*na → 4*na
- apply_cql_gradient: added b3_i32
- compute_expected_q: added b3

Also updated max_branch calculations to include b3.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
2026-04-08 14:59:19 +02:00
..