[ET-VK] Add optional blocklist and allowlist to vulkan partitioner to aid debugging

[SS-JIA]

Summary:

Changes

• Add operator_allowlist and operator_blocklist optional arguments to VulkanPartitioner
• operator_blocklist will prevent operators in the block list to be lowered to Vulkan
• operator_allowlist will only allow operators in the allow list to be lowered to Vulkan
• operator_allowlist takes precedence over operator_blocklist

Context

When debugging models, it is useful to be able to prevent certain operators from being lowered to Vulkan, or to only allow certain operators from being lowered to Vulkan. This can help isolate which ops are causing model output to be incorrect.

Test Plan:

Test Plan

Tested this feature locally while debugging example models.

[SS-JIA]

Stack from ghstack (oldest at bottom):

• [DNL] small change to trigger CI testing #13300
• [ET-VK][CI] Add vulkan CI to test exporting and running models #13295
• [ET-VK][examples] Create export script for Vulkan examples #13294
• [ET-VK][ez] Fix vulkan_schema interface and link Vulkan to pybindings #13327
• -> [ET-VK] Add optional blocklist and allowlist to vulkan partitioner to aid debugging #13326

[pytorch-bot]

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/executorch/13326

• 📄 Preview Python docs built from this PR

Note: Links to docs will display an error until the docs builds have been completed.

❌ 1 New Failure, 1 Pending

As of commit fb3083d with merge base 71c9a4f ():

NEW FAILURE - The following job has failed:

• Build documentation / build (buck2) / Build doc (gh)

This comment was automatically generated by Dr. CI and updates every 15 minutes.

[github-actions]

This PR needs a release notes: label

If your change should be included in the release notes (i.e. would users of this library care about this change?), please use a label starting with release notes:. This helps us keep track and include your important work in the next release notes.

To add a label, you can comment to pytorchbot, for example
@pytorchbot label "release notes: none"

For more information, see
https://github.com/pytorch/pytorch/wiki/PyTorch-AutoLabel-Bot#why-categorize-for-release-notes-and-how-does-it-work.