mirror of https://github.com/llvm/torch-mlir
218b4875d5
This commit adds support for type refinement when `torch.tensor_static_info_cast`s are involved, even when there are users of the casted tensor that don't allow type refinements. Originally the canonicalization pattern for `torch.tensor_static_info_cast` would check if all the users of the casted tensor allowed type refinements before making any changes. This means that if at least one of the users did not allow type refinements, the pattern would fail. This becomes an issue when doing shape calculations because the calculations need the shape information of each input tensor to be available before the calculation can be simplified. |
||
|---|---|---|
| .. | ||
| Torch | ||
| TorchConversion | ||
| CMakeLists.txt | ||