pytorch/torch/testing/_internal
PyTorch MergeBot fdc387ec7c Revert "refine fp32 precision api (#125888)"
This reverts commit 4c11b26158.

Reverted https://github.com/pytorch/pytorch/pull/125888 on behalf of https://github.com/huydhn due to Sorry for reverting your change but it seems to cause some failures on ROCm ([comment](https://github.com/pytorch/pytorch/pull/125888#issuecomment-2869274791))
2025-05-11 00:35:46 +00:00
..
codegen
data
distributed [BE][DTensor] move torch.distributed._tensor import to torch.distributed.tensor in test files (#153225) 2025-05-09 20:40:54 +00:00
generated
opinfo [ROCm] Upgrade ROCm CI to ROCm6.4 (#151368) 2025-05-08 16:12:16 +00:00
optests [opcheck] Improve error reporting; allow atol/rtol overrides (#146488) 2025-02-05 21:25:06 +00:00
test_module
__init__.py
autocast_test_lists.py
autograd_function_db.py
check_kernel_launches.py
common_cuda.py [TEST][ATen][CUDA] Skip row-wise scaled matrix mmultiplication tests on sm_120+ (#152814) 2025-05-08 19:34:20 +00:00
common_device_type.py Fix instantiate_device_type_tests() for 3rd-party devices (#152177) 2025-04-30 06:25:59 +00:00
common_dist_composable.py
common_distributed.py [c10d] Reduce test verbosity (#153116) 2025-05-08 22:22:22 +00:00
common_dtype.py
common_fsdp.py Enable FSDP tests on XPU device (#147518) 2025-03-04 23:49:37 +00:00
common_jit.py
common_methods_invocations.py [ROCm] Upgrade ROCm CI to ROCm6.4 (#151368) 2025-05-08 16:12:16 +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 Move additional MPS Unary ops to Iterator (#152876) 2025-05-07 00:06:54 +00:00
common_nn.py ROCm: Enable tf32 testing on test_nn (#148945) 2025-04-28 23:01:04 +00:00
common_optimizers.py Remove code for Python < 3.9 (#147097) 2025-02-14 03:22:49 +00:00
common_pruning.py
common_quantization.py [Codemod][AddExplicitStrictExportForTrainingInferenceArg] caffe2/ (#149595) 2025-04-03 23:50:13 +00:00
common_quantized.py Ensure mxfp8 scaled_mm works w/ max-autotune (#152744) 2025-05-06 01:16:57 +00:00
common_subclass.py
common_utils.py Revert "refine fp32 precision api (#125888)" 2025-05-11 00:35:46 +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 Support torch.compile rng selective activation checkpointing with cudagraph (#146878) 2025-02-28 00:47:03 +00:00
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 Fix linter F821 error (#146665) 2025-02-08 07:19:37 +00:00
logging_tensor.py [BE] Migrate dtype_abbrs into one location (#152229) 2025-04-28 03:52:47 +00:00
logging_utils.py Update ruff linter for PEP585 (#147540) 2025-02-22 04:45:17 +00:00
quantization_torch_package_models.py
static_module.py
subclasses.py [subclass] testing WrapperSubclass respect outer_size, outer_stride (#146897) 2025-02-13 15:21:19 +00:00
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