Skip to content

Commit fdc6bf1

Browse files
committed
Update base for Update on "[ET-VK] Making stride equals dilation the default mode for conv2d dw."
This diff makes changes make stride equals dilation the default mode for conv2d dw output op. Adds a different source file to handle stride not equal dilation case. Differential Revision: [D67979760](https://our.internmc.facebook.com/intern/diff/D67979760/) [ghstack-poisoned]
1 parent 01d48a2 commit fdc6bf1

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

backends/vulkan/runtime/graph/ops/glsl/conv2d_dw_output_tile.glsl

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -130,7 +130,11 @@ void main() {
130130

131131
#else
132132
void main() {
133-
const ivec3 pos = idx_to_ipos_x_wise(gl_GlobalInvocationID.x, out_limits.x, out_limits.y);
133+
const uint div_by_x = gl_GlobalInvocationID.x / out_limits.x;
134+
const ivec3 pos = ivec3(
135+
gl_GlobalInvocationID.x % out_limits.x,
136+
div_by_x % out_limits.y,
137+
div_by_x / out_limits.y);
134138

135139
if (any(greaterThanEqual(pos, out_limits))) {
136140
return;

0 commit comments

Comments
 (0)