File tree Expand file tree Collapse file tree 3 files changed +9
-5
lines changed
Expand file tree Collapse file tree 3 files changed +9
-5
lines changed Original file line number Diff line number Diff line change @@ -64,6 +64,7 @@ def all_flows() -> dict[str, TestFlow]:
6464
6565 try :
6666 from executorch .backends .test .suite .flows .vulkan import VULKAN_TEST_FLOW
67+
6768 flows += [
6869 VULKAN_TEST_FLOW ,
6970 ]
Original file line number Diff line number Diff line change 1- from executorch .backends .vulkan .test .tester import VulkanTester
21from executorch .backends .test .suite .flow import TestFlow
2+ from executorch .backends .vulkan .test .tester import VulkanTester
3+
34
45def _create_vulkan_flow (
5- name : str ,
6- quantize : bool = False ,
6+ name : str ,
7+ quantize : bool = False ,
78) -> TestFlow :
89 return TestFlow (
910 name ,
@@ -12,4 +13,5 @@ def _create_vulkan_flow(
1213 quantize = quantize ,
1314 )
1415
16+
1517VULKAN_TEST_FLOW = _create_vulkan_flow ("vulkan" )
Original file line number Diff line number Diff line change 1010import executorch .backends .test .harness .stages as BaseStages
1111
1212import torch
13- from executorch .backends .vulkan .partitioner .vulkan_partitioner import VulkanPartitioner
1413from executorch .backends .test .harness import Tester as TesterBase
1514from executorch .backends .test .harness .stages import StageType
15+ from executorch .backends .vulkan .partitioner .vulkan_partitioner import VulkanPartitioner
1616from executorch .exir import EdgeCompileConfig
1717from executorch .exir .backend .partitioner import Partitioner
1818
@@ -33,7 +33,8 @@ def __init__(
3333 super ().__init__ (
3434 default_partitioner_cls = VulkanPartitioner ,
3535 partitioners = partitioners ,
36- edge_compile_config = edge_compile_config or EdgeCompileConfig (_check_ir_validity = False ),
36+ edge_compile_config = edge_compile_config
37+ or EdgeCompileConfig (_check_ir_validity = False ),
3738 )
3839
3940
You can’t perform that action at this time.
0 commit comments