Skip to content

Commit 8964e6e

Browse files
committed
Update on "[ET-VK][13/n] split_with_sizes with more test codegen"
Life is fun when the code-gen is more challenging than writing the operator itself. 1. Test codegen update to include vector of Tensor as output: `Tensor(a)[]`. 2. `aten.split_with_sizes.default` 3. `aten.split.Tensor` 4. Improve `DimUtils` for better dimension reasoning. Differential Revision: [D56660525](https://our.internmc.facebook.com/intern/diff/D56660525/) [ghstack-poisoned]
2 parents 4bbbde3 + 210716f commit 8964e6e

File tree

2 files changed

+0
-4
lines changed

2 files changed

+0
-4
lines changed

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

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

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

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,8 +8,6 @@
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/Copy.h>
1412

1513
#include <executorch/backends/vulkan/runtime/graph/ops/impl/utils/DimUtils.h>

0 commit comments

Comments
 (0)