frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

[BE] NO MORE discrepancy between forloop foreach capturable YAY (#121269)

Browse Source 
and I will not let it happen again

Pull Request resolved: https://github.com/pytorch/pytorch/pull/121269
Approved by: https://github.com/albanD
ghstack dependencies: #121260, #121264
This commit is contained in:
Jane Xu
2024-03-07 12:32:46 -08:00
committed by PyTorch MergeBot
parent 9d6c5be781
commit f76e541ec7
2 changed files with 0 additions and 107 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
torch/testing/_internal/common_optimizers.py
View File

@ -114,8 +114,6 @@ class OptimizerInfo:
supports_param_groups: bool = True,
# whether the optimizer supports parameters on multiple devices
supports_multiple_devices: bool = True,
# whether the optimizer ONLY supports capturable on foreach vs. both foreach and forloop
only_supports_capturable_on_foreach: bool = False,
skips=(), # Indicates which tests to skip
decorators=None, # Additional decorators to apply to generated tests
optim_error_inputs_func=None, # Function to generate optim inputs that error
@ -128,7 +126,6 @@ class OptimizerInfo:
self.step_requires_closure = step_requires_closure
self.supports_param_groups = supports_param_groups
self.supports_multiple_devices = supports_multiple_devices
self.only_supports_capturable_on_foreach = only_supports_capturable_on_foreach
self.decorators = (
*(decorators if decorators else []),
*(skips if skips else []),