viable/strict/1790972922: [MPS] conv1d to metal (#198878)
- PyTorch: 1843 events in the last 90 days
- PyTorch: 1819th Release in the last 90 days
- Previous: earlier the same day · viable/strict/1790970701: Document the new issue and PR workflow (#198527)
What happened
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…
Summary assembled by rule from the sources below
Why it's spreading
Sources
Release
- viable/strict/1790972922: [MPS] conv1d to metal (#198878)PyTorch Releases