mirror of https://github.com/llvm/torch-mlir
718f53ff8a
In PyTorch, the `NumberType` is equal to `Union[int, float, complex]`. However, the abstract interpretation library was treating the `NumberType` as `Union[int, float]`, resulting in type mismatches when reifying certain dtype functions. This commit fixes the type inconsistency by having the abstract interpretation functions take as an input a `Union[int, float, complex]` for the ops that take `!torch.number` inputs. |
||
|---|---|---|
| .. | ||
| CAPI | ||
| Conversion | ||
| Dialect | ||
| RefBackend | ||
| python | ||
| CMakeLists.txt | ||
| lit.cfg.py | ||
| lit.site.cfg.py.in | ||