Skip to content

Fix NaN attention scores on MPS from uninitialized baddbmm buffer - #14459

Open
RudraMantri123 wants to merge 3 commits into
huggingface:mainfrom
RudraMantri123:fix-mps-sliced-attention-nan
Open

Fix NaN attention scores on MPS from uninitialized baddbmm buffer#14459
RudraMantri123 wants to merge 3 commits into
huggingface:mainfrom
RudraMantri123:fix-mps-sliced-attention-nan

Add a device-agnostic equivalence test for get_attention_scores

edbe78a
Select commit
Loading
Failed to load commit list.
Sign in for the full log view
size-label
succeeded Aug 27, 2026 in 3s