update Vulkan nn.module test to support not decomposition (#4054)

Summary:
Pull Request resolved: #4054

In order to test the operators that are disabled to decompose, we need different APIs from excutorch to compile the model. We need to use `_to_edge_transform_and_lower()` by passing the Vulkan partitioner.  If we use this API, we don't use the combination of `to_edge()` and `to_backend()`.

https://www.internalfb.com/code/fbsource/[1176a06a4b48]/xplat/executorch/exir/program/_program.py?lines=862

Add a test option in `lower_module_and_test_output`. By default, it will call normal APIs, with `not_decompose` enabled, it will call
`_to_edge_transform_and_lower()` instead.

Reviewed By: jorgep31415

Differential Revision: D58907317

Author: Yujie Hui

backends/vulkan/test/test_vulkan_delegate.py

@@ -20,12 +20,13 @@
 from executorch.backends.vulkan.partitioner.vulkan_partitioner import VulkanPartitioner
 from executorch.backends.vulkan.vulkan_preprocess import VulkanBackend
 
-from executorch.exir import EdgeCompileConfig, EdgeProgramManager, to_edge
+from executorch.exir import EdgeCompileConfig
 from torch.export import Dim, export, ExportedProgram
 
 ctypes.CDLL("libvulkan.so.1")
 
 
+from executorch.exir.program._program import _to_edge_transform_and_lower
 from executorch.extension.pybindings.portable_lib import (  # @manual
     _load_for_executorch_from_buffer,
 )
@@ -119,16 +120,15 @@ def run_test(memory_layout):
             program: ExportedProgram = export(
                 model, sample_inputs, dynamic_shapes=dynamic_shapes
             )
-            edge_program: EdgeProgramManager = to_edge(
-                program, compile_config=self._edge_compile_config
-            )
 
-            edge_program = edge_program.transform(
-                [I64toI32(self._edge_compile_config._skip_dim_order), MeanToSumDiv()]
+            edge_program = _to_edge_transform_and_lower(
+                program,
+                transform_passes=[
+                    I64toI32(self._edge_compile_config._skip_dim_order),
+                    MeanToSumDiv(),
+                ],
+                partitioner=[VulkanPartitioner(compile_options)],
             )
-
-            edge_program = edge_program.to_backend(VulkanPartitioner(compile_options))
-
             executorch_program = edge_program.to_executorch()
 
             self.assertEqual(