Commit 3cebce2
authored
[TorchToLinalg] Direct lowering from Torch to Linalg for torch.aten.convolution_backward (#4384)
### Description:
- Added the direct lowering pass for `torch.aten.convolution_backward`
from Torch to Linalg. The pattern generates
`linalg.generic` ops instead of `linalg.conv_<>` for better lowering.
- Created new lit tests for backward convolution in the separate file
`convolution_backward.mlir`. Also added more test cases for better test
coverage.
- Added new e2e tests for backward convolution for better test coverage.
### Issue:
- #43551 parent e3200d9 commit 3cebce2
File tree
4 files changed
+1147
-7
lines changed- lib/Conversion/TorchToLinalg
- projects/pt1
- e2e_testing
- python/torch_mlir_e2e_test/test_suite
- test/Conversion/TorchToLinalg
4 files changed
+1147
-7
lines changed
0 commit comments