mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
a9dca53438
RFC: https://github.com/pytorch/rfcs/pull/54 First commit is the contents of https://github.com/Quansight-Labs/numpy_pytorch_interop/ We have already been using this in core for the last few months as a external dependency. This PR pulls all these into core. In the next commits, I do a number of things in this order - Fix a few small issues - Make the tests that this PR adds pass - Bend backwards until lintrunner passes - Remove the optional dependency on `torch_np` and simply rely on the upstreamed code - Fix a number dynamo tests that were passing before (they were not tasting anything I think) and are not passing now. Missing from this PR (but not blocking): - Have a flag that deactivates tracing NumPy functions and simply breaks. There used to be one but after the merge stopped working and I removed it. @lezcano to investigate. - https://github.com/pytorch/pytorch/pull/106431#issuecomment-1667079543. @voznesenskym to submit a fix after we merge. All the tests in `tests/torch_np` take about 75s to run. This was a work by @ev-br, @rgommers @honno and I. I did not create this PR via ghstack (which would have been convenient) as this is a collaboration, and ghstack doesn't allow for shared contributions. Pull Request resolved: https://github.com/pytorch/pytorch/pull/106211 Approved by: https://github.com/ezyang
880 lines
26 KiB
Python
880 lines
26 KiB
Python
import torch
|
|
|
|
# These two dicts are autogenerated with autogen/gen_dtypes.py,
|
|
# using numpy version 1.23.5.
|
|
|
|
_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.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.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.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.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.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.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.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.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.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.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.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.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.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.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.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.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.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.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.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.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.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.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.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.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.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.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.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.int8: False,
|
|
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.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.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.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.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.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.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.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.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.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.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.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.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.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.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.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.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.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.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.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.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.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.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.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.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.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.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.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.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.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.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.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.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.int8: torch.int16,
|
|
torch.int16: torch.int16,
|
|
torch.int32: torch.int32,
|
|
torch.int64: torch.int64,
|
|
torch.bool: torch.uint8,
|
|
},
|
|
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.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.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.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.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.int8: torch.int8,
|
|
torch.int16: torch.int16,
|
|
torch.int32: torch.int32,
|
|
torch.int64: torch.int64,
|
|
torch.bool: torch.bool,
|
|
},
|
|
}
|