Skip to content

Commit 2220d72

Browse files
committed
Update base for Update on "[ET-VK][9/n] clone node"
Introduce a clone node for copy operation. Also register `aten.clone` to this node. Important to note that during model export, possible to point the lvalue of `aten.clone` to the underlying shared object of the rvalue to achieve no-copy. Differential Revision: [D56441547](https://our.internmc.facebook.com/intern/diff/D56441547/) [ghstack-poisoned]
1 parent 3f88753 commit 2220d72

File tree

2 files changed

+2
-3
lines changed

2 files changed

+2
-3
lines changed

backends/vulkan/runtime/graph/ops/impl/Permute.h

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,8 @@
66
* LICENSE file in the root directory of this source tree.
77
*/
88

9+
#pragma once
10+
911
#include <executorch/backends/vulkan/runtime/api/api.h>
1012

1113
#include <executorch/backends/vulkan/runtime/graph/ComputeGraph.h>

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

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

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

11-
#include <executorch/backends/vulkan/runtime/api/api.h>
12-
1311
#include <executorch/backends/vulkan/runtime/graph/ops/impl/Permute.h>
1412
#include <executorch/backends/vulkan/runtime/graph/ops/impl/utils/KernelUtils.h>
15-
#include <executorch/backends/vulkan/runtime/graph/ops/impl/utils/TensorUtils.h>
1613
#include <executorch/backends/vulkan/runtime/graph/ops/utils/ShaderNameUtils.h>
1714

1815
namespace vkcompute {

0 commit comments

Comments
 (0)