Skip to content

Commit c90d054

Browse files
committed
Update on "[ET-VK][12/n] aten.cat with more codegen"
1. The `aten.cat` operation are very straightforward using the `copy_*_node`. 2. Complexity comes from the code-gen. We need to introduce a `AT_TENSOR_LIST` type, which contains a list of `AT_TENSOR` with `is_in=True`. The tensor list itself as a container is not an `IOValueRef`, but the element inside are. It leads to some ugly if-then-else in the codegen. Differential Revision: [D56626865](https://our.internmc.facebook.com/intern/diff/D56626865/) [ghstack-poisoned]
2 parents 838e770 + 962b3df commit c90d054

File tree

1 file changed

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

1 file changed

+0
-3
lines changed

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

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

0 commit comments

Comments
 (0)