pytorch/torch/testing/_internal
Rohan Varma c11412b4a8 [DDP] Support optim in backward after DDP init (#105995)
This allows in backward optimizers to be configured after DDP init, in
addition to before as was previously supported.

Differential Revision: [D47783347](https://our.internmc.facebook.com/intern/diff/D47783347/)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/105995
Approved by: https://github.com/fegin
2023-07-29 01:36:25 +00:00
..
codegen [BE] f-stringify torch/ and scripts (#105538) 2023-07-21 19:35:24 +00:00
data
distributed [DDP] Support optim in backward after DDP init (#105995) 2023-07-29 01:36:25 +00:00
generated
opinfo [foreach] Set SavedVariable.is_output to true for grad_fn->result_ (#105504) 2023-07-26 14:29:32 +00:00
optests [operator_compile_check] Add FakeTensor testing (#103595) 2023-06-16 16:55:51 +00:00
test_module
__init__.py
autocast_test_lists.py
autograd_function_db.py
check_kernel_launches.py [BE] Enable ruff's UP rules and autoformat testing/ (#105425) 2023-07-18 21:04:39 +00:00
common_cuda.py [BE] Enable ruff's UP rules and autoformat testing/ (#105425) 2023-07-18 21:04:39 +00:00
common_device_type.py [BE]: Update Ruff to 0.0.280 (#105724) 2023-07-22 23:03:34 +00:00
common_dist_composable.py
common_distributed.py [Composable FSDP] Enable HSDP (#105206) 2023-07-26 21:03:55 +00:00
common_dtype.py
common_fsdp.py [BE] Enable ruff's UP rules and autoformat testing/ (#105425) 2023-07-18 21:04:39 +00:00
common_jit.py
common_methods_invocations.py [pt2] add meta for argsort.stable, use sort samples in OpInfo (#106025) 2023-07-27 03:49:17 +00:00
common_modules.py Added ModuleInfo test for meta device ctx init (#105871) 2023-07-26 01:57:54 +00:00
common_nn.py [BE] f-stringify torch/ and scripts (#105538) 2023-07-21 19:35:24 +00:00
common_pruning.py [BE] Enable ruff's UP rules and autoformat testing/ (#105425) 2023-07-18 21:04:39 +00:00
common_quantization.py [BE] f-stringify torch/ and scripts (#105538) 2023-07-21 19:35:24 +00:00
common_quantized.py
common_subclass.py
common_utils.py Print env var name instead of flag name for commandline repros (#106223) 2023-07-28 23:22:27 +00:00
composite_compliance.py
control_flow_opinfo_db.py
custom_op_db.py Update custom op API (#105947) 2023-07-28 13:30:58 +00:00
dist_utils.py [BE] Enable ruff's UP rules and autoformat testing/ (#105425) 2023-07-18 21:04:39 +00:00
hypothesis_utils.py
inductor_utils.py
jit_metaprogramming_utils.py [BE] f-stringify torch/ and scripts (#105538) 2023-07-21 19:35:24 +00:00
jit_utils.py [BE]: Update Ruff to 0.0.280 (#105724) 2023-07-22 23:03:34 +00:00
logging_tensor.py Upgrade LoggingTensor mode and add traceback collection (#103734) 2023-06-21 22:04:30 +00:00
logging_utils.py [Dynamo] Enable torch._dynamo.config.suppress_errors by default (#105307) 2023-07-21 19:17:46 +00:00
quantization_torch_package_models.py