[ExecuTorch][WebGPU] Op-test for pow.Tensor_Scalar#21163
Conversation
This PR needs a
|
There was a problem hiding this comment.
Pull request overview
Adds WebGPU op-test coverage for aten.pow.Tensor_Scalar by introducing a small module wrapper with a baked scalar exponent and registering a corresponding declarative test suite in the WebGPU op-test registry.
Changes:
- Added
PowScalarModuleandPOW_SCALAR_CONFIGS(square and sqrt exponents) to the unary activations test helpers. - Registered a new
pow_scalarop-test suite incases.pythat generates deterministic positive-base inputs to avoidpow(neg, frac) -> NaNbehavior.
Reviewed changes
Copilot reviewed 2 out of 2 changed files in this pull request and generated no comments.
| File | Description |
|---|---|
| backends/webgpu/test/ops/test_unary_activations.py | Adds PowScalarModule plus config constants used by the declarative op-test suites. |
| backends/webgpu/test/op_tests/cases.py | Registers the pow_scalar suite and wires it into the op-test case generation. |
💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.
psiddh
left a comment
There was a problem hiding this comment.
Approving full WebGPU stack
Stack from ghstack (oldest at bottom):
Adds the pow_scalar op-test suite (square + sqrt exponents, positive base) for the op ported in the diff below.
Key changes:
test/ops/test_unary_activations.py—PowScalarModule+POW_SCALAR_CONFIGS.test/op_tests/cases.py—_pow_scalar_suite.@exported-using-ghexport
Differential Revision: D112257612
Differential Revision: D112257612