pytorch/torch/testing/_internal
2024-12-03 01:32:50 +00:00
..
codegen
data
distributed [DTensorTestbase] Fix TestFunc typing issue (#141513) 2024-11-26 19:48:34 +00:00
generated
opinfo Pass rounding_mode for div reference inputs through kwargs (#136308) 2024-11-29 21:28:24 +00:00
optests
test_module
__init__.py
autocast_test_lists.py
autograd_function_db.py
check_kernel_launches.py
common_cuda.py
common_device_type.py Add API query for available per-process CUDA memory (#140620) 2024-12-03 00:24:03 +00:00
common_dist_composable.py
common_distributed.py Revert "[CI] Reduce distributed test timeout to 60s (#141168)" 2024-11-22 15:46:37 +00:00
common_dtype.py
common_fsdp.py Revert "[BE]: Update mypy to 1.13.0 (#140808)" 2024-12-02 20:47:43 +00:00
common_jit.py
common_methods_invocations.py Pass rounding_mode for div reference inputs through kwargs (#136308) 2024-11-29 21:28:24 +00:00
common_mkldnn.py
common_modules.py [BE] Use torch.log1p(x) instead of torch.log(1+x) (#141167) 2024-11-21 00:36:20 +00:00
common_nn.py
common_optimizers.py Support tensor betas in Adam and AdamW (#134171) 2024-11-15 21:55:55 +00:00
common_pruning.py Fix type-safety of torch.nn.Module instances (#141240) 2024-11-22 00:05:05 +00:00
common_quantization.py Revert "Add missing data types at torch export serialization (#138561)" 2024-12-03 01:32:50 +00:00
common_quantized.py [BE]: Apply PERF401 autofixes from ruff (#140980) 2024-11-20 17:52:07 +00:00
common_subclass.py
common_utils.py [Intel GPU] qconv at XPU backend (#133080) 2024-11-26 02:24:30 +00:00
composite_compliance.py
custom_op_db.py
custom_tensor.py
dist_utils.py
dynamo_test_failures.py
fake_config_module.py Add type annotations to Configs (#139833) 2024-11-07 03:49:09 +00:00
hop_db.py [invoke_subgraph] User facing API to support arbitrary args and kwargs (#139162) 2024-11-08 03:31:19 +00:00
hypothesis_utils.py [BE]: Apply PERF401 autofixes from ruff (#140980) 2024-11-20 17:52:07 +00:00
inductor_utils.py Revert "Refactor test_torchinductor_strided_blocks to also support triton CPU (#141587)" 2024-11-28 19:41:03 +00:00
jit_metaprogramming_utils.py
jit_utils.py
logging_tensor.py add float8 types to LoggingTensor (#141385) 2024-11-26 23:39:57 +00:00
logging_utils.py
quantization_torch_package_models.py
static_module.py
subclasses.py [aotd] coerce_same_metadata_as_tangent with expected_type for e.g.AsyncCollectiveTensor (#139095) 2024-11-07 16:24:48 +00:00
torchbind_impls.py
triton_utils.py [RFC] Implement caching for user defined triton kernels (#140326) 2024-11-16 02:37:16 +00:00
two_tensor.py