Nikita Shulga 13966d0bf5 [BE] Migrate dtype_abbrs into one location (#152229) ...

Namely `torch.utils._dtype_abbrs.dtype_abbrs`

Before that it was defined in various forms of completeness in
c02edba863/torch/fx/graph.py (L215),
c02edba863/torch/testing/_internal/common_utils.py (L5226)
 and c02edba863/torch/testing/_internal/logging_tensor.py (L17)

TODO:
 - Add linter that `torch.testing._internal` module is not referenced from any of the public facing APIs, as it can have extra dependencies such as `expect_test`

Fixes https://github.com/pytorch/pytorch/issues/152225

Pull Request resolved: https://github.com/pytorch/pytorch/pull/152229
Approved by: https://github.com/clee2000, https://github.com/Skylion007

2025-04-28 03:52:47 +00:00

..

_strobelight

[BE]: Enable ruff rule SIM113 (#147290)

2025-02-16 22:41:16 +00:00

_sympy

Fix broken URLs (#152237)

2025-04-27 09:56:42 +00:00

backcompat

Flip default value for mypy disallow_untyped_defs [9/11] (#127846)

2024-06-08 18:50:06 +00:00

benchmark

[BE][Ez]: Use itertools.chain.from_iterable when possible (#148190)

2025-03-06 20:37:06 +00:00

bottleneck

PEP585 update - torch/utils (#145201)

2025-01-21 21:04:10 +00:00

data

Optimize dataloader Self typing (#146816)

2025-04-08 03:52:23 +00:00

hipify

Fix broken URLs (#152237)

2025-04-27 09:56:42 +00:00

jit

PEP585 update - torch/utils (#145201)

2025-01-21 21:04:10 +00:00

model_dump

PEP585: More UP006 fixes (#146392)

2025-02-20 06:18:13 +00:00

serialization

Make record/storage alignment in torch.save configurable (#147788)

2025-03-06 12:04:46 +00:00

tensorboard

Fix broken URLs (#152237)

2025-04-27 09:56:42 +00:00

viz

Fix ReferenceError: weakly-referenced object no longer exists in cycle detector (#146922)

2025-02-24 22:27:39 +00:00

__init__.py

[BE] enable UFMT in torch.utils.data (#127705)

2024-06-27 23:16:24 +00:00

_appending_byte_serializer.py

Check integrity of bytes in AppendingByteSerializer (#152139)

2025-04-26 18:10:58 +00:00

_backport_slots.py

PEP585 update - torch/utils (#145201)

2025-01-21 21:04:10 +00:00

_config_module.py

[dynamo] context manager/decorator for dynamo config patching during tracing (#150586)

2025-04-23 09:12:13 +00:00

_config_typing.pyi

Use Generic TypeAlias (PEP 585) and Union Type (PEP 604) in .pyi stub files (#129419)

2024-06-29 09:23:39 +00:00

_content_store.py

Revert "Use the device interface for detecting Triton availability (#139171)"

2025-03-11 18:49:21 +00:00

_contextlib.py

Flip default value for mypy disallow_untyped_defs [9/11] (#127846)

2024-06-08 18:50:06 +00:00

_cpp_embed_headers.py

[BE] Strip #pragma once when embedding the headers (#146871)

2025-02-11 16:49:00 +00:00

_cpp_extension_versioner.py

xpu: support sycl with torch.utils.cpp_extension APIs (#132945)

2025-02-16 16:50:59 +00:00

_cxx_pytree.py

Gracefully handle optree less than minimum version, part 2 (#151257)

2025-04-15 13:08:26 +00:00

_device.py

Remove torch functions that do not support device arguments from _device_constructor (#150290)

2025-04-08 15:13:55 +00:00

_dtype_abbrs.py

[BE] Migrate dtype_abbrs into one location (#152229)

2025-04-28 03:52:47 +00:00

_exposed_in.py

Type exposed_in decorator (#141894)

2024-12-03 16:28:17 +00:00

_filelock.py

filelock: Make waitcounter variant to use (#139816)

2024-12-12 01:18:34 +00:00

_foreach_utils.py

[HPU] Add hpu to fused kernels supported devices (#148666)

2025-03-07 04:28:33 +00:00

_freeze.py

PEP585 update - torch/utils (#145201)

2025-01-21 21:04:10 +00:00

_functools.py

dynamo tracing perf: cache on import_source: 52.9 -> 52.58 (#143058)

2024-12-13 18:20:48 +00:00

_get_clean_triton.py

Reland: [inductor] Simplify grid handling (#148305)

2025-03-12 15:52:16 +00:00

_import_utils.py

remove allow-untyped-defs from utils/_import_utils.py (#144089)

2025-01-03 01:21:13 +00:00

_mode_utils.py

Flip default value for mypy disallow_untyped_defs [9/11] (#127846)

2024-06-08 18:50:06 +00:00

_ordered_set.py

[BE]: Make OrderedSet reversible (#146904)

2025-02-13 15:11:48 +00:00

_python_dispatch.py

PEP585 update - torch/utils (#145201)

2025-01-21 21:04:10 +00:00

_pytree.py

Gracefully handle optree less than minimum version, part 2 (#151257)

2025-04-15 13:08:26 +00:00

_stats.py

PEP585 update - torch/utils (#145201)

2025-01-21 21:04:10 +00:00

_thunk.py

Refactor thunkify to return proper thunk abstraction (#132407)

2024-08-06 02:35:45 +00:00

_traceback.py

PEP585 update - torch/utils (#145201)

2025-01-21 21:04:10 +00:00

_triton.py

[Inductor] Remove triton dtype patch which has landed (#149611)

2025-04-10 03:42:55 +00:00

_typing_utils.py

Revert "Fix type annotation of Linear.bias (#142326)"

2025-01-26 03:41:00 +00:00

_zip.py

Flip default value for mypy disallow_untyped_defs [9/11] (#127846)

2024-06-08 18:50:06 +00:00

backend_registration.py

PEP585 update - torch/utils (#145201)

2025-01-21 21:04:10 +00:00

bundled_inputs.py

PEP585 update - torch/utils (#145201)

2025-01-21 21:04:10 +00:00

checkpoint.py

[BE] typing for decorators (#144161)

2025-01-04 16:40:09 +00:00

collect_env.py

Fix broken URLs (#152237)

2025-04-27 09:56:42 +00:00

cpp_backtrace.py

Flip default value for mypy disallow_untyped_defs [9/11] (#127846)

2024-06-08 18:50:06 +00:00

cpp_extension.py

[xpu] set aot device flags in cpp_extension (#149459)

2025-04-24 22:55:52 +00:00

deterministic.py

Flip default value for mypy disallow_untyped_defs [9/11] (#127846)

2024-06-08 18:50:06 +00:00

dlpack.py

Add __all__ for torch.utils.dlpack (#149026)

2025-04-11 22:03:24 +00:00

file_baton.py

Warn user of existing lock file to avoid infinite waiting (#149382)

2025-04-15 20:25:29 +00:00

flop_counter.py

[NJT] fix flop counter for SDPA & test (#147032)

2025-02-13 07:14:58 +00:00

hooks.py

PEP585 update - torch/utils (#145201)

2025-01-21 21:04:10 +00:00

mkldnn.py

Flip default value for mypy disallow_untyped_defs [10/11] (#127847)

2024-06-08 18:50:06 +00:00

mobile_optimizer.py

PEP585 update - torch/utils (#145201)

2025-01-21 21:04:10 +00:00

model_zoo.py

…

module_tracker.py

PEP585 update - torch/utils (#145201)

2025-01-21 21:04:10 +00:00

show_pickle.py

Use typing.IO[bytes] instead of io.BytesIO in annotations (#144994)

2025-01-27 18:08:07 +00:00

throughput_benchmark.py

Flip default value for mypy disallow_untyped_defs [10/11] (#127847)

2024-06-08 18:50:06 +00:00

weak.py

[BE]: Update mypy to 1.11.2 (#133816)

2024-09-16 19:44:11 +00:00