add support not decompose into vulkan_partition (#4056)

Summary:
Pull Request resolved: #4056

Operator `aten.upsample_nearest2d.vec` will be decomposed into several operators when lowering to Vulkan delegation. While we already have the implementation of this op in Vulkan. Add support to disable decomposition in Vulkan Partitioner.

Follow usage in this [example](https://www.internalfb.com/code/fbsource/[697a6da34c78]/fbcode/executorch/exir/backend/test/op_partitioner_demo.py?lines=158).

Reviewed By: jorgep31415

Differential Revision: D58900868

fbshipit-source-id: e251b6f0c96ed80731766d3451af1cd8c20fce96

Author: Yujie Hui

backends/vulkan/partitioner/vulkan_partitioner.py

@@ -7,7 +7,7 @@
 # pyre-strict
 
 import logging
-from typing import Any, Dict, final, List, Mapping, Optional
+from typing import Any, Callable, Dict, final, List, Mapping, Optional, Tuple
 
 import executorch.backends.vulkan.serialization.vulkan_graph_schema as vk_graph_schema
 
@@ -33,6 +33,11 @@
 
 from torch.fx.passes.operator_support import OperatorSupportBase
 
+# pyre-ignore
+ops_not_to_decompose = [
+    torch.ops.aten.upsample_nearest2d.vec,
+]
+
 
 class VulkanSupportedOperators(OperatorSupportBase):
     _ops: OpList = enumerate_supported_ops()
@@ -150,6 +155,11 @@ def __init__(self, compile_options: Optional[Dict[str, Any]] = None) -> None:
         compile_spec = parse_compile_options(self.options)
         self.delegation_spec = DelegationSpec(VulkanBackend.__name__, compile_spec)
 
+    def ops_to_not_decompose(
+        self, ep: ExportedProgram
+    ) -> Tuple[List[torch._ops.OpOverload], Optional[Callable[[torch.fx.Node], bool]]]:
+        return (ops_not_to_decompose, None)
+
     def partition(self, exported_program: ExportedProgram) -> PartitionResult:
         # Run the CapabilityBasedPartitioner to return the largest possible
         # subgraphs containing the nodes with the tags