Repository navigation
Commit d63dade
authored
Arm backend: Update VGF test for fused ReLU qparams (pytorch#22088)
Update the VGF rewrite-convolution regression for the ReLU fusion
behavior introduced by D116922192.
The ReLU is fused into the convolution before ConvertToClampPass runs,
so the clamp is fused away and no clamp node remains. Verify that the
fused convolution owns the output quantization parameters with qmin
rewritten to the zero point.
Rename the adjacent lowering test to describe the fused Conv-ReLU-Cat
path.
Test plan:
- `buck test @fbcode//mode/dev
fbcode//executorch/backends/arm/test:rewrite_conv_pass -- --exact
"fbcode//executorch/backends/arm/test:rewrite_conv_pass -
test_rewrite_conv_pass.py::test_fold_and_annotate_q_params_vgf_quant_tracks_fused_relu_qparams"`
- `buck test @fbcode//mode/dev
fbcode//executorch/backends/arm/test:rewrite_conv_pass -- --exact
"fbcode//executorch/backends/arm/test:rewrite_conv_pass -
test_rewrite_conv_pass.py::test_rewrite_conv_vgf_quant_handles_fused_conv_relu_cat_branch"
"fbcode//executorch/backends/arm/test:rewrite_conv_pass -
test_rewrite_conv_pass.py::test_rewrite_conv_vgf_quant_infers_quantized_bias_dtype_from_inputs"`
- `arc lint -a` and `arc lint` on changed files
- `python -m py_compile
backends/arm/test/passes/test_rewrite_conv_pass.py`
This PR was authored with Codex.
cc @digantdesai @freddan80 @per @zingo @oscarandersson8218 @mansnils
@Sebastian-Larsson @robell @rascani1 parent dd3b096 commit d63dade
1 file changed
Lines changed: 7 additions & 7 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
534 | 534 | | |
535 | 535 | | |
536 | 536 | | |
537 | | - | |
538 | | - | |
539 | | - | |
| 537 | + | |
540 | 538 | | |
541 | 539 | | |
542 | 540 | | |
543 | 541 | | |
544 | | - | |
| 542 | + | |
545 | 543 | | |
546 | 544 | | |
547 | | - | |
548 | | - | |
| 545 | + | |
| 546 | + | |
| 547 | + | |
| 548 | + | |
549 | 549 | | |
550 | 550 | | |
551 | 551 | | |
552 | | - | |
| 552 | + | |
553 | 553 | | |
554 | 554 | | |
555 | 555 | | |
| |||
0 commit comments