transformer_nuggets icon indicating copy to clipboard operation
transformer_nuggets copied to clipboard

FlexAttention currently evaluates on partial blocks all values and this can lead to IMA

Open drisspg opened this issue 8 months ago • 1 comments
trafficstars

See https://github.com/pytorch/pytorch/issues/147551#issuecomment-2683700299

drisspg avatar Feb 26 '25 02:02 drisspg