viable/strict/1790972922: [MPS] conv1d to metal (#198878)

发生了什么

Moves every MPS conv1d forward off the MPSGraph convolution. Short outputs run im2col + addmm , all other shapes run Metal kernels. Which kernel runs, first match wins: Kernel size >= 256: mps_convolution_2d_large_kernel , the existing direct conv2d Metal kernel. int32 index overflow: conv3d_metal_forward . 1 input channel per group and ( groups > 1 or min(C_out, L_out) ): conv1d_depthwise . groups == 1 and ( L_out…

摘要按规则整理自下方来源原文

来源