pytorch/torch/testing/_internal
2025-03-25 17:58:40 +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
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
common_distributed.py Enable FSDP tests on XPU device (#147518) 2025-03-04 23:49:37 +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 [aotd] Guess tangents stride as output strides (#144579) 2025-03-20 15:41:36 +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
common_optimizers.py Remove code for Python < 3.9 (#147097) 2025-02-14 03:22:49 +00:00
common_pruning.py
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
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-03-25 17:58:40 +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
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 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