Skip to content

Commit 4ae840d

Browse files
committed
Update
[ghstack-poisoned]
1 parent 11a5a02 commit 4ae840d

File tree

3 files changed

+9
-5
lines changed

3 files changed

+9
-5
lines changed

backends/test/suite/flow.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff 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
]
Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,10 @@
1-
from executorch.backends.vulkan.test.tester import VulkanTester
21
from executorch.backends.test.suite.flow import TestFlow
2+
from executorch.backends.vulkan.test.tester import VulkanTester
3+
34

45
def _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+
1517
VULKAN_TEST_FLOW = _create_vulkan_flow("vulkan")

backends/vulkan/test/tester.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,9 +10,9 @@
1010
import executorch.backends.test.harness.stages as BaseStages
1111

1212
import torch
13-
from executorch.backends.vulkan.partitioner.vulkan_partitioner import VulkanPartitioner
1413
from executorch.backends.test.harness import Tester as TesterBase
1514
from executorch.backends.test.harness.stages import StageType
15+
from executorch.backends.vulkan.partitioner.vulkan_partitioner import VulkanPartitioner
1616
from executorch.exir import EdgeCompileConfig
1717
from 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

0 commit comments

Comments
 (0)