OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 00:21:07 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
c869862875
pytorch/torch/utils
History
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
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
_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
_content_store.py Revert "Use the device interface for detecting Triton availability (#139171)" 2025-03-11 18:49:21 +00:00
_contextlib.py
_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
_filelock.py
_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
_get_clean_triton.py Reland: [inductor] Simplify grid handling (#148305) 2025-03-12 15:52:16 +00:00
_import_utils.py
_mode_utils.py
_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
_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
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
collect_env.py Fix broken URLs (#152237) 2025-04-27 09:56:42 +00:00
cpp_backtrace.py
cpp_extension.py [xpu] set aot device flags in cpp_extension (#149459) 2025-04-24 22:55:52 +00:00
deterministic.py
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
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
weak.py