Commit Graph

4015 Commits

Author SHA1 Message Date
jgrusewski
abb0795759 feat: mag_concat_qdir + strided_accumulate kernels for Layer 3
mag_concat_qdir: computes direction E[Q] from softmax over C51 atoms,
normalizes by delta_z for matched gradient scale, concatenates with h_s2
to produce [B, SH2+3] input for magnitude branch FC.

strided_accumulate: extracts first SH2 columns from d_mag_concat [B, SH2+3]
into d_h_s2 [B, SH2] with beta accumulation for backward pass.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
2026-04-14 22:32:20 +02:00
jgrusewski
f49ae6858e fix: Layer 3 plan — add Q_dir normalization by delta_z
Raw E[Q] ~ 0.1 while h_s2 ~ 1.0 post-ReLU. Without normalization,
Q_dir columns learn 10x slower. Dividing by delta_z scales to ~1-10.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
2026-04-14 22:30:29 +02:00
jgrusewski
39ce4c15db fix: asymmetric spread gradient zero-sum — prevents Q-value inflation
Old: taken=+1, challenger=+0.5, rest=-0.33. Sum=+1.17 (biased up).
New: taken=+1, challenger=+0.5, rest=-1.5. Sum=0.0 (zero-mean).
Rest actions absorb -(1+0.5)/(A_d-2) to balance the challenger's push.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
2026-04-14 22:29:56 +02:00
jgrusewski
3f9a75264c fix: reward_rank_normalize cross-block race — two-buffer read/write split
In-place ranking races across blocks: block X writes shaped rewards[i]
before block Y loads it in tile scan. __syncthreads is block-local.
Fix: read from rewards_out (raw), write to gpu_batch.rewards (clone).
The DtoD clone from collect_experiences_gpu provides the output buffer.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
2026-04-14 22:28:44 +02:00
jgrusewski
8d2ed799e6 plan: direction-conditioned magnitude head (Layer 3) — 7 tasks
Tasks: CUDA kernel, param widening, forward conditioning, backward split,
zero-init, experience collector, H100 integration test.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
2026-04-14 22:19:41 +02:00
jgrusewski
234870d966 spec: direction-conditioned magnitude head (Layer 3)
Magnitude FC input changes from h_s2[B, SH2] to [h_s2; Q_dir][B, SH2+3].
Direction E[Q] values (3 scalars per sample) are computed from branch 0
logits and concatenated. w_b1fc grows by 3*adv_h parameters (768 for AH=256).

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
2026-04-14 22:14:49 +02:00
jgrusewski
b2eb1b7741 feat: asymmetric spread + per-branch scaling + distributional variance sizing + reward_std guard
Four improvements to the training pipeline:

1. Asymmetric spread gradient: challenger action (adjacent to taken) gets
   pushed UP at half strength instead of DOWN. Creates a two-horse race
   instead of single-action monopoly.

2. Per-branch spread scaling: spread_grad *= branch_scale. Direction branch
   (high impact) gets more spread than urgency (low impact).

3. Distributional variance position sizing (Layer 4): Var[Q] = E[Z²] - E[Z]²
   computed in compute_expected_q. Position scaled by 1/(1+sqrt(Var[Q_taken])).
   High uncertainty → smaller position. Kelly criterion from C51 atoms.

4. Reward std guard: skip rank normalization when observed_reward_std ≈ 0
   (epoch 0). Prevents zeroing all rewards before std is observed.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
2026-04-14 22:06:58 +02:00
jgrusewski
bde30c6c4d feat: Q-gap momentum modulates spread gradient — backs off during growth
spread_scale = inv_batch * delta_z * velocity_modulator
velocity_mod = clamp(1 - (Q_gap - Q_gap_ema) / delta_z, 0.1, 2.0)

Growing Q-gap → mod=0.1 (10% spread, let C51 learn).
Plateau → mod=1.0 (full spread, push Q-values apart).
Shrinking Q-gap → mod=2.0 (emergency spread, prevent collapse).
10% floor proven by train-w6qfd (val_Sharpe=21+ sustained).

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
2026-04-14 21:58:02 +02:00
jgrusewski
93e4185fa5 feat: rank-preserving signed reward standardization — SNR 0.001→0.5
r_shaped = sign(r) * rank(|r|) * std_ema. Counting-sort rank within
batch gives outlier-resistant, scale-free normalization. Sign preserved
for absolute direction. std_ema rescales to original magnitude range.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
2026-04-14 21:52:52 +02:00
jgrusewski
e9e4bd54ac plan: Adaptive Training Dynamics Layers 1+2 implementation plan
Task 1: Reward rank normalization kernel (SNR amplification)
Task 2: Q-gap momentum for spread gradient (plateau modulation)
Task 3: H100 integration test (50 epochs, compare to train-w6qfd)

Layers 3+4 (dir→mag conditioning, variance sizing) deferred to
separate plan after L1+L2 validation.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
2026-04-14 21:42:57 +02:00
jgrusewski
a5945db538 spec: Adaptive Training Dynamics v2 — 4-layer design for breaking Sharpe plateau
Layer 1: Rank-preserving signed reward standardization (SNR 0.001→0.5)
Layer 2: Temporal Q-gap momentum (spread modulation at plateaus)
Layer 3: Direction-conditioned magnitude (dir Q-values → mag head input)
Layer 4: Distributional variance position sizing (Kelly from C51 atoms)

Based on train-w6qfd results: val_Sharpe=21-25 sustained, Q-gap=0.138
growing. Builds on 32 fixes from this session. Implementation order:
1→2→4→3 (signal → momentum → variance → architecture).

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
2026-04-14 21:40:03 +02:00
jgrusewski
7eccfc53c9 feat: Q-gap floor gradient — perpetual action differentiation pressure
Adds a spread gradient to the C51 advantage logits that pushes the
taken action's distribution toward higher atoms and non-taken toward
lower. Scale = inv_batch * delta_z (adaptive to per-sample atom
resolution, zero hardcoded constants).

This gradient is ORTHOGONAL to the Bellman equation — it depends on
atom position, not target match. Active on ALL samples, providing
perpetual pressure to differentiate Q-values even when the C51
cross-entropy gradient vanishes at convergence. Prevents the Q-gap
plateau where all actions have identical Q-values.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
2026-04-14 21:07:23 +02:00
jgrusewski
6271322b06 fix: stochastic SARSA uses Adam step counter — fixed seed was deterministic
Philox seed used batch_size+sample_id (constant across steps) so the
same action was always sampled for the same sample → at the fixed point,
identical target every step → zero gradient. Now uses t_buf (Adam step
counter, device-mapped, increments every step). Each training step
samples a DIFFERENT action per sample → target is truly stochastic →
gradient variance prevents convergence.

Also moves philox_uniform to common_device_functions.cuh for reuse.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
2026-04-14 19:05:08 +02:00
jgrusewski
7a85a40788 fix: stochastic Expected SARSA — sample one action per target, not mixture
Deterministic softmax mixture produced a fixed target that the model
matched exactly → zero gradient → Q-gap frozen at 0.167, val_Sharpe
frozen at -5.62 from epoch 22. Stochastic sampling picks ONE action
per sample from the softmax weights. Expected target over many steps
is identical (unbiased) but per-step target varies → model can never
match a moving target → perpetual gradient → no convergence trap.

Also moves philox_uniform to common_device_functions.cuh so all
kernels can use it.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
2026-04-14 18:59:11 +02:00
jgrusewski
b1b5682e2b fix: Expected SARSA tau floor scales with Q magnitude — was fixed 0.01
Floor = max(|mean_Q| * 0.01, 1e-6). At Q-mean=0.005: floor=5e-5.
At Q-mean=1.0: floor=0.01. Fully adaptive, zero hardcoded constants.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
2026-04-14 18:10:10 +02:00
jgrusewski
69b911a611 fix: Expected SARSA tau=Q_gap (scale-invariant) — tau=Q_gap/3 caused convergence trap
tau=Q_gap/3 gave 20:1 weight ratio (95%/4%/1%), making the target
essentially argmax. Once the online distribution matched this sharp
target, cross-entropy gradient vanished → Q-gap froze at 0.158,
val_Sharpe locked at 2.40 from epoch 22.

tau=Q_gap gives a CONSTANT e:1 ≈ 2.72:1 ratio (63%/23%/14% for 3
actions) regardless of Q-gap magnitude. The target always maintains
meaningful mixture → perpetual gradient pressure → Q-values keep
evolving. Scale-invariant: the softmax ratio depends only on relative
Q-differences, not their absolute magnitude.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
2026-04-14 18:00:09 +02:00
jgrusewski
0b59b30549 fix: eval mode uses Boltzmann sampling — argmax froze val_Sharpe at 0.00
All 4 branches (dir, mag, order, urgency) now use Boltzmann sampling
in eval mode instead of argmax. argmax(softmax(Q/tau)) = argmax(Q) —
temperature is meaningless with argmax. Every state picked the same
winning action → identical trades → Sharpe=0.00 from epoch 8 onward.
Boltzmann sampling with Philox seed gives deterministic but
action-diverse evaluation. When Q-values differentiate, Boltzmann
naturally sharpens toward the best action.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
2026-04-14 16:58:26 +02:00
jgrusewski
c55d9d0276 fix: Expected SARSA temperature prevents uniform-averaging plateau
Softmax weights used implicit tau=1, producing near-uniform weights
when Q-gap=0.06: exp(0.06)≈1.06 → weights [0.35, 0.33, 0.32].
The mixture averaged all distributions → zero gradient to differentiate
actions → Q-gap frozen → val_Sharpe locked at 0.00 after epoch 19.

Now uses tau = max(Q_gap/3, 0.001). With Q-gap=0.06 and tau=0.02:
exp(0.06/0.02) = exp(3) ≈ 20:1 weight ratio between best and worst.
The 0.001 floor means even tiny Q-differences produce sharp weights,
preventing the uniform-averaging fixed point.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
2026-04-14 16:10:20 +02:00
jgrusewski
d06a18ef9f fix: IQN readiness update moved outside graph capture — cuStreamSynchronize invalidated capture
read_total_loss() calls cuStreamSynchronize which is illegal during
CUDA Graph stream capture. Was inside submit_aux_ops (captured in
graph_mega/graph_aux). Moved to Step 4 (conditional ops outside graph).

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
2026-04-14 15:26:53 +02:00
jgrusewski
7c00793c9f fix: exposure action select uses Boltzmann everywhere — argmax caused flips
branching_action_select exposure head used hard argmax while order/
urgency used Boltzmann. experience_action_select eval mode used pure
argmax for direction. Both replaced with Boltzmann softmax:
  tau = max(Q_max - Q_min, 0.01)
When Q-values differentiate: converges to argmax. When flat: spreads
evenly. Eval mode takes argmax of softmax probs (deterministic).

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
2026-04-14 15:19:44 +02:00
jgrusewski
1634dec042 fix: eval v_range width from Q-gap, baseline from Q-std — eliminates -40 dips
Two fixes:
1. Width: half = max(10*q_gap, 3*q_std, 0.1) instead of fixed 3*q_std+1.0.
   With q_gap=0.06 and 51 atoms, gives ~5 atoms of action resolution
   (was 1.5 atoms with the 1.0 floor). No more catastrophic -40 dips
   from atom underresolution.

2. Baseline: uses q_std_ema (proportional to natural Q oscillation)
   instead of fixed 0.01. Normal Q-oscillation (±q_std) no longer
   triggers aggressive alpha — only abnormal shifts do.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
2026-04-14 15:13:36 +02:00
jgrusewski
60528390e1 fix: adaptive IQN lambda + complete bf16 elimination (312 calls total)
Adaptive IQN lambda: reads IQN total_loss synchronously, tracks EMA
with adaptive alpha, computes readiness = (loss_initial - loss_ema) /
loss_initial. SAXPY scale = iqn_lambda_base * readiness. Suppresses
noisy IQN gradient early (readiness≈0), ramps to full weight as IQN
converges (readiness→1).

bf16 cleanup: 185 wrapper calls replaced across 12 .cu files
(attention, backtest_ppo, backtest_supervised, curiosity, dqn_utility,
dt, ensemble, her, iqn_cvar, monitoring, mse_loss, statistics).
Definitions removed from common_device_functions.cuh (legacy compat
block retained for PPO kernels only). Total: 312 bf16 calls eliminated
across both sessions.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
2026-04-14 14:46:17 +02:00
jgrusewski
7000151493 fix: IQL readiness uses adaptive EMA — was binary pop-on
Readiness jumped from 0 to ~0.8 in one step when CV first improved,
causing per-sample support to snap from [-1,1] to V(s)-centered
discontinuously. Same adaptive alpha pattern as eval_v_range:
α = |error| / (|error| + 0.05), clamped [0.01, 0.3]. Smooth ramp-in.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
2026-04-14 14:26:05 +02:00
jgrusewski
e3bd44f170 fix: eval EMA seeds from neutral v_range — snap-to-first caused epoch 3 spike
EMA init used == 0.0 sentinel and snapped to first non-zero Q-stats,
causing a discontinuous v_range transition → val_Sharpe=30 spike at
epoch 3 followed by -1 correction. Now seeds from neutral center
(mean=0, std=1 → v_range=[-4,4]) and the adaptive alpha tracks
to actual Q-stats smoothly over 2-3 steps.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
2026-04-14 14:24:59 +02:00
jgrusewski
9c2f5d75b5 fix: eval v_range EMA uses adaptive alpha — fixed α=0.05 lagged on Q-shifts
When Q-mean shifted from +0.01 to -0.01, fixed α=0.05 took ~20 updates
to catch up → misaligned backtest atoms → val_Sharpe dipped to -20.
Adaptive alpha: α = |error| / (|error| + 0.01), clamped [0.01, 0.5].
Tracks fast on Q-shifts (α→0.5), smooths when stable (α→0.01).

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
2026-04-14 14:01:49 +02:00
jgrusewski
8c3b7f5c5e fix: validation backtest uses EMA-smoothed eval v_range — was fixed [-240, 240]
Fixed v_range from hyperparams (reward_scale/gamma=±240) was 4800x too
wide for Q-values near 0. All Q-values mapped to the same 2-3 center
atoms, making backtest action selection essentially random → val_Sharpe
jumped between -11 and +0.14 arbitrarily. Now uses the same
EMA-smoothed eval_v_range that the experience collector uses.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
2026-04-14 13:39:44 +02:00
jgrusewski
5d877234b3 fix: Q-stats readback is synchronous — stale double-buffer caused 2-state oscillation
reduce_current_q_stats returned PREVIOUS call's results via async
double-buffer. When the eval_v_range EMA converged, stale readback
created a limit cycle: Q-mean alternated between exactly -0.0190 and
-0.2102 for 20+ epochs. Now uses synchronous cuStreamSynchronize +
cuMemcpyDtoH (5µs cost every 50 steps). Removes dead
flush_q_stats_readback and q_stats_ready field.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
2026-04-14 13:16:26 +02:00
jgrusewski
5c725f7c9d fix: C51 target uses softmax-weighted mixture instead of hard argmax
Hard argmax picks actions based on noise when Q-values are flat
(all near zero). This causes policy collapse — the model locks into
a single action pattern after 1 epoch (Trades=56647 locked, action
distribution frozen). Expected SARSA-style softmax mixture:
  p_target(z) = Σ_a softmax(E[Q(s',a)])[a] * p_target_a(z)
When Q-values differentiate, converges to argmax. When flat, averages
across actions → stable gradient signal → breaks policy collapse.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
2026-04-14 12:07:59 +02:00
jgrusewski
dea2ce1db7 fix: EMA-smooth eval v_range updates — per-step snapping caused val_Sharpe instability
update_eval_v_range snapped to current Q-stats every 50 steps. With
per-sample IQL support (the circular dependency fix), Q-stats fluctuate
between steps, causing the backtest atom placement to jump → val_Sharpe
oscillated between -42 and 0. EMA (α=0.05) smooths the transition,
giving the backtest consistent atoms across evaluations.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
2026-04-14 11:38:15 +02:00
jgrusewski
36c803f520 fix: Q-stats use per-sample IQL support — breaks circular eval_v_range feedback
compute_expected_q used eval_v_range to compute Q-values, then Q-stats
updated eval_v_range from those Q-values — circular dependency that
amplified any drift. Q-mean collapsed 0 → -0.69 in 10 epochs despite
mean_reward ≈ 0. Now uses per_sample_support from IQL (independently
trained V(s)), breaking the feedback loop.

Also removes 127 bf16 identity wrapper calls across 5 CUDA kernels
(c51_loss, epsilon_greedy, dqn_utility, experience, attention_backward).

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
2026-04-14 11:01:21 +02:00
jgrusewski
2b612b478f cleanup: remove bf16 wrappers from training guard Q-value kernels
qvalue_stats_reduce and qvalue_divergence_check used bf16() identity
wrappers on native f32 data — legacy from the bf16 era. These
wrappers can cause subtle precision differences across GPU
architectures. Replaced with native fminf/fmaxf/__shfl_xor_sync.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
2026-04-14 09:35:32 +02:00
jgrusewski
9260100200 fix: forward cuBLASLt also uses heuristic — AlgoGetIds caused Q-value drift on H100
Forward cached GEMMs (plain + RELU_BIAS) used AlgoGetIds which can
select graph-incompatible algorithms on H100 SM90. The DDQN target
forward pass uses these — biased target Q-values cause bootstrapping
collapse (Q-mean drifted from 0 to -0.69 over 10 epochs despite
mean_reward≈0). Heuristic selects graph-safe algorithms.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
2026-04-14 09:30:30 +02:00
jgrusewski
6bdda454b5 fix: adversarial regime uses ratio with hysteresis — consecutive counter was too brittle
Consecutive counter reset on a single epoch above threshold, so the
trigger=5 was never reached when max_dd naturally bounced every 3-4
epochs. Now uses EMA of binary low-dd signal (α=0.15) with hysteresis:
activates at ratio>0.6 (sustained low dd), deactivates at ratio<0.4
(recovered). No fixed trigger count, no single-epoch reset.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
2026-04-14 09:00:57 +02:00
jgrusewski
bdad095de1 fix: adversarial regime uses adaptive EMA threshold — was permanently locked on
Fixed threshold 0.5% MaxDD was calibrated for zero-gradient era.
With real gradients, max_dd=0.01-0.08% is always below 0.5%, so
adversarial activated at epoch 5 and never released — every epoch
trained under 3x spread, 2x tx_cost, 0.5x fill permanently.
Now uses EMA(max_dd) * 0.5 as threshold, self-calibrating to the
model's actual drawdown scale.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
2026-04-14 08:41:12 +02:00
jgrusewski
f74d8d018c fix: episode starts use epoch counter in Philox seed — breaks deterministic repetition
domain_rand_episode_starts used constant seed (i, 0, 9999) producing
identical episode starting bars every epoch. With deterministic starts
and alternating weight updates, the OOS Sharpe oscillated perfectly
between +0.85 and -1.0. Adding epoch to the Philox hash key diversifies
starting positions across epochs.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
2026-04-14 08:21:37 +02:00
jgrusewski
bf4e92bdb7 cleanup: remove BUFFER_DIAG, RECAPTURE_DIAG, GRAD_DIAG diagnostic infrastructure
Root causes are fixed (cuBLASLt heuristic, entropy inv_batch, overflow,
IQL OOB). Removes: run_buffer_diagnostics, run_recapture_diagnostics,
debug_buffer_norm_f32, diag_recapture_remaining field,
gradient_stage_diagnostics field + FOXHUNT_GRAD_DIAG env var.
Eliminates ~4 stream syncs per step for first 3 steps each fold.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
2026-04-14 08:10:46 +02:00
jgrusewski
a5a754ec53 cleanup: remove FOXHUNT_NO_GRAPH diagnostic code path
The ungraphed path served its purpose — confirmed cuBLASLt AlgoGetIds
was selecting graph-incompatible algorithms on H100. Now that the
heuristic fix is in place, the diagnostic is dead code.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
2026-04-14 08:06:36 +02:00
jgrusewski
172ed8690f fix: MSE grad uses IQL branch_scales — removes hardcoded 4.0x magnitude
MSE gradient kernel had hardcoded branch_scale=(d==1)?4.0:1.0 for
magnitude branch. Now uses the same IQL branch_scales [B,4] buffer
as c51_grad_kernel. Also fixes MSE entropy missing inv_batch.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
2026-04-14 07:53:00 +02:00
jgrusewski
91613c45b4 fix: MSE gradient entropy also missing inv_batch — same bug as C51
Per-branch entropy boost in mse_grad_kernel added O(1) term without
1/B normalization. Same pattern as the C51 entropy fix.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
2026-04-14 07:49:02 +02:00
jgrusewski
1393cf36f8 fix: entropy gradient missing inv_batch — grad_norm scaled with batch_size
The entropy regularization term in c51_grad_kernel was added without
the 1/B factor, making it O(1) per sample vs O(1/B) for the C51 loss
term. At batch_size=16384, entropy dominated by 16384x, causing
grad_norm=640 (should be ~33). Local batch=16: 5.5 → 0.29.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
2026-04-14 07:44:59 +02:00
jgrusewski
15369c8f50 fix: grad_norm_partials sized for d_logits clipping — 12KB overflow per step
d_logits clipping uses val_blocks(2048) + adv_blocks(2304) = 4352
partials, but buffer was sized for total_params(1298). 3054-slot
(12KB) overflow corrupted adjacent GPU memory on EVERY training step
inside graph_forward. This was captured in the CUDA Graph and
replayed every step — the actual root cause of zero grad_buf on H100.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
2026-04-14 07:26:20 +02:00
jgrusewski
d3d9c4a145 fix: IQL kernels decode factored actions into branch indices — eliminates OOB
iql_gather_q_taken and iql_compute_advantage_weights indexed q_out
with factored action (0..80) into a [B,12] buffer — massive OOB.
compute-sanitizer found 2586 errors. Now decodes factored action
(dir*b1*b2*b3 + mag*b2*b3 + ord*b3 + urg) into 4 branch indices
and sums per-branch Q-values. Also fixes total_actions config
(was product 81, now sum 12). Sanitizer: 0 errors.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
2026-04-14 00:19:26 +02:00
jgrusewski
2d704eaa07 fix: backward cuBLASLt uses heuristic algo — AlgoGetIds selects graph-incompatible algo on H100
AlgoGetIds picks algorithms by ID order. On H100 (SM90), the first
valid algo uses split-K/stream-K with internal workspace allocation
that silently produces zero output when replayed via CUDA Graph.
The heuristic selects graph-safe algorithms by design.

Also removes FOXHUNT_NO_GRAPH temp env var and raw DtoH diagnostic probe.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
2026-04-14 00:01:34 +02:00
jgrusewski
d6ada5266b diag: enable FOXHUNT_NO_GRAPH in Argo train template — isolate H100 graph issue
Temporarily sets FOXHUNT_NO_GRAPH=1 in train-best pods to run
forward+backward+Adam ungraphed. If grad_buf is non-zero ungraphed,
the root cause is CUDA Graph capture of cuBLASLt on H100 SM90.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
2026-04-13 23:44:10 +02:00
jgrusewski
407b3b04ff diag: FOXHUNT_NO_GRAPH env var bypasses all CUDA Graph capture
Runs forward+backward+Adam ungraphed to isolate whether H100
grad_buf=0 is caused by graph capture or the cuBLAS operation itself.
NVIDIA docs confirm cublasLtMatmul inside graphs can silently fail
on Hopper when algorithm uses internal workspace allocation.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
2026-04-13 23:42:59 +02:00
jgrusewski
9dce707348 diag: raw DtoH probe on grad_buf — bypass kernel to confirm H100 data
Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
2026-04-13 23:30:17 +02:00
jgrusewski
fee5a7cb3a fix: guard diagnostic buffer overflow — d_adv needed 2304 blocks, partials had 1137
debug_buffer_norm_f32 reuses grad_norm_partials (sized for total_params)
as scratch. d_adv_logits with batch_size*tba=589K elements needs 2304
blocks but partials only has 1137 — GPU memory overflow corrupting
adjacent allocations. On H100 this likely trashed grad_norm_buf,
explaining the persistent grad_norm=0.000000. Guard added to skip
oversized norms.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
2026-04-13 23:23:03 +02:00
jgrusewski
4159d201a0 fix: eliminate grad_buf pointer duplication — avg_grad=0 root cause
All code now uses ptrs.grad_buf (u64) as single source of truth.
Vaccine rewritten to use launch_cublas_backward_to(scratch) instead
of std::mem::swap on CudaSlice which caused pointer divergence
between diagnostics and Adam/graphs. Locally verified: avg_grad=6.13.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
2026-04-13 23:05:43 +02:00
jgrusewski
03cac6ffda perf: accumulator uses mapped pinned memory — eliminates memcpy_dtoh
volatile float* bypasses GPU L2 cache for correct read-modify-write
across kernel launches. CPU reads directly via read_volatile at epoch
boundary. Removes dead ptr variable in update_eval_v_range.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
2026-04-13 22:15:05 +02:00
jgrusewski
40cf48868a fix: tile attention backward per-sample grads — eliminates H100 OOM
attention d_params_per_sample was B*P*4 = 1.6GB at batch_size=16384.
Caused CUDA_ERROR_OUT_OF_MEMORY on H100.

Same tile fix as IQL: process min(B,256) samples per tile, accumulate
d_params via +=. Buffer shrinks from 1.6GB to 26MB.

Also fixed: attn_weight_grad_reduce uses += for tile accumulation.
Also fixed: IQL VRAM log now shows tiled allocation size.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
2026-04-13 21:55:13 +02:00