pytorch/torch/testing/_internal
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
..
codegen
data
distributed Fix DTensorTestBase to barrier with device ids (#150896) 2025-04-22 20:22:55 +00:00
generated
opinfo [BE] Switch TestConsistency to MPS device (#147893) 2025-04-26 01:19:21 +00:00
optests
test_module
__init__.py
autocast_test_lists.py
autograd_function_db.py
check_kernel_launches.py
common_cuda.py [ROCm] opportunistic fastatomics for ReduceAdd operations for MI300 GPUs (#146264) 2025-04-22 21:55:40 +00:00
common_device_type.py Fix setUpClass() / tearDownClass() for device-specific tests (#151129) 2025-04-16 02:18:42 +00:00
common_dist_composable.py
common_distributed.py Reapply "ProcessGroupGloo: support lazy_init (#150801)" (#151031) 2025-04-11 01:58:35 +00:00
common_dtype.py
common_fsdp.py
common_jit.py
common_methods_invocations.py Fix broken URLs (#152237) 2025-04-27 09:56:42 +00:00
common_mkldnn.py [BE]: Apply ruff PERF403 to use dict comprehensions more often (#149257) 2025-03-18 00:46:07 +00:00
common_modules.py Remove outdated skipCUDAIfCudnnVersionLessThan decoration (#148940) 2025-03-13 18:02:50 +00:00
common_mps.py [BE] Switch TestConsistency to MPS device (#147893) 2025-04-26 01:19:21 +00:00
common_nn.py
common_optimizers.py
common_pruning.py
common_quantization.py [Codemod][AddExplicitStrictExportForTrainingInferenceArg] caffe2/ (#149595) 2025-04-03 23:50:13 +00:00
common_quantized.py wire torch._scaled_mm with fp4 operands to the cublas nvfp4 kernel (#148792) 2025-03-27 17:32:20 +00:00
common_subclass.py
common_utils.py [BE] Migrate dtype_abbrs into one location (#152229) 2025-04-28 03:52:47 +00:00
composite_compliance.py [pytree] add APIs to determine a class is a namedtuple or PyStructSequence (#113257) 2025-04-01 10:40:43 +00:00
custom_op_db.py
custom_tensor.py
dist_utils.py
dynamo_test_failures.py
fake_config_module.py
fake_config_module2.py
fake_config_module3.py
hop_db.py
hypothesis_utils.py
inductor_utils.py cpp_wrapper: Fix even more tests (#147225) 2025-04-07 14:20:06 +00:00
jit_metaprogramming_utils.py
jit_utils.py
logging_tensor.py [BE] Migrate dtype_abbrs into one location (#152229) 2025-04-28 03:52:47 +00:00
logging_utils.py
quantization_torch_package_models.py
static_module.py
subclasses.py
torchbind_impls.py Fakify torchbind objects in compile_fx and add tests for SigridTransformsInstanceTorchBind (#149529) 2025-03-21 18:58:28 +00:00
triton_utils.py [AOTInductor] Fix autotuning code's codegen (#150522) 2025-04-03 00:08:19 +00:00
two_tensor.py Support subclass constructor capturing in export (#147014) 2025-03-16 18:19:19 +00:00