You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Update on "[ET-VK] Creating specialized version of conv2d pw shader for X and Y stride = 1 and padding = 0."
This diff creates a specialized version of the conv2d pw shader for X and Y stride equals 1 and padding equals 0.
* It adds a new file `conv2d_pw_s1p0.glsl`, which implements the conv2d pw shader for X and Y stride equals 1 and padding equals 0.
* It adds a new file `conv2d_pw_s1p0.yaml`, which defines the parameters and shader variants for the specialized conv2d pw shader.
* The file `Convolution.cpp` is modified to add a new parameter `stride_1_padding_0` to the `conv2d` function, which enables the use of the specialized shader.
Differential Revision: [D75423931](https://our.internmc.facebook.com/intern/diff/D75423931/)
[ghstack-poisoned]
0 commit comments