SIGN IN SIGN UP

float8: snapshot FSDP precomputed scale (#4555)

Read the precomputed scale once in fsdp_pre_all_gather so concurrent
updates between FSDP iterations cannot change which value is used after
the branch is selected.

Assisted-by: Claude Opus 4.7

Co-authored-by: Ralf Gommers <ralf.gommers@gmail.com>
N
Neil Schemenauer committed
dd0efc7529dec0cb92d9e8cd3c027bc34c6c36c9
Parent: 3c466f5
Committed by GitHub <noreply@github.com> on 7/6/2026, 12:34:33 PM