trunk/67af8e140dce108c4c2fe6ffdec8317f2b93deb1: Fix AVX512 grid sampler mask gather (#197635)
- PyTorch 近 90 天出现 1578 次
- PyTorch 近 90 天第 1559 次发版
- 上一次:同一天稍早 · trunk/536fcc5eeabf50519f9173c7bf8410993d5b2440
发生了什么
This PR fixes an incorrect OP AI-assisted summary, reviewed by the author: Summary The AVX512 mask_gather converted an all-ones mask to a bitmask with floating-point equality. The all-ones pattern is NaN as float or double, so every comparison was false and grid_sample returned zeros when the AVX512 kernel ran. Extract the sign bit with _mm512_movepi{64,32}_mask , matching AVX2. Also make the sign bit the active-lan…
摘要按规则整理自下方来源原文