mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
879d01afcb
We should support these to whatever extent we can. They corresponding `torch.uint<w>` types are defined, so I don't see an issue with generating the various casting rules and allowing them to trace. Pull Request resolved: https://github.com/pytorch/pytorch/pull/125717 Approved by: https://github.com/lezcano
1368 lines
42 KiB
Python
1368 lines
42 KiB
Python
# mypy: ignore-errors
|
|
|
|
import torch
|
|
|
|
# These two dicts are autogenerated with autogen/gen_dtypes.py,
|
|
# using numpy version 1.24.3.
|
|
|
|
_can_cast_dict = {
|
|
"no": {
|
|
torch.float16: {
|
|
torch.float16: True,
|
|
torch.float32: False,
|
|
torch.float64: False,
|
|
torch.complex64: False,
|
|
torch.complex128: False,
|
|
torch.uint8: False,
|
|
torch.uint16: False,
|
|
torch.uint32: False,
|
|
torch.uint64: False,
|
|
torch.int8: False,
|
|
torch.int16: False,
|
|
torch.int32: False,
|
|
torch.int64: False,
|
|
torch.bool: False,
|
|
},
|
|
torch.float32: {
|
|
torch.float16: False,
|
|
torch.float32: True,
|
|
torch.float64: False,
|
|
torch.complex64: False,
|
|
torch.complex128: False,
|
|
torch.uint8: False,
|
|
torch.uint16: False,
|
|
torch.uint32: False,
|
|
torch.uint64: False,
|
|
torch.int8: False,
|
|
torch.int16: False,
|
|
torch.int32: False,
|
|
torch.int64: False,
|
|
torch.bool: False,
|
|
},
|
|
torch.float64: {
|
|
torch.float16: False,
|
|
torch.float32: False,
|
|
torch.float64: True,
|
|
torch.complex64: False,
|
|
torch.complex128: False,
|
|
torch.uint8: False,
|
|
torch.uint16: False,
|
|
torch.uint32: False,
|
|
torch.uint64: False,
|
|
torch.int8: False,
|
|
torch.int16: False,
|
|
torch.int32: False,
|
|
torch.int64: False,
|
|
torch.bool: False,
|
|
},
|
|
torch.complex64: {
|
|
torch.float16: False,
|
|
torch.float32: False,
|
|
torch.float64: False,
|
|
torch.complex64: True,
|
|
torch.complex128: False,
|
|
torch.uint8: False,
|
|
torch.uint16: False,
|
|
torch.uint32: False,
|
|
torch.uint64: False,
|
|
torch.int8: False,
|
|
torch.int16: False,
|
|
torch.int32: False,
|
|
torch.int64: False,
|
|
torch.bool: False,
|
|
},
|
|
torch.complex128: {
|
|
torch.float16: False,
|
|
torch.float32: False,
|
|
torch.float64: False,
|
|
torch.complex64: False,
|
|
torch.complex128: True,
|
|
torch.uint8: False,
|
|
torch.uint16: False,
|
|
torch.uint32: False,
|
|
torch.uint64: False,
|
|
torch.int8: False,
|
|
torch.int16: False,
|
|
torch.int32: False,
|
|
torch.int64: False,
|
|
torch.bool: False,
|
|
},
|
|
torch.uint8: {
|
|
torch.float16: False,
|
|
torch.float32: False,
|
|
torch.float64: False,
|
|
torch.complex64: False,
|
|
torch.complex128: False,
|
|
torch.uint8: True,
|
|
torch.uint16: False,
|
|
torch.uint32: False,
|
|
torch.uint64: False,
|
|
torch.int8: False,
|
|
torch.int16: False,
|
|
torch.int32: False,
|
|
torch.int64: False,
|
|
torch.bool: False,
|
|
},
|
|
torch.uint16: {
|
|
torch.float16: False,
|
|
torch.float32: False,
|
|
torch.float64: False,
|
|
torch.complex64: False,
|
|
torch.complex128: False,
|
|
torch.uint8: False,
|
|
torch.uint16: True,
|
|
torch.uint32: False,
|
|
torch.uint64: False,
|
|
torch.int8: False,
|
|
torch.int16: False,
|
|
torch.int32: False,
|
|
torch.int64: False,
|
|
torch.bool: False,
|
|
},
|
|
torch.uint32: {
|
|
torch.float16: False,
|
|
torch.float32: False,
|
|
torch.float64: False,
|
|
torch.complex64: False,
|
|
torch.complex128: False,
|
|
torch.uint8: False,
|
|
torch.uint16: False,
|
|
torch.uint32: True,
|
|
torch.uint64: False,
|
|
torch.int8: False,
|
|
torch.int16: False,
|
|
torch.int32: False,
|
|
torch.int64: False,
|
|
torch.bool: False,
|
|
},
|
|
torch.uint64: {
|
|
torch.float16: False,
|
|
torch.float32: False,
|
|
torch.float64: False,
|
|
torch.complex64: False,
|
|
torch.complex128: False,
|
|
torch.uint8: False,
|
|
torch.uint16: False,
|
|
torch.uint32: False,
|
|
torch.uint64: True,
|
|
torch.int8: False,
|
|
torch.int16: False,
|
|
torch.int32: False,
|
|
torch.int64: False,
|
|
torch.bool: False,
|
|
},
|
|
torch.int8: {
|
|
torch.float16: False,
|
|
torch.float32: False,
|
|
torch.float64: False,
|
|
torch.complex64: False,
|
|
torch.complex128: False,
|
|
torch.uint8: False,
|
|
torch.uint16: False,
|
|
torch.uint32: False,
|
|
torch.uint64: False,
|
|
torch.int8: True,
|
|
torch.int16: False,
|
|
torch.int32: False,
|
|
torch.int64: False,
|
|
torch.bool: False,
|
|
},
|
|
torch.int16: {
|
|
torch.float16: False,
|
|
torch.float32: False,
|
|
torch.float64: False,
|
|
torch.complex64: False,
|
|
torch.complex128: False,
|
|
torch.uint8: False,
|
|
torch.uint16: False,
|
|
torch.uint32: False,
|
|
torch.uint64: False,
|
|
torch.int8: False,
|
|
torch.int16: True,
|
|
torch.int32: False,
|
|
torch.int64: False,
|
|
torch.bool: False,
|
|
},
|
|
torch.int32: {
|
|
torch.float16: False,
|
|
torch.float32: False,
|
|
torch.float64: False,
|
|
torch.complex64: False,
|
|
torch.complex128: False,
|
|
torch.uint8: False,
|
|
torch.uint16: False,
|
|
torch.uint32: False,
|
|
torch.uint64: False,
|
|
torch.int8: False,
|
|
torch.int16: False,
|
|
torch.int32: True,
|
|
torch.int64: False,
|
|
torch.bool: False,
|
|
},
|
|
torch.int64: {
|
|
torch.float16: False,
|
|
torch.float32: False,
|
|
torch.float64: False,
|
|
torch.complex64: False,
|
|
torch.complex128: False,
|
|
torch.uint8: False,
|
|
torch.uint16: False,
|
|
torch.uint32: False,
|
|
torch.uint64: False,
|
|
torch.int8: False,
|
|
torch.int16: False,
|
|
torch.int32: False,
|
|
torch.int64: True,
|
|
torch.bool: False,
|
|
},
|
|
torch.bool: {
|
|
torch.float16: False,
|
|
torch.float32: False,
|
|
torch.float64: False,
|
|
torch.complex64: False,
|
|
torch.complex128: False,
|
|
torch.uint8: False,
|
|
torch.uint16: False,
|
|
torch.uint32: False,
|
|
torch.uint64: False,
|
|
torch.int8: False,
|
|
torch.int16: False,
|
|
torch.int32: False,
|
|
torch.int64: False,
|
|
torch.bool: True,
|
|
},
|
|
},
|
|
"equiv": {
|
|
torch.float16: {
|
|
torch.float16: True,
|
|
torch.float32: False,
|
|
torch.float64: False,
|
|
torch.complex64: False,
|
|
torch.complex128: False,
|
|
torch.uint8: False,
|
|
torch.uint16: False,
|
|
torch.uint32: False,
|
|
torch.uint64: False,
|
|
torch.int8: False,
|
|
torch.int16: False,
|
|
torch.int32: False,
|
|
torch.int64: False,
|
|
torch.bool: False,
|
|
},
|
|
torch.float32: {
|
|
torch.float16: False,
|
|
torch.float32: True,
|
|
torch.float64: False,
|
|
torch.complex64: False,
|
|
torch.complex128: False,
|
|
torch.uint8: False,
|
|
torch.uint16: False,
|
|
torch.uint32: False,
|
|
torch.uint64: False,
|
|
torch.int8: False,
|
|
torch.int16: False,
|
|
torch.int32: False,
|
|
torch.int64: False,
|
|
torch.bool: False,
|
|
},
|
|
torch.float64: {
|
|
torch.float16: False,
|
|
torch.float32: False,
|
|
torch.float64: True,
|
|
torch.complex64: False,
|
|
torch.complex128: False,
|
|
torch.uint8: False,
|
|
torch.uint16: False,
|
|
torch.uint32: False,
|
|
torch.uint64: False,
|
|
torch.int8: False,
|
|
torch.int16: False,
|
|
torch.int32: False,
|
|
torch.int64: False,
|
|
torch.bool: False,
|
|
},
|
|
torch.complex64: {
|
|
torch.float16: False,
|
|
torch.float32: False,
|
|
torch.float64: False,
|
|
torch.complex64: True,
|
|
torch.complex128: False,
|
|
torch.uint8: False,
|
|
torch.uint16: False,
|
|
torch.uint32: False,
|
|
torch.uint64: False,
|
|
torch.int8: False,
|
|
torch.int16: False,
|
|
torch.int32: False,
|
|
torch.int64: False,
|
|
torch.bool: False,
|
|
},
|
|
torch.complex128: {
|
|
torch.float16: False,
|
|
torch.float32: False,
|
|
torch.float64: False,
|
|
torch.complex64: False,
|
|
torch.complex128: True,
|
|
torch.uint8: False,
|
|
torch.uint16: False,
|
|
torch.uint32: False,
|
|
torch.uint64: False,
|
|
torch.int8: False,
|
|
torch.int16: False,
|
|
torch.int32: False,
|
|
torch.int64: False,
|
|
torch.bool: False,
|
|
},
|
|
torch.uint8: {
|
|
torch.float16: False,
|
|
torch.float32: False,
|
|
torch.float64: False,
|
|
torch.complex64: False,
|
|
torch.complex128: False,
|
|
torch.uint8: True,
|
|
torch.uint16: False,
|
|
torch.uint32: False,
|
|
torch.uint64: False,
|
|
torch.int8: False,
|
|
torch.int16: False,
|
|
torch.int32: False,
|
|
torch.int64: False,
|
|
torch.bool: False,
|
|
},
|
|
torch.uint16: {
|
|
torch.float16: False,
|
|
torch.float32: False,
|
|
torch.float64: False,
|
|
torch.complex64: False,
|
|
torch.complex128: False,
|
|
torch.uint8: False,
|
|
torch.uint16: True,
|
|
torch.uint32: False,
|
|
torch.uint64: False,
|
|
torch.int8: False,
|
|
torch.int16: False,
|
|
torch.int32: False,
|
|
torch.int64: False,
|
|
torch.bool: False,
|
|
},
|
|
torch.uint32: {
|
|
torch.float16: False,
|
|
torch.float32: False,
|
|
torch.float64: False,
|
|
torch.complex64: False,
|
|
torch.complex128: False,
|
|
torch.uint8: False,
|
|
torch.uint16: False,
|
|
torch.uint32: True,
|
|
torch.uint64: False,
|
|
torch.int8: False,
|
|
torch.int16: False,
|
|
torch.int32: False,
|
|
torch.int64: False,
|
|
torch.bool: False,
|
|
},
|
|
torch.uint64: {
|
|
torch.float16: False,
|
|
torch.float32: False,
|
|
torch.float64: False,
|
|
torch.complex64: False,
|
|
torch.complex128: False,
|
|
torch.uint8: False,
|
|
torch.uint16: False,
|
|
torch.uint32: False,
|
|
torch.uint64: True,
|
|
torch.int8: False,
|
|
torch.int16: False,
|
|
torch.int32: False,
|
|
torch.int64: False,
|
|
torch.bool: False,
|
|
},
|
|
torch.int8: {
|
|
torch.float16: False,
|
|
torch.float32: False,
|
|
torch.float64: False,
|
|
torch.complex64: False,
|
|
torch.complex128: False,
|
|
torch.uint8: False,
|
|
torch.uint16: False,
|
|
torch.uint32: False,
|
|
torch.uint64: False,
|
|
torch.int8: True,
|
|
torch.int16: False,
|
|
torch.int32: False,
|
|
torch.int64: False,
|
|
torch.bool: False,
|
|
},
|
|
torch.int16: {
|
|
torch.float16: False,
|
|
torch.float32: False,
|
|
torch.float64: False,
|
|
torch.complex64: False,
|
|
torch.complex128: False,
|
|
torch.uint8: False,
|
|
torch.uint16: False,
|
|
torch.uint32: False,
|
|
torch.uint64: False,
|
|
torch.int8: False,
|
|
torch.int16: True,
|
|
torch.int32: False,
|
|
torch.int64: False,
|
|
torch.bool: False,
|
|
},
|
|
torch.int32: {
|
|
torch.float16: False,
|
|
torch.float32: False,
|
|
torch.float64: False,
|
|
torch.complex64: False,
|
|
torch.complex128: False,
|
|
torch.uint8: False,
|
|
torch.uint16: False,
|
|
torch.uint32: False,
|
|
torch.uint64: False,
|
|
torch.int8: False,
|
|
torch.int16: False,
|
|
torch.int32: True,
|
|
torch.int64: False,
|
|
torch.bool: False,
|
|
},
|
|
torch.int64: {
|
|
torch.float16: False,
|
|
torch.float32: False,
|
|
torch.float64: False,
|
|
torch.complex64: False,
|
|
torch.complex128: False,
|
|
torch.uint8: False,
|
|
torch.uint16: False,
|
|
torch.uint32: False,
|
|
torch.uint64: False,
|
|
torch.int8: False,
|
|
torch.int16: False,
|
|
torch.int32: False,
|
|
torch.int64: True,
|
|
torch.bool: False,
|
|
},
|
|
torch.bool: {
|
|
torch.float16: False,
|
|
torch.float32: False,
|
|
torch.float64: False,
|
|
torch.complex64: False,
|
|
torch.complex128: False,
|
|
torch.uint8: False,
|
|
torch.uint16: False,
|
|
torch.uint32: False,
|
|
torch.uint64: False,
|
|
torch.int8: False,
|
|
torch.int16: False,
|
|
torch.int32: False,
|
|
torch.int64: False,
|
|
torch.bool: True,
|
|
},
|
|
},
|
|
"safe": {
|
|
torch.float16: {
|
|
torch.float16: True,
|
|
torch.float32: True,
|
|
torch.float64: True,
|
|
torch.complex64: True,
|
|
torch.complex128: True,
|
|
torch.uint8: False,
|
|
torch.uint16: False,
|
|
torch.uint32: False,
|
|
torch.uint64: False,
|
|
torch.int8: False,
|
|
torch.int16: False,
|
|
torch.int32: False,
|
|
torch.int64: False,
|
|
torch.bool: False,
|
|
},
|
|
torch.float32: {
|
|
torch.float16: False,
|
|
torch.float32: True,
|
|
torch.float64: True,
|
|
torch.complex64: True,
|
|
torch.complex128: True,
|
|
torch.uint8: False,
|
|
torch.uint16: False,
|
|
torch.uint32: False,
|
|
torch.uint64: False,
|
|
torch.int8: False,
|
|
torch.int16: False,
|
|
torch.int32: False,
|
|
torch.int64: False,
|
|
torch.bool: False,
|
|
},
|
|
torch.float64: {
|
|
torch.float16: False,
|
|
torch.float32: False,
|
|
torch.float64: True,
|
|
torch.complex64: False,
|
|
torch.complex128: True,
|
|
torch.uint8: False,
|
|
torch.uint16: False,
|
|
torch.uint32: False,
|
|
torch.uint64: False,
|
|
torch.int8: False,
|
|
torch.int16: False,
|
|
torch.int32: False,
|
|
torch.int64: False,
|
|
torch.bool: False,
|
|
},
|
|
torch.complex64: {
|
|
torch.float16: False,
|
|
torch.float32: False,
|
|
torch.float64: False,
|
|
torch.complex64: True,
|
|
torch.complex128: True,
|
|
torch.uint8: False,
|
|
torch.uint16: False,
|
|
torch.uint32: False,
|
|
torch.uint64: False,
|
|
torch.int8: False,
|
|
torch.int16: False,
|
|
torch.int32: False,
|
|
torch.int64: False,
|
|
torch.bool: False,
|
|
},
|
|
torch.complex128: {
|
|
torch.float16: False,
|
|
torch.float32: False,
|
|
torch.float64: False,
|
|
torch.complex64: False,
|
|
torch.complex128: True,
|
|
torch.uint8: False,
|
|
torch.uint16: False,
|
|
torch.uint32: False,
|
|
torch.uint64: False,
|
|
torch.int8: False,
|
|
torch.int16: False,
|
|
torch.int32: False,
|
|
torch.int64: False,
|
|
torch.bool: False,
|
|
},
|
|
torch.uint8: {
|
|
torch.float16: True,
|
|
torch.float32: True,
|
|
torch.float64: True,
|
|
torch.complex64: True,
|
|
torch.complex128: True,
|
|
torch.uint8: True,
|
|
torch.uint16: True,
|
|
torch.uint32: True,
|
|
torch.uint64: True,
|
|
torch.int8: False,
|
|
torch.int16: True,
|
|
torch.int32: True,
|
|
torch.int64: True,
|
|
torch.bool: False,
|
|
},
|
|
torch.uint16: {
|
|
torch.float16: False,
|
|
torch.float32: True,
|
|
torch.float64: True,
|
|
torch.complex64: True,
|
|
torch.complex128: True,
|
|
torch.uint8: False,
|
|
torch.uint16: True,
|
|
torch.uint32: True,
|
|
torch.uint64: True,
|
|
torch.int8: False,
|
|
torch.int16: False,
|
|
torch.int32: True,
|
|
torch.int64: True,
|
|
torch.bool: False,
|
|
},
|
|
torch.uint32: {
|
|
torch.float16: False,
|
|
torch.float32: False,
|
|
torch.float64: True,
|
|
torch.complex64: False,
|
|
torch.complex128: True,
|
|
torch.uint8: False,
|
|
torch.uint16: False,
|
|
torch.uint32: True,
|
|
torch.uint64: True,
|
|
torch.int8: False,
|
|
torch.int16: False,
|
|
torch.int32: False,
|
|
torch.int64: True,
|
|
torch.bool: False,
|
|
},
|
|
torch.uint64: {
|
|
torch.float16: False,
|
|
torch.float32: False,
|
|
torch.float64: True,
|
|
torch.complex64: False,
|
|
torch.complex128: True,
|
|
torch.uint8: False,
|
|
torch.uint16: False,
|
|
torch.uint32: False,
|
|
torch.uint64: True,
|
|
torch.int8: False,
|
|
torch.int16: False,
|
|
torch.int32: False,
|
|
torch.int64: False,
|
|
torch.bool: False,
|
|
},
|
|
torch.int8: {
|
|
torch.float16: True,
|
|
torch.float32: True,
|
|
torch.float64: True,
|
|
torch.complex64: True,
|
|
torch.complex128: True,
|
|
torch.uint8: False,
|
|
torch.uint16: False,
|
|
torch.uint32: False,
|
|
torch.uint64: False,
|
|
torch.int8: True,
|
|
torch.int16: True,
|
|
torch.int32: True,
|
|
torch.int64: True,
|
|
torch.bool: False,
|
|
},
|
|
torch.int16: {
|
|
torch.float16: False,
|
|
torch.float32: True,
|
|
torch.float64: True,
|
|
torch.complex64: True,
|
|
torch.complex128: True,
|
|
torch.uint8: False,
|
|
torch.uint16: False,
|
|
torch.uint32: False,
|
|
torch.uint64: False,
|
|
torch.int8: False,
|
|
torch.int16: True,
|
|
torch.int32: True,
|
|
torch.int64: True,
|
|
torch.bool: False,
|
|
},
|
|
torch.int32: {
|
|
torch.float16: False,
|
|
torch.float32: False,
|
|
torch.float64: True,
|
|
torch.complex64: False,
|
|
torch.complex128: True,
|
|
torch.uint8: False,
|
|
torch.uint16: False,
|
|
torch.uint32: False,
|
|
torch.uint64: False,
|
|
torch.int8: False,
|
|
torch.int16: False,
|
|
torch.int32: True,
|
|
torch.int64: True,
|
|
torch.bool: False,
|
|
},
|
|
torch.int64: {
|
|
torch.float16: False,
|
|
torch.float32: False,
|
|
torch.float64: True,
|
|
torch.complex64: False,
|
|
torch.complex128: True,
|
|
torch.uint8: False,
|
|
torch.uint16: False,
|
|
torch.uint32: False,
|
|
torch.uint64: False,
|
|
torch.int8: False,
|
|
torch.int16: False,
|
|
torch.int32: False,
|
|
torch.int64: True,
|
|
torch.bool: False,
|
|
},
|
|
torch.bool: {
|
|
torch.float16: True,
|
|
torch.float32: True,
|
|
torch.float64: True,
|
|
torch.complex64: True,
|
|
torch.complex128: True,
|
|
torch.uint8: True,
|
|
torch.uint16: True,
|
|
torch.uint32: True,
|
|
torch.uint64: True,
|
|
torch.int8: True,
|
|
torch.int16: True,
|
|
torch.int32: True,
|
|
torch.int64: True,
|
|
torch.bool: True,
|
|
},
|
|
},
|
|
"same_kind": {
|
|
torch.float16: {
|
|
torch.float16: True,
|
|
torch.float32: True,
|
|
torch.float64: True,
|
|
torch.complex64: True,
|
|
torch.complex128: True,
|
|
torch.uint8: False,
|
|
torch.uint16: False,
|
|
torch.uint32: False,
|
|
torch.uint64: False,
|
|
torch.int8: False,
|
|
torch.int16: False,
|
|
torch.int32: False,
|
|
torch.int64: False,
|
|
torch.bool: False,
|
|
},
|
|
torch.float32: {
|
|
torch.float16: True,
|
|
torch.float32: True,
|
|
torch.float64: True,
|
|
torch.complex64: True,
|
|
torch.complex128: True,
|
|
torch.uint8: False,
|
|
torch.uint16: False,
|
|
torch.uint32: False,
|
|
torch.uint64: False,
|
|
torch.int8: False,
|
|
torch.int16: False,
|
|
torch.int32: False,
|
|
torch.int64: False,
|
|
torch.bool: False,
|
|
},
|
|
torch.float64: {
|
|
torch.float16: True,
|
|
torch.float32: True,
|
|
torch.float64: True,
|
|
torch.complex64: True,
|
|
torch.complex128: True,
|
|
torch.uint8: False,
|
|
torch.uint16: False,
|
|
torch.uint32: False,
|
|
torch.uint64: False,
|
|
torch.int8: False,
|
|
torch.int16: False,
|
|
torch.int32: False,
|
|
torch.int64: False,
|
|
torch.bool: False,
|
|
},
|
|
torch.complex64: {
|
|
torch.float16: False,
|
|
torch.float32: False,
|
|
torch.float64: False,
|
|
torch.complex64: True,
|
|
torch.complex128: True,
|
|
torch.uint8: False,
|
|
torch.uint16: False,
|
|
torch.uint32: False,
|
|
torch.uint64: False,
|
|
torch.int8: False,
|
|
torch.int16: False,
|
|
torch.int32: False,
|
|
torch.int64: False,
|
|
torch.bool: False,
|
|
},
|
|
torch.complex128: {
|
|
torch.float16: False,
|
|
torch.float32: False,
|
|
torch.float64: False,
|
|
torch.complex64: True,
|
|
torch.complex128: True,
|
|
torch.uint8: False,
|
|
torch.uint16: False,
|
|
torch.uint32: False,
|
|
torch.uint64: False,
|
|
torch.int8: False,
|
|
torch.int16: False,
|
|
torch.int32: False,
|
|
torch.int64: False,
|
|
torch.bool: False,
|
|
},
|
|
torch.uint8: {
|
|
torch.float16: True,
|
|
torch.float32: True,
|
|
torch.float64: True,
|
|
torch.complex64: True,
|
|
torch.complex128: True,
|
|
torch.uint8: True,
|
|
torch.uint16: True,
|
|
torch.uint32: True,
|
|
torch.uint64: True,
|
|
torch.int8: True,
|
|
torch.int16: True,
|
|
torch.int32: True,
|
|
torch.int64: True,
|
|
torch.bool: False,
|
|
},
|
|
torch.uint16: {
|
|
torch.float16: True,
|
|
torch.float32: True,
|
|
torch.float64: True,
|
|
torch.complex64: True,
|
|
torch.complex128: True,
|
|
torch.uint8: True,
|
|
torch.uint16: True,
|
|
torch.uint32: True,
|
|
torch.uint64: True,
|
|
torch.int8: True,
|
|
torch.int16: True,
|
|
torch.int32: True,
|
|
torch.int64: True,
|
|
torch.bool: False,
|
|
},
|
|
torch.uint32: {
|
|
torch.float16: True,
|
|
torch.float32: True,
|
|
torch.float64: True,
|
|
torch.complex64: True,
|
|
torch.complex128: True,
|
|
torch.uint8: True,
|
|
torch.uint16: True,
|
|
torch.uint32: True,
|
|
torch.uint64: True,
|
|
torch.int8: True,
|
|
torch.int16: True,
|
|
torch.int32: True,
|
|
torch.int64: True,
|
|
torch.bool: False,
|
|
},
|
|
torch.uint64: {
|
|
torch.float16: True,
|
|
torch.float32: True,
|
|
torch.float64: True,
|
|
torch.complex64: True,
|
|
torch.complex128: True,
|
|
torch.uint8: True,
|
|
torch.uint16: True,
|
|
torch.uint32: True,
|
|
torch.uint64: True,
|
|
torch.int8: True,
|
|
torch.int16: True,
|
|
torch.int32: True,
|
|
torch.int64: True,
|
|
torch.bool: False,
|
|
},
|
|
torch.int8: {
|
|
torch.float16: True,
|
|
torch.float32: True,
|
|
torch.float64: True,
|
|
torch.complex64: True,
|
|
torch.complex128: True,
|
|
torch.uint8: False,
|
|
torch.uint16: False,
|
|
torch.uint32: False,
|
|
torch.uint64: False,
|
|
torch.int8: True,
|
|
torch.int16: True,
|
|
torch.int32: True,
|
|
torch.int64: True,
|
|
torch.bool: False,
|
|
},
|
|
torch.int16: {
|
|
torch.float16: True,
|
|
torch.float32: True,
|
|
torch.float64: True,
|
|
torch.complex64: True,
|
|
torch.complex128: True,
|
|
torch.uint8: False,
|
|
torch.uint16: False,
|
|
torch.uint32: False,
|
|
torch.uint64: False,
|
|
torch.int8: True,
|
|
torch.int16: True,
|
|
torch.int32: True,
|
|
torch.int64: True,
|
|
torch.bool: False,
|
|
},
|
|
torch.int32: {
|
|
torch.float16: True,
|
|
torch.float32: True,
|
|
torch.float64: True,
|
|
torch.complex64: True,
|
|
torch.complex128: True,
|
|
torch.uint8: False,
|
|
torch.uint16: False,
|
|
torch.uint32: False,
|
|
torch.uint64: False,
|
|
torch.int8: True,
|
|
torch.int16: True,
|
|
torch.int32: True,
|
|
torch.int64: True,
|
|
torch.bool: False,
|
|
},
|
|
torch.int64: {
|
|
torch.float16: True,
|
|
torch.float32: True,
|
|
torch.float64: True,
|
|
torch.complex64: True,
|
|
torch.complex128: True,
|
|
torch.uint8: False,
|
|
torch.uint16: False,
|
|
torch.uint32: False,
|
|
torch.uint64: False,
|
|
torch.int8: True,
|
|
torch.int16: True,
|
|
torch.int32: True,
|
|
torch.int64: True,
|
|
torch.bool: False,
|
|
},
|
|
torch.bool: {
|
|
torch.float16: True,
|
|
torch.float32: True,
|
|
torch.float64: True,
|
|
torch.complex64: True,
|
|
torch.complex128: True,
|
|
torch.uint8: True,
|
|
torch.uint16: True,
|
|
torch.uint32: True,
|
|
torch.uint64: True,
|
|
torch.int8: True,
|
|
torch.int16: True,
|
|
torch.int32: True,
|
|
torch.int64: True,
|
|
torch.bool: True,
|
|
},
|
|
},
|
|
"unsafe": {
|
|
torch.float16: {
|
|
torch.float16: True,
|
|
torch.float32: True,
|
|
torch.float64: True,
|
|
torch.complex64: True,
|
|
torch.complex128: True,
|
|
torch.uint8: True,
|
|
torch.uint16: True,
|
|
torch.uint32: True,
|
|
torch.uint64: True,
|
|
torch.int8: True,
|
|
torch.int16: True,
|
|
torch.int32: True,
|
|
torch.int64: True,
|
|
torch.bool: True,
|
|
},
|
|
torch.float32: {
|
|
torch.float16: True,
|
|
torch.float32: True,
|
|
torch.float64: True,
|
|
torch.complex64: True,
|
|
torch.complex128: True,
|
|
torch.uint8: True,
|
|
torch.uint16: True,
|
|
torch.uint32: True,
|
|
torch.uint64: True,
|
|
torch.int8: True,
|
|
torch.int16: True,
|
|
torch.int32: True,
|
|
torch.int64: True,
|
|
torch.bool: True,
|
|
},
|
|
torch.float64: {
|
|
torch.float16: True,
|
|
torch.float32: True,
|
|
torch.float64: True,
|
|
torch.complex64: True,
|
|
torch.complex128: True,
|
|
torch.uint8: True,
|
|
torch.uint16: True,
|
|
torch.uint32: True,
|
|
torch.uint64: True,
|
|
torch.int8: True,
|
|
torch.int16: True,
|
|
torch.int32: True,
|
|
torch.int64: True,
|
|
torch.bool: True,
|
|
},
|
|
torch.complex64: {
|
|
torch.float16: True,
|
|
torch.float32: True,
|
|
torch.float64: True,
|
|
torch.complex64: True,
|
|
torch.complex128: True,
|
|
torch.uint8: True,
|
|
torch.uint16: True,
|
|
torch.uint32: True,
|
|
torch.uint64: True,
|
|
torch.int8: True,
|
|
torch.int16: True,
|
|
torch.int32: True,
|
|
torch.int64: True,
|
|
torch.bool: True,
|
|
},
|
|
torch.complex128: {
|
|
torch.float16: True,
|
|
torch.float32: True,
|
|
torch.float64: True,
|
|
torch.complex64: True,
|
|
torch.complex128: True,
|
|
torch.uint8: True,
|
|
torch.uint16: True,
|
|
torch.uint32: True,
|
|
torch.uint64: True,
|
|
torch.int8: True,
|
|
torch.int16: True,
|
|
torch.int32: True,
|
|
torch.int64: True,
|
|
torch.bool: True,
|
|
},
|
|
torch.uint8: {
|
|
torch.float16: True,
|
|
torch.float32: True,
|
|
torch.float64: True,
|
|
torch.complex64: True,
|
|
torch.complex128: True,
|
|
torch.uint8: True,
|
|
torch.uint16: True,
|
|
torch.uint32: True,
|
|
torch.uint64: True,
|
|
torch.int8: True,
|
|
torch.int16: True,
|
|
torch.int32: True,
|
|
torch.int64: True,
|
|
torch.bool: True,
|
|
},
|
|
torch.uint16: {
|
|
torch.float16: True,
|
|
torch.float32: True,
|
|
torch.float64: True,
|
|
torch.complex64: True,
|
|
torch.complex128: True,
|
|
torch.uint8: True,
|
|
torch.uint16: True,
|
|
torch.uint32: True,
|
|
torch.uint64: True,
|
|
torch.int8: True,
|
|
torch.int16: True,
|
|
torch.int32: True,
|
|
torch.int64: True,
|
|
torch.bool: True,
|
|
},
|
|
torch.uint32: {
|
|
torch.float16: True,
|
|
torch.float32: True,
|
|
torch.float64: True,
|
|
torch.complex64: True,
|
|
torch.complex128: True,
|
|
torch.uint8: True,
|
|
torch.uint16: True,
|
|
torch.uint32: True,
|
|
torch.uint64: True,
|
|
torch.int8: True,
|
|
torch.int16: True,
|
|
torch.int32: True,
|
|
torch.int64: True,
|
|
torch.bool: True,
|
|
},
|
|
torch.uint64: {
|
|
torch.float16: True,
|
|
torch.float32: True,
|
|
torch.float64: True,
|
|
torch.complex64: True,
|
|
torch.complex128: True,
|
|
torch.uint8: True,
|
|
torch.uint16: True,
|
|
torch.uint32: True,
|
|
torch.uint64: True,
|
|
torch.int8: True,
|
|
torch.int16: True,
|
|
torch.int32: True,
|
|
torch.int64: True,
|
|
torch.bool: True,
|
|
},
|
|
torch.int8: {
|
|
torch.float16: True,
|
|
torch.float32: True,
|
|
torch.float64: True,
|
|
torch.complex64: True,
|
|
torch.complex128: True,
|
|
torch.uint8: True,
|
|
torch.uint16: True,
|
|
torch.uint32: True,
|
|
torch.uint64: True,
|
|
torch.int8: True,
|
|
torch.int16: True,
|
|
torch.int32: True,
|
|
torch.int64: True,
|
|
torch.bool: True,
|
|
},
|
|
torch.int16: {
|
|
torch.float16: True,
|
|
torch.float32: True,
|
|
torch.float64: True,
|
|
torch.complex64: True,
|
|
torch.complex128: True,
|
|
torch.uint8: True,
|
|
torch.uint16: True,
|
|
torch.uint32: True,
|
|
torch.uint64: True,
|
|
torch.int8: True,
|
|
torch.int16: True,
|
|
torch.int32: True,
|
|
torch.int64: True,
|
|
torch.bool: True,
|
|
},
|
|
torch.int32: {
|
|
torch.float16: True,
|
|
torch.float32: True,
|
|
torch.float64: True,
|
|
torch.complex64: True,
|
|
torch.complex128: True,
|
|
torch.uint8: True,
|
|
torch.uint16: True,
|
|
torch.uint32: True,
|
|
torch.uint64: True,
|
|
torch.int8: True,
|
|
torch.int16: True,
|
|
torch.int32: True,
|
|
torch.int64: True,
|
|
torch.bool: True,
|
|
},
|
|
torch.int64: {
|
|
torch.float16: True,
|
|
torch.float32: True,
|
|
torch.float64: True,
|
|
torch.complex64: True,
|
|
torch.complex128: True,
|
|
torch.uint8: True,
|
|
torch.uint16: True,
|
|
torch.uint32: True,
|
|
torch.uint64: True,
|
|
torch.int8: True,
|
|
torch.int16: True,
|
|
torch.int32: True,
|
|
torch.int64: True,
|
|
torch.bool: True,
|
|
},
|
|
torch.bool: {
|
|
torch.float16: True,
|
|
torch.float32: True,
|
|
torch.float64: True,
|
|
torch.complex64: True,
|
|
torch.complex128: True,
|
|
torch.uint8: True,
|
|
torch.uint16: True,
|
|
torch.uint32: True,
|
|
torch.uint64: True,
|
|
torch.int8: True,
|
|
torch.int16: True,
|
|
torch.int32: True,
|
|
torch.int64: True,
|
|
torch.bool: True,
|
|
},
|
|
},
|
|
}
|
|
|
|
|
|
_result_type_dict = {
|
|
torch.float16: {
|
|
torch.float16: torch.float16,
|
|
torch.float32: torch.float32,
|
|
torch.float64: torch.float64,
|
|
torch.complex64: torch.complex64,
|
|
torch.complex128: torch.complex128,
|
|
torch.uint8: torch.float16,
|
|
torch.uint16: torch.float32,
|
|
torch.uint32: torch.float64,
|
|
torch.uint64: torch.float64,
|
|
torch.int8: torch.float16,
|
|
torch.int16: torch.float32,
|
|
torch.int32: torch.float64,
|
|
torch.int64: torch.float64,
|
|
torch.bool: torch.float16,
|
|
},
|
|
torch.float32: {
|
|
torch.float16: torch.float32,
|
|
torch.float32: torch.float32,
|
|
torch.float64: torch.float64,
|
|
torch.complex64: torch.complex64,
|
|
torch.complex128: torch.complex128,
|
|
torch.uint8: torch.float32,
|
|
torch.uint16: torch.float32,
|
|
torch.uint32: torch.float64,
|
|
torch.uint64: torch.float64,
|
|
torch.int8: torch.float32,
|
|
torch.int16: torch.float32,
|
|
torch.int32: torch.float64,
|
|
torch.int64: torch.float64,
|
|
torch.bool: torch.float32,
|
|
},
|
|
torch.float64: {
|
|
torch.float16: torch.float64,
|
|
torch.float32: torch.float64,
|
|
torch.float64: torch.float64,
|
|
torch.complex64: torch.complex128,
|
|
torch.complex128: torch.complex128,
|
|
torch.uint8: torch.float64,
|
|
torch.uint16: torch.float64,
|
|
torch.uint32: torch.float64,
|
|
torch.uint64: torch.float64,
|
|
torch.int8: torch.float64,
|
|
torch.int16: torch.float64,
|
|
torch.int32: torch.float64,
|
|
torch.int64: torch.float64,
|
|
torch.bool: torch.float64,
|
|
},
|
|
torch.complex64: {
|
|
torch.float16: torch.complex64,
|
|
torch.float32: torch.complex64,
|
|
torch.float64: torch.complex128,
|
|
torch.complex64: torch.complex64,
|
|
torch.complex128: torch.complex128,
|
|
torch.uint8: torch.complex64,
|
|
torch.uint16: torch.complex64,
|
|
torch.uint32: torch.complex128,
|
|
torch.uint64: torch.complex128,
|
|
torch.int8: torch.complex64,
|
|
torch.int16: torch.complex64,
|
|
torch.int32: torch.complex128,
|
|
torch.int64: torch.complex128,
|
|
torch.bool: torch.complex64,
|
|
},
|
|
torch.complex128: {
|
|
torch.float16: torch.complex128,
|
|
torch.float32: torch.complex128,
|
|
torch.float64: torch.complex128,
|
|
torch.complex64: torch.complex128,
|
|
torch.complex128: torch.complex128,
|
|
torch.uint8: torch.complex128,
|
|
torch.uint16: torch.complex128,
|
|
torch.uint32: torch.complex128,
|
|
torch.uint64: torch.complex128,
|
|
torch.int8: torch.complex128,
|
|
torch.int16: torch.complex128,
|
|
torch.int32: torch.complex128,
|
|
torch.int64: torch.complex128,
|
|
torch.bool: torch.complex128,
|
|
},
|
|
torch.uint8: {
|
|
torch.float16: torch.float16,
|
|
torch.float32: torch.float32,
|
|
torch.float64: torch.float64,
|
|
torch.complex64: torch.complex64,
|
|
torch.complex128: torch.complex128,
|
|
torch.uint8: torch.uint8,
|
|
torch.uint16: torch.uint16,
|
|
torch.uint32: torch.uint32,
|
|
torch.uint64: torch.uint64,
|
|
torch.int8: torch.int16,
|
|
torch.int16: torch.int16,
|
|
torch.int32: torch.int32,
|
|
torch.int64: torch.int64,
|
|
torch.bool: torch.uint8,
|
|
},
|
|
torch.uint16: {
|
|
torch.float16: torch.float32,
|
|
torch.float32: torch.float32,
|
|
torch.float64: torch.float64,
|
|
torch.complex64: torch.complex64,
|
|
torch.complex128: torch.complex128,
|
|
torch.uint8: torch.uint16,
|
|
torch.uint16: torch.uint16,
|
|
torch.uint32: torch.uint32,
|
|
torch.uint64: torch.uint64,
|
|
torch.int8: torch.int32,
|
|
torch.int16: torch.int32,
|
|
torch.int32: torch.int32,
|
|
torch.int64: torch.int64,
|
|
torch.bool: torch.uint16,
|
|
},
|
|
torch.uint32: {
|
|
torch.float16: torch.float64,
|
|
torch.float32: torch.float64,
|
|
torch.float64: torch.float64,
|
|
torch.complex64: torch.complex128,
|
|
torch.complex128: torch.complex128,
|
|
torch.uint8: torch.uint32,
|
|
torch.uint16: torch.uint32,
|
|
torch.uint32: torch.uint32,
|
|
torch.uint64: torch.uint64,
|
|
torch.int8: torch.int64,
|
|
torch.int16: torch.int64,
|
|
torch.int32: torch.int64,
|
|
torch.int64: torch.int64,
|
|
torch.bool: torch.uint32,
|
|
},
|
|
torch.uint64: {
|
|
torch.float16: torch.float64,
|
|
torch.float32: torch.float64,
|
|
torch.float64: torch.float64,
|
|
torch.complex64: torch.complex128,
|
|
torch.complex128: torch.complex128,
|
|
torch.uint8: torch.uint64,
|
|
torch.uint16: torch.uint64,
|
|
torch.uint32: torch.uint64,
|
|
torch.uint64: torch.uint64,
|
|
torch.int8: torch.float64,
|
|
torch.int16: torch.float64,
|
|
torch.int32: torch.float64,
|
|
torch.int64: torch.float64,
|
|
torch.bool: torch.uint64,
|
|
},
|
|
torch.int8: {
|
|
torch.float16: torch.float16,
|
|
torch.float32: torch.float32,
|
|
torch.float64: torch.float64,
|
|
torch.complex64: torch.complex64,
|
|
torch.complex128: torch.complex128,
|
|
torch.uint8: torch.int16,
|
|
torch.uint16: torch.int32,
|
|
torch.uint32: torch.int64,
|
|
torch.uint64: torch.float64,
|
|
torch.int8: torch.int8,
|
|
torch.int16: torch.int16,
|
|
torch.int32: torch.int32,
|
|
torch.int64: torch.int64,
|
|
torch.bool: torch.int8,
|
|
},
|
|
torch.int16: {
|
|
torch.float16: torch.float32,
|
|
torch.float32: torch.float32,
|
|
torch.float64: torch.float64,
|
|
torch.complex64: torch.complex64,
|
|
torch.complex128: torch.complex128,
|
|
torch.uint8: torch.int16,
|
|
torch.uint16: torch.int32,
|
|
torch.uint32: torch.int64,
|
|
torch.uint64: torch.float64,
|
|
torch.int8: torch.int16,
|
|
torch.int16: torch.int16,
|
|
torch.int32: torch.int32,
|
|
torch.int64: torch.int64,
|
|
torch.bool: torch.int16,
|
|
},
|
|
torch.int32: {
|
|
torch.float16: torch.float64,
|
|
torch.float32: torch.float64,
|
|
torch.float64: torch.float64,
|
|
torch.complex64: torch.complex128,
|
|
torch.complex128: torch.complex128,
|
|
torch.uint8: torch.int32,
|
|
torch.uint16: torch.int32,
|
|
torch.uint32: torch.int64,
|
|
torch.uint64: torch.float64,
|
|
torch.int8: torch.int32,
|
|
torch.int16: torch.int32,
|
|
torch.int32: torch.int32,
|
|
torch.int64: torch.int64,
|
|
torch.bool: torch.int32,
|
|
},
|
|
torch.int64: {
|
|
torch.float16: torch.float64,
|
|
torch.float32: torch.float64,
|
|
torch.float64: torch.float64,
|
|
torch.complex64: torch.complex128,
|
|
torch.complex128: torch.complex128,
|
|
torch.uint8: torch.int64,
|
|
torch.uint16: torch.int64,
|
|
torch.uint32: torch.int64,
|
|
torch.uint64: torch.float64,
|
|
torch.int8: torch.int64,
|
|
torch.int16: torch.int64,
|
|
torch.int32: torch.int64,
|
|
torch.int64: torch.int64,
|
|
torch.bool: torch.int64,
|
|
},
|
|
torch.bool: {
|
|
torch.float16: torch.float16,
|
|
torch.float32: torch.float32,
|
|
torch.float64: torch.float64,
|
|
torch.complex64: torch.complex64,
|
|
torch.complex128: torch.complex128,
|
|
torch.uint8: torch.uint8,
|
|
torch.uint16: torch.uint16,
|
|
torch.uint32: torch.uint32,
|
|
torch.uint64: torch.uint64,
|
|
torch.int8: torch.int8,
|
|
torch.int16: torch.int16,
|
|
torch.int32: torch.int32,
|
|
torch.int64: torch.int64,
|
|
torch.bool: torch.bool,
|
|
},
|
|
}
|