pytorch/torch/testing/_internal
2025-03-18 00:46:07 +00:00
..
codegen
data
distributed Revert "[PGNCCL] Launch kernel on current stream & remove record_stream entirely (#148590)" 2025-03-17 22:43:15 +00:00
generated
opinfo Implement gradient for the residuals of torch.linalg.lstsq (#148526) 2025-03-10 12:35:09 +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 PEP585 update - torch/testing (#145200) 2025-01-20 22:42:42 +00:00
common_cuda.py [BE]: Apply ruff PERF403 to use dict comprehensions more often (#149257) 2025-03-18 00:46:07 +00:00
common_device_type.py cpp_wrapper: use largeTensorTest for test memory checks (#146991) 2025-02-27 00:30:21 +00:00
common_dist_composable.py Migrate from Tuple -> tuple in torch/testing (#144256) 2025-01-10 06:37:55 +00:00
common_distributed.py Enable FSDP tests on XPU device (#147518) 2025-03-04 23:49:37 +00:00
common_dtype.py [BE]: Enable ruff SLOT checks (#146276) 2025-02-04 19:18:23 +00:00
common_fsdp.py Enable FSDP tests on XPU device (#147518) 2025-03-04 23:49:37 +00:00
common_jit.py PEP585 update - torch/testing (#145200) 2025-01-20 22:42:42 +00:00
common_methods_invocations.py [ROCm] Bump AOTriton to 0.9.2b (#148433) 2025-03-07 22:10:07 +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_nn.py PEP585 update - torch/testing (#145200) 2025-01-20 22:42:42 +00:00
common_optimizers.py Remove code for Python < 3.9 (#147097) 2025-02-14 03:22:49 +00:00
common_pruning.py PEP585 update - torch/testing (#145200) 2025-01-20 22:42:42 +00:00
common_quantization.py [BE]: Enable ruff rule SIM113 (#147290) 2025-02-16 22:41:16 +00:00
common_quantized.py
common_subclass.py
common_utils.py Add __context/cause/suppress_context/traceback__ to Exception (#146499) 2025-03-11 18:55:45 +00:00
composite_compliance.py Revert "[pytree] add APIs to determine a class is a namedtuple or PyStructSequence (#113257)" 2025-03-14 23:13:34 +00:00
custom_op_db.py Migrate from Tuple -> tuple in torch/testing (#144256) 2025-01-10 06:37:55 +00:00
custom_tensor.py Support getattr for tensor subclasses in pre-dispatch export via patching tensor.getattr (#143946) 2025-01-06 23:55:50 +00:00
dist_utils.py Migrate from Tuple -> tuple in torch/testing (#144256) 2025-01-10 06:37:55 +00:00
dynamo_test_failures.py [BE] Remove test_ops from FIXME_inductor_dont_reset_dynamo (#145307) 2025-01-27 18:12:39 +00:00
fake_config_module.py config: Support str env variables (#145980) 2025-01-30 00:13:02 +00:00
fake_config_module2.py Add multi env variable support to configs (#145288) 2025-01-24 10:04:24 +00:00
fake_config_module3.py [minifier] Fix config generator for callables (#144518) 2025-01-14 17:18:13 +00:00
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 [Inductor UT] Enable PYTORCH_TESTING_DEVICE_ONLY_FOR test case filter for test_torchinductor.py (#149023) 2025-03-13 05:15:28 +00:00
jit_metaprogramming_utils.py [BE]: Enable ruff SLOT checks (#146276) 2025-02-04 19:18:23 +00:00
jit_utils.py Fix linter F821 error (#146665) 2025-02-08 07:19:37 +00:00
logging_tensor.py PEP585 update - torch/testing (#145200) 2025-01-20 22:42:42 +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 Consolidate torchbind fake class registration (#149063) 2025-03-13 06:57:13 +00:00
triton_utils.py Fix None and equal_to_1 arguments issue in Triton kernel generated by AOTI (#148102) 2025-03-01 18:38:33 +00:00
two_tensor.py Support subclass constructor capturing in export (#147014) 2025-03-16 18:19:19 +00:00