Skip to content

Commit 131e309

Browse files
committed
Update base for 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]
1 parent b9e4fb5 commit 131e309

File tree

1 file changed

+1
-5
lines changed

1 file changed

+1
-5
lines changed

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)