Skip to content

Commit 11d278e

Browse files
committed
Update on "[ET-VK][10/n] copy node, aten.repeat"
1. Introduce a `CopyNode` for generic copy-with-offset operations. 2. `aten.repeat` on all dimensions. 2.1 Use `CopyNode` where possible. 2.2. Specialized `repeat_channel` shader to handle packings 3. Update codegen to support `Methods` variant only operations. Need a new route to trigger the dispatch. Differential Revision: [D56499329](https://our.internmc.facebook.com/intern/diff/D56499329/) [ghstack-poisoned]
2 parents e7439ea + 19b4d7e commit 11d278e

File tree

1 file changed

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

1 file changed

+0
-5
lines changed

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

Lines changed: 0 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -8,15 +8,10 @@
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-
1411
#include <executorch/backends/vulkan/runtime/graph/ops/impl/utils/KernelUtils.h>
1512
#include <executorch/backends/vulkan/runtime/graph/ops/impl/utils/TensorUtils.h>
1613
#include <executorch/backends/vulkan/runtime/graph/ops/utils/ShaderNameUtils.h>
1714

18-
#include <iostream>
19-
2015
namespace vkcompute {
2116

2217
void add_copy_offset_node(

0 commit comments

Comments
 (0)