[ExecuTorch][WebGPU] select_copy op test suite (cases.py op-test framework)

Pull Request resolved: #20363

Registers `aten.select_copy.int` in the `cases.py` op-test framework: a `_select_suite` of 4 configs (leading/middle/last dim + negative index) that `generate_op_tests` exports and compares to a torch golden on Dawn. Also adds `test/ops/select/test_select.py` (`SelectModule` + `CONFIGS` + an export-delegation/eager smoke test) and the `aten.select_copy.int` partitioner-allowlist entry in `tester.py`.
ghstack-source-id: 397026513
@exported-using-ghexport

Differential Revision: [D108793161](https://our.internmc.facebook.com/intern/diff/D108793161/)

Author: JCNTH

backends/webgpu/test/op_tests/cases.py

@@ -40,6 +40,10 @@
     CONFIGS as _MUL_CONFIGS,
     MulModule,
 )
+from executorch.backends.webgpu.test.ops.test_select import (
+    CONFIGS as _SELECT_CONFIGS,
+    SelectModule,
+)
 from executorch.backends.webgpu.test.ops.test_view_copy import (
     CONFIGS as _VIEW_CONFIGS,
     ViewModule,
@@ -144,3 +148,8 @@ def _fn_config_suite(module_cls, configs) -> WebGPUTestSuite:
 @register_op_test("view_copy")
 def _view_copy_suite() -> WebGPUTestSuite:
     return _fn_config_suite(ViewModule, _VIEW_CONFIGS)
+
+
+@register_op_test("select")
+def _select_suite() -> WebGPUTestSuite:
+    return _fn_config_suite(SelectModule, _SELECT_CONFIGS)

backends/webgpu/test/ops/test_select.py

@@ -0,0 +1,67 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the BSD-style license found in the
+# LICENSE file in the root directory of this source tree.
+
+"""`aten.select_copy.int` module + configs for the WebGPU op-test framework.
+
+`SelectModule` + `CONFIGS` are imported by `cases.py` to drive the declarative
+op-test suite. `SelectTest` is the export-delegation smoke test.
+Configs cover the leading, middle, and last dim plus a negative index (output rank =
+input rank - 1).
+"""
+
+import unittest
+
+import torch
+
+from executorch.backends.vulkan.partitioner.vulkan_partitioner import VulkanPartitioner
+from executorch.exir import to_edge_transform_and_lower
+
+# name -> (input_shape, select_fn)
+CONFIGS = {
+    "dim0": ((3, 8, 4), lambda x: x[1]),
+    "middle": ((3, 8, 4), lambda x: x[:, 2]),
+    "last": ((3, 8, 4), lambda x: x[..., 3]),
+    "neg_idx": ((3, 8, 4), lambda x: x[:, -1]),
+}
+
+
+class SelectModule(torch.nn.Module):
+    def __init__(self, fn):
+        super().__init__()
+        self.fn = fn
+
+    def forward(self, x: torch.Tensor) -> torch.Tensor:
+        return self.fn(x)
+
+
+def _det_input(shape):
+    g = torch.Generator().manual_seed(0)
+    return torch.randn(*shape, generator=g, dtype=torch.float32)
+
+
+def _export(fn, x: torch.Tensor):
+    ep = torch.export.export(SelectModule(fn).eval(), (x,))
+    return to_edge_transform_and_lower(
+        ep, partitioner=[VulkanPartitioner()]
+    ).to_executorch()
+
+
+def _delegated(et) -> bool:
+    return any(
+        d.id == "VulkanBackend"
+        for plan in et.executorch_program.execution_plan
+        for d in plan.delegates
+    )
+
+
+class SelectTest(unittest.TestCase):
+    def test_export_delegates(self) -> None:
+        for name, (shape, fn) in CONFIGS.items():
+            with self.subTest(name=name):
+                et = _export(fn, _det_input(shape))
+                self.assertTrue(
+                    _delegated(et), f"Expected a VulkanBackend delegate (select {name})"
+                )

backends/webgpu/test/tester.py

@@ -23,6 +23,7 @@
     exir_ops.edge.et_vk.rms_norm.default,
     exir_ops.edge.aten.mul.Tensor,
     exir_ops.edge.aten.view_copy.default,
+    exir_ops.edge.aten.select_copy.int,
 ]