[MLIR][Torch] Add TorchToLinalg lowering for AtenAvgPool3dOp

[vivekkhandelwal1]

This commit also fixes the average pool op' test failing for OnnxToLinalg lowering.

Signed-Off By: Vivek Khandelwal [email protected]

[AmosLewis]

You will need this commit to
[ONNX] Fix padding attributes for onnx.AveragePool eb3b62f

[AmosLewis]

Could you also test with model nod-ai/SHARK-TestSuite#185 in shark-testsuites with --torchtolinag flag before iree bump:
python ./run.py --torchmlirbuild ../../torch-mlir/build --tolerance 0.001 0.001 --cachedir ./huggingface_cache --ireebuild ../../iree-build -f onnx -g models --mode onnx --report --tests onnx/models/VideoResNet_vaiq_int8 --torchtolinalg

[vivekkhandelwal1]

Could you also test with model nod-ai/SHARK-TestSuite#185 in shark-testsuites with --torchtolinag flag before iree bump: python ./run.py --torchmlirbuild ../../torch-mlir/build --tolerance 0.001 0.001 --cachedir ./huggingface_cache --ireebuild ../../iree-build -f onnx -g models --mode onnx --report --tests onnx/models/VideoResNet_vaiq_int8 --torchtolinalg

It fails during iree-compile with failure for 3d conv but passed for torch-mlir.

[vivekkhandelwal1]

You will need this commit to [ONNX] Fix padding attributes for onnx.AveragePool eb3b62f

We can do this in a seperate PR, since the current PR doesn't make any change in the Onnx implementation. Or, you can create a PR with your commit for the same.

[vivekkhandelwal1]

Could you also test with model nod-ai/SHARK-TestSuite#185 in shark-testsuites with --torchtolinag flag before iree bump: python ./run.py --torchmlirbuild ../../torch-mlir/build --tolerance 0.001 0.001 --cachedir ./huggingface_cache --ireebuild ../../iree-build -f onnx -g models --mode onnx --report --tests onnx/models/VideoResNet_vaiq_int8 --torchtolinalg

I'm getting this:

Status report for run: test-onnx using mode:onnx todtype:default backend:llvm-cpu

| tests                             | model-run   | onnx-import   | torch-mlir   | iree-compile   | inference   |
|:----------------------------------|:------------|:--------------|:-------------|:---------------|:------------|
| onnx/models/VideoResNet_vaiq_int8 | passed      | passed        | passed       | passed         | mismatch    |

[vivekkhandelwal1]

Hi @rsuderman, can you please re-review this PR?

[vivekkhandelwal1]

Hi @rsuderman, can you please re-review this?

[vivekkhandelwal1]

Hi @rsuderman, can you please re-review this?

Hi @rsuderman, just a gentle reminder to review this PR.