fix: bf16 precision failures in attention index construction for large inputs#236

jonfunk21 wants to merge 3 commits intoRosettaCommons:productionRosettaCommons/foundry:productionfrom

jonfunk21:fix/extend-index-mask-bfloat16-precisionjonfunk21/foundry:fix/extend-index-mask-bfloat16-precisionCopy head branch name to clipboard