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
..
2026-03-13 10:18:35 +01:00
2026-04-02 16:29:55 +02:00
2026-03-12 19:10:18 +01:00
2026-03-14 11:35:15 +01:00
2026-03-13 10:18:35 +01:00
2026-03-13 10:18:35 +01:00
2026-03-01 22:47:39 +01:00
2026-04-08 14:59:19 +02:00
2026-03-13 10:18:35 +01:00
2026-03-13 10:18:35 +01:00
2026-03-13 10:18:35 +01:00
2026-04-04 01:24:19 +02:00
2026-03-01 22:47:39 +01:00
2026-03-13 10:18:35 +01:00
2026-04-08 14:34:04 +02:00
2026-03-29 22:35:37 +02:00
2026-03-29 22:35:37 +02:00
2026-03-31 12:37:25 +02:00
2026-03-29 19:18:47 +02:00
2026-03-19 00:39:03 +01:00
2026-03-13 10:18:35 +01:00
2026-03-13 10:18:35 +01:00
2026-03-31 00:39:59 +02:00
2026-03-16 21:01:28 +01:00
2026-03-13 10:18:35 +01:00
2026-03-13 10:18:35 +01:00
2026-03-13 10:18:35 +01:00
2026-03-10 13:18:57 +01:00
2026-03-28 01:51:45 +01:00
2026-03-13 10:18:35 +01:00
2026-03-13 10:18:35 +01:00
2026-03-13 10:18:35 +01:00
2026-03-18 00:53:47 +01:00
2026-03-14 11:35:15 +01:00
2026-03-01 22:47:39 +01:00
2026-03-15 11:59:31 +01:00
2026-03-01 22:47:39 +01:00
2026-03-03 22:16:35 +01:00