mirror of https://github.com/llvm/torch-mlir
d0933b0eb6
Following up from the discussion in <https://github.com/llvm/torch-mlir/pull/3550>, I've edited the lowering to prevent OOB extracts in a more direct fashion (i.e., just clamping directly). I don't think this affects the lit tests at all, but I've tested the changes in our external test suite at <https://github.com/nod-ai/SHARK-TestSuite/tree/main/>. I found the issue when I was unexpectedly getting `nan`'s along the output image border for a resize test there. |
||
|---|---|---|
| .. | ||
| CMakeLists.txt | ||
| DataMovement.cpp | ||
| IndirectDataMovement.cpp | ||
| Linear.cpp | ||
| Pooling.cpp | ||
| PopulatePatterns.h | ||
| Random.cpp | ||
| Reduction.cpp | ||
| TensorConstructors.cpp | ||
| TensorScalarInterop.cpp | ||
| TorchToLinalg.cpp | ||
| Uncategorized.cpp | ||
| Utils.cpp | ||