viable/strict/1790592790: Fix AVX512 grid sampler mask gather (#197635)
PyTorch PR #197635 修复了 AVX512 版 mask_gather 的缺陷：该实现用浮点相等比较把全 1 掩码转换为位掩码，而全 1 位模式在 float/double 下是 NaN，导致所有比较为假，AVX512 内核运行时 grid_sample 返回全零。修复方式改为用 _mm512_movepi{64,32}_mask 提取符号位，与 AVX2 保持一致，并把符号位作为 SVE 与标量回退路径的活跃通道规则。mask_gather 现在在所有后端把掩码视为只读，使 GridSampler 可避免冗余掩码拷贝。新增向量级测试覆盖零、仅低位、仅符号位和全 1 掩码，并验证 mask_gather 不修改掩码。
Development
- First Reportviable/strict/1790592790: Fix AVX512 grid sampler mask gather (#197635)PyTorch Core
- Industry Responsetrunk/67af8e140dce108c4c2fe6ffdec8317f2b93deb1: Fix AVX512 grid sampler mask gather (#197635)PyTorch Core
- Current Assessment此类缺陷不会改变模型能力或市场格局，但反映底层算子库在 AI 加速路径上的维护风险：多后端 SIMD 实现各自为政时，正确性依赖逐后端测试覆盖。对依赖 PyTorch CPU 推理的团队，静默返回全零比崩溃更难排查。可验证下一信号：PyTorch 是否在 CI 中增加跨后端掩码语义一致性测试，或发布补丁版本说明。Agent Pulse · analysis
PyTorch 合并 PR #197635，修复 AVX512 grid sampler 的 mask_gather 错误。原实现通过浮点相等把全 1 掩码转成位掩码，但全 1 位模式在 float/double 下是 NaN，比较恒为假，导致 AVX512 内核下 grid_sample 输出全零。修复改用 _mm512_movepi{64,32}_mask 提取符号位，与 AVX2 对齐，并将符号位作为 SVE 和标量回退的活跃通道规则。同时 mask_gather 在所有后端将掩码视为只读，GridSampler 因此可省去冗余掩码拷贝。新增向量级测试覆盖零、仅低位、仅符号位、全 1 掩码，并验证掩码不被修改。作者说明该向量测试未在本地运行，因为 ARM64 检出未重新构建；现有 AVX512 验证命令为 ATEN_CPU_CAPABILITY=avx512 python test/test_nn.py TestNN -k test_grid_sample。
这是一处典型的掩码语义与浮点表示冲突：把位掩码当作浮点值做相等比较，在 NaN 位模式下必然失效，且只在 AVX512 路径暴露，说明不同 SIMD 后端对同一掩码约定的实现存在漂移。修复统一为符号位活跃通道规则并声明掩码只读，属于接口契约收紧，可能影响依赖掩码被就地修改的调用方。可验证下一信号：后续 PR 是否把该只读约定写入算子文档或测试，以及 SVE/标量路径是否出现同类回归。
此类缺陷不会改变模型能力或市场格局，但反映底层算子库在 AI 加速路径上的维护风险：多后端 SIMD 实现各自为政时，正确性依赖逐后端测试覆盖。对依赖 PyTorch CPU 推理的团队，静默返回全零比崩溃更难排查。可验证下一信号：PyTorch 是否在 CI 中增加跨后端掩码语义一致性测试，或发布补丁版本说明。
对使用 PyTorch CPU 推理做图像采样、变形或空间变换的团队，该修复消除了 AVX512 机器上静默输出全零的风险，减少一类难以定位的线上精度事故；掩码只读带来的拷贝消除也可能小幅降低采样算子开销。价值大小取决于受影响工作负载是否落在 AVX512 路径。
若该修复进入正式发布，AVX512 与 SVE 路径的 grid_sample 行为将趋于一致，掩码只读约定也可能被更多算子采纳。需要观察的是新增向量测试能否在 CI 中实际运行，以及是否出现针对掩码拷贝优化的后续改动。