WAR sort_chunks_by_index intermittent failures in L0 JAX unitttest#2730

Merged

tdophung merged 2 commits into

NVIDIA:mainfrom

tdophung:sort-chunks-WAR

Mar 5, 2026

Commits on Mar 4, 2026

Add WAR to alias input (output_grads) to output (input_grads) of the sort_chunk_by_map bwd function
tdophung
committed
adapt pytorch call site to extra dummy pointer
tdophung
committed