Skip to content

Commit 4899cb8

Browse files
committed
Update on "[ET-VK] Introduce Vulkan partitioner"
## Context Introduce `VulkanPartitioner`. I based the implementation of `VulkanPartitioner` on `ArmPartitioner`. Differential Revision: [D54128090](https://our.internmc.facebook.com/intern/diff/D54128090/) [ghstack-poisoned]
2 parents 77d6c8c + 131e309 commit 4899cb8

File tree

2 files changed

+1
-6
lines changed

2 files changed

+1
-6
lines changed

backends/vulkan/test/TARGETS

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,6 @@ python_unittest(
1717
"//executorch/backends/vulkan:vulkan_preprocess",
1818
"//executorch/backends/vulkan/partitioner:vulkan_partitioner",
1919
"//executorch/exir:lib",
20-
"//executorch/exir/backend:backend_api",
2120
"//executorch/extension/pybindings:portable_lib", # @manual
2221
"//executorch/extension/pytree:pylib",
2322
"//executorch/kernels/portable:custom_ops_generated_lib",

backends/vulkan/vulkan_preprocess.py

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -4,11 +4,10 @@
44
# This source code is licensed under the BSD-style license found in the
55
# LICENSE file in the root directory of this source tree.
66

7-
from typing import final, List, Optional
7+
from typing import final, List
88

99
import executorch.backends.vulkan.serialization.vulkan_graph_schema as vk_graph_schema
1010

11-
import torch
1211
from executorch.backends.vulkan.serialization.vulkan_graph_builder import VkGraphBuilder
1312
from executorch.backends.vulkan.serialization.vulkan_graph_serialize import (
1413
serialize_vulkan_graph,
@@ -24,10 +23,7 @@
2423
from executorch.exir.passes import MemoryPlanningPass, SpecPropPass
2524

2625
from executorch.exir.program._program import _copy_module
27-
from executorch.exir.tensor import TensorSpec
2826
from torch import dtype, float32
29-
from torch._export.utils import get_buffer, get_param, is_buffer, is_param
30-
from torch.fx import Node
3127

3228
DEFAULT_DEBUG_HANDLE = 65535
3329

0 commit comments

Comments
 (0)