Skip to content

Commit 2668b10

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 7411f25 + 8228ecf commit 2668b10

File tree

1 file changed

+3
-0
lines changed
  • backends/vulkan/runtime/graph/ops/impl

1 file changed

+3
-0
lines changed

backends/vulkan/runtime/graph/ops/impl/Copy.cpp

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,9 @@
88

99
#include <executorch/backends/vulkan/runtime/graph/ops/OperatorRegistry.h>
1010

11+
#include <executorch/backends/vulkan/runtime/api/api.h>
12+
#include <executorch/backends/vulkan/runtime/graph/Logging.h>
13+
1114
#include <executorch/backends/vulkan/runtime/graph/ops/impl/utils/DimUtils.h>
1215
#include <executorch/backends/vulkan/runtime/graph/ops/impl/utils/KernelUtils.h>
1316
#include <executorch/backends/vulkan/runtime/graph/ops/impl/utils/TensorUtils.h>

0 commit comments

Comments
 (0)