Update

[ghstack-poisoned]

Author: GregoryComer

backends/qualcomm/tests/tester.py

@@ -13,8 +13,8 @@
 from executorch.backends.qualcomm._passes.qnn_pass_manager import QnnPassManager
 from executorch.backends.qualcomm.partition.qnn_partitioner import QnnPartitioner
 from executorch.backends.qualcomm.utils.utils import (
-    generate_qnn_executorch_compiler_spec,
     generate_htp_compiler_spec,
+    generate_qnn_executorch_compiler_spec,
     get_soc_to_chipset_map,
 )
 from executorch.backends.test.harness import Tester as TesterBase
@@ -36,7 +36,7 @@ def __init__(
         self,
         partitioners: Optional[List[Partitioner]] = None,
         edge_compile_config: Optional[EdgeCompileConfig] = None,
-        soc_model: str = "SM8650"
+        soc_model: str = "SM8650",
     ):
         backend_options = generate_htp_compiler_spec(use_fp16=True)
         self.chipset = get_soc_to_chipset_map()[soc_model]
@@ -47,7 +47,8 @@ def __init__(
 
         super().__init__(
             partitioners=partitioners or [QnnPartitioner(self.compiler_specs)],
-            edge_compile_config=edge_compile_config or EdgeCompileConfig(_check_ir_validity=False),
+            edge_compile_config=edge_compile_config
+            or EdgeCompileConfig(_check_ir_validity=False),
             default_partitioner_cls=QnnPartitioner,
         )
 
@@ -69,7 +70,7 @@ def __init__(
         module: torch.nn.Module,
         example_inputs: Tuple[torch.Tensor],
         dynamic_shapes: Optional[Tuple[Any]] = None,
-    ): 
+    ):
         # Specialize for Qualcomm
         stage_classes = (
             executorch.backends.test.harness.Tester.default_stage_classes()

backends/test/suite/flow.py

@@ -64,6 +64,7 @@ def all_flows() -> dict[str, TestFlow]:
 
     try:
         from executorch.backends.test.suite.flows.vulkan import VULKAN_TEST_FLOW
+
         flows += [
             VULKAN_TEST_FLOW,
         ]
@@ -72,6 +73,7 @@ def all_flows() -> dict[str, TestFlow]:
 
     try:
         from executorch.backends.test.suite.flows.qualcomm import QUALCOMM_TEST_FLOW
+
         flows += [
             QUALCOMM_TEST_FLOW,
         ]

backends/test/suite/flows/qualcomm.py

@@ -1,9 +1,10 @@
 from executorch.backends.qualcomm.tests.tester import QualcommTester
 from executorch.backends.test.suite.flow import TestFlow
 
+
 def _create_qualcomm_flow(
-    name: str, 
-    quantize: bool = False, 
+    name: str,
+    quantize: bool = False,
 ) -> TestFlow:
     return TestFlow(
         name,
@@ -12,4 +13,5 @@ def _create_qualcomm_flow(
         quantize=quantize,
     )
 
+
 QUALCOMM_TEST_FLOW = _create_qualcomm_flow("qualcomm")

backends/test/suite/flows/vulkan.py

@@ -1,9 +1,10 @@
-from executorch.backends.vulkan.test.tester import VulkanTester
 from executorch.backends.test.suite.flow import TestFlow
+from executorch.backends.vulkan.test.tester import VulkanTester
+
 
 def _create_vulkan_flow(
-    name: str, 
-    quantize: bool = False, 
+    name: str,
+    quantize: bool = False,
 ) -> TestFlow:
     return TestFlow(
         name,
@@ -12,4 +13,5 @@ def _create_vulkan_flow(
         quantize=quantize,
     )
 
+
 VULKAN_TEST_FLOW = _create_vulkan_flow("vulkan")

backends/vulkan/test/tester.py

@@ -10,9 +10,9 @@
 import executorch.backends.test.harness.stages as BaseStages
 
 import torch
-from executorch.backends.vulkan.partitioner.vulkan_partitioner import VulkanPartitioner
 from executorch.backends.test.harness import Tester as TesterBase
 from executorch.backends.test.harness.stages import StageType
+from executorch.backends.vulkan.partitioner.vulkan_partitioner import VulkanPartitioner
 from executorch.exir import EdgeCompileConfig
 from executorch.exir.backend.partitioner import Partitioner
 
@@ -33,7 +33,8 @@ def __init__(
         super().__init__(
             default_partitioner_cls=VulkanPartitioner,
             partitioners=partitioners,
-            edge_compile_config=edge_compile_config or EdgeCompileConfig(_check_ir_validity=False),
+            edge_compile_config=edge_compile_config
+            or EdgeCompileConfig(_check_ir_validity=False),
         )