[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 2 commits 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. ❌ 3 New Failures, 1 Pending, 1 Unrelated FailureAs of commit f6f8a62 with merge base e03f777 ( NEW FAILURES - The following jobs have failed:
BROKEN TRUNK - The following job failed but were 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
|
…lden) Pull Request resolved: #20465 Adds the test suite for the aten.index.Tensor op (stacked on the op diff): - test/ops/index/test_index.py: exports a module computing x[idx] through 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. - test/native/test_index.cpp: a standalone Dawn binary that loads each .pte, feeds 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. - Wired into CMake (webgpu_index_test), test/TARGETS (python_unittest test_index), and the Dawn native CI script. ghstack-source-id: 397026576 @exported-using-ghexport Differential Revision: [D109479000](https://our.internmc.facebook.com/intern/diff/D109479000/)
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