Skip to content

Commit d8191ee

Browse files
committed
Update on "[ET-VK][11/n] copy_channel_offsets node"
1. Add a node `copy_channel_offsets` specifically for copying along the channel dimension, it needs extra attention at the boundaries due to channel packing. 1.1. `copy_channel_offsets` will be useful for `aten.cat` and `aten.split`. 2. Create `etvk.*` operators to facilitate testing. Add test case for both `copy_offset` and `copy_channel_offset`. Differential Revision: [D56554426](https://our.internmc.facebook.com/intern/diff/D56554426/) [ghstack-poisoned]
2 parents 8f518b3 + ba0bd85 commit d8191ee

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

backends/vulkan/test/vulkan_compute_api_test.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1133,7 +1133,7 @@ TEST(
11331133
}
11341134

11351135
for (int n_idx = 0; n_idx < n; n_idx++) {
1136-
for (int c_idx = b_dst_offset + b_range + 1; c_idx < c; c_idx++) {
1136+
for (int c_idx = b_dst_offset + b_range; c_idx < c; c_idx++) {
11371137
for (int h_idx = 0; h_idx < h; h_idx++) {
11381138
for (int w_idx = 0; w_idx < w; w_idx++) {
11391139
auto dst_idx = get_buf_idx(graph, out, {n_idx, c_idx, h_idx, w_idx});

0 commit comments

Comments
 (0)