Skip to content

Commit 3cebce2

Browse files
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: - #4355
1 parent e3200d9 commit 3cebce2

File tree

4 files changed

+1147
-7
lines changed

4 files changed

+1147
-7
lines changed

0 commit comments

Comments
 (0)