mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-21 13:44:15 +08:00
9bce208dfb
This is a lot of files changed! Don't panic! Here's how it works: * Previously, we set `follow_imports = silent` for our mypy.ini configuration. Per https://mypy.readthedocs.io/en/stable/running_mypy.html#follow-imports, what this does is whenever we have an import to a module which is not listed as a file to be typechecked in mypy, we typecheck it as normal but suppress all errors that occurred in that file. * When mypy is run inside lintrunner, the list of files is precisely the files covered by the glob in lintrunner.toml, but with files in excludes excluded. * The top-level directive `# mypy: ignore-errors` instructs mypy to typecheck the file as normal, but ignore all errors. * Therefore, it should be equivalent to set `follow_imports = normal`, if we put `# mypy: ignore-errors` on all files that were previously excluded from the file list. * Having done this, we can remove the exclude list from .lintrunner.toml, since excluding a file from typechecking is baked into the files themselves. * torch/_dynamo and torch/_inductor were previously in the exclude list, because they were covered by MYPYINDUCTOR. It is not OK to mark these as `# mypy: ignore-errors` as this will impede typechecking on the alternate configuration. So they are temporarily being checked twice, but I am suppressing the errors in these files as the configurations are not quite the same. I plan to unify the configurations so this is only a temporary state. * There were some straggler type errors after these changes somehow, so I fixed them as needed. There weren't that many. In the future, to start type checking a file, just remove the ignore-errors directive from the top of the file. The codemod was done with this script authored by GPT-4: ``` import glob exclude_patterns = [ ... ] for pattern in exclude_patterns: for filepath in glob.glob(pattern, recursive=True): if filepath.endswith('.py'): with open(filepath, 'r+') as f: content = f.read() f.seek(0, 0) f.write('# mypy: ignore-errors\n\n' + content) ``` Signed-off-by: Edward Z. Yang <ezyang@meta.com> Pull Request resolved: https://github.com/pytorch/pytorch/pull/118414 Approved by: https://github.com/thiagocrepaldi, https://github.com/albanD
882 lines
26 KiB
Python
882 lines
26 KiB
Python
# mypy: ignore-errors
|
|
|
|
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,
|
|
},
|
|
}
|