Skip to content

Commit 519af1d

Browse files
committed
Update base for Update on "[ET-VK] Adding batch processing in x axis to conv2d dw shader by caching input texel for reuse."
This diff adds batch processing in the x axis to the conv2d dw shader by reusing input texel overlapping between consecutive tiles. The changes include modifying the glsl code for the conv2d dw output tile, adding a new parameter to the yaml file, and modifying the Convolution.cpp file to use the new parameter. Differential Revision: [D67868671](https://our.internmc.facebook.com/intern/diff/D67868671/) [ghstack-poisoned]
1 parent a93a2f2 commit 519af1d

File tree

1 file changed

+1
-4
lines changed

1 file changed

+1
-4
lines changed

backends/vulkan/runtime/graph/ops/glsl/indexing_utils_u16.h

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -13,10 +13,7 @@
1313

1414
u16vec3 idx_to_u16pos_x_wise(uint idx, int size_x, int size_y) {
1515
const uint div_by_x = idx / size_x;
16-
return u16vec3(
17-
idx % size_x,
18-
div_by_x % size_y,
19-
div_by_x / size_y);
16+
return u16vec3(idx % size_x, div_by_x % size_y, div_by_x / size_y);
2017
}
2118

2219
#endif // INDEXING_UTILS_U16_H

0 commit comments

Comments
 (0)