#4528·mlx

[BUG] 金属 `col_reduce_longcolumn` 对负步长视图产生错误结果

作者: mateuuszzzzz创建于 2026年9月17日更新于 2026年9月17日

☑️ 我了解严禁使用 AI 编写问题。 **描述错误** 在 Metal 后端中,在负步长情况下,对某些视图的归并返回了错误的结果。 CPU 后端和 NumPy 返回了正确的结果(对于没有反转的相同数组也是如此)。 代码段使用 `sum`,但对于 `prod`、`max`、`min`、`all`、`any` 以及 `mean` 和 `var`(均衡于 `sum`)也会出现相同的情况。 形状影响正确性,只有其中一个列归并内核受影响,而其他形状(例如 `(2, 512, 64)`)则被分配到正确工作的内核中。 **如何重现** 请参阅以下代码: