[ExecuTorch][WebGPU] aten.index.Tensor test suite (export + native golden)#20465
[ExecuTorch][WebGPU] aten.index.Tensor test suite (export + native golden)#20465JulianCloudNTH wants to merge 1 commit into
Conversation
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/executorch/20465
Note: Links to docs will display an error until the docs builds have been completed. ❗ 1 Active SEVsThere are 1 currently active SEVs. If your PR is affected, please view them below: ❌ 4 New Failures, 4 Unrelated FailuresAs of commit 8f74aad with merge base 1b726b2 ( NEW FAILURES - The following jobs have failed:
FLAKY - The following job failed but was likely due to flakiness present on trunk:
BROKEN TRUNK - The following jobs failed but was present on the merge base:👉 Rebase onto the `viable/strict` branch to avoid these failures
This comment was automatically generated by Dr. CI and updates every 15 minutes. |
This PR needs a
|
Stack from ghstack (oldest at bottom):
Adds the test suite for the aten.index.Tensor op (stacked on the op diff):
VulkanPartitioner for four configs (reorder/repeat indices over distinct self
values, so a wrong-gather is visible), asserts a VulkanBackend delegate with
index.Tensor absorbed (not a CPU fallback), and writes per-config .pte +
.self/.idx/.golden.bin.
self (fp32) + index (int64 at the program boundary, narrowed to the int32 buffer)
and compares the gather against the torch golden at 1e-3, with a single-output
shape guard.
and the Dawn native CI script.
Differential Revision: D109479000