torch-mlir/lib/Dialect/Torch
Ramiro Leal-Cavazos 73bd32d06c
Make `getTensorRank` safer by changing return to `Optional<unsigned>` (#1707)
Currently `getTensorRank` returns -1 if it was unable to get the rank
of the tensor. However, not every use in the codebase was checking the
return value, and in some cases, the return value was casted to
unsigned leading to some infinte loops when an unranked tensor reached
a decomposition.

This commit changes the return of `getTensorRank` to
`Optional<unsigned>` to make it clear to the user that the function
can fail.

This commit also changes a couple of for loops that iterate a vector
in reverse order that can potentially become infinite loops into
range-based for loops.
2022-12-12 08:56:28 -08:00
..
IR Make `getTensorRank` safer by changing return to `Optional<unsigned>` (#1707) 2022-12-12 08:56:28 -08:00
Transforms Make `getTensorRank` safer by changing return to `Optional<unsigned>` (#1707) 2022-12-12 08:56:28 -08:00
Utils Make `getTensorRank` safer by changing return to `Optional<unsigned>` (#1707) 2022-12-12 08:56:28 -08:00
CMakeLists.txt E2e support for aten.softmax.int and aten.embedding 2021-10-18 17:57:45 -04:00