Replace `export_for_training` with `torch.export.export` (#2724)
* Deprecate old TORCH_VERSION variables **Summary:** This commit deprecates the following variables: ``` TORCH_VERSION_AT_LEAST_2_5 TORCH_VERSION_AT_LEAST_2_4 TORCH_VERSION_AT_LEAST_2_3 TORCH_VERSION_AT_LEAST_2_2 TORCH_VERSION_AFTER_2_5 TORCH_VERSION_AFTER_2_4 TORCH_VERSION_AFTER_2_3 TORCH_VERSION_AFTER_2_2 ``` As of this commit, the latest released version of PyTorch is 2.8, which means we can drop support for 2.5 and before since we only support 3 of the latest releases. The next commit will remove usages of all of these variables from within torchao. **Test Plan:** ``` python test/test_utils.py -k torch_version_deprecation ``` [ghstack-poisoned] * Update on "Deprecate old TORCH_VERSION variables" **Summary:** This commit deprecates the following variables: ``` TORCH_VERSION_AT_LEAST_2_5 TORCH_VERSION_AT_LEAST_2_4 TORCH_VERSION_AT_LEAST_2_3 TORCH_VERSION_AT_LEAST_2_2 TORCH_VERSION_AFTER_2_5 TORCH_VERSION_AFTER_2_4 TORCH_VERSION_AFTER_2_3 TORCH_VERSION_AFTER_2_2 ``` As of this commit, the latest released version of PyTorch is 2.8, which means we can drop support for 2.5 and before since we only support 3 of the latest releases. The next commit will remove usages of all of these variables from within torchao. **Test Plan:** ``` python test/test_utils.py -k torch_version_deprecation ``` [ghstack-poisoned] * Drop support for PyTorch 2.5 and before **Summary:** We gate on the PyTorch version throughout the repo. Recently PyTorch 2.8 was released, so the oldest PyTorch version we need to support is 2.6. After this commit, we assume the user is running PyTorch 2.6+, and remove all references to the following variables, which are deprecated. ``` TORCH_VERSION_AT_LEAST_2_6 TORCH_VERSION_AT_LEAST_2_5 TORCH_VERSION_AT_LEAST_2_4 TORCH_VERSION_AT_LEAST_2_3 TORCH_VERSION_AT_LEAST_2_2 TORCH_VERSION_AFTER_2_5 TORCH_VERSION_AFTER_2_4 TORCH_VERSION_AFTER_2_3 TORCH_VERSION_AFTER_2_2 ``` **Test Plan:** CI [ghstack-poisoned] * Remove old `change_linear_weights_to_*` APIs **Summary:** This commit removes these super old quantization APIs that aren't even accessible by the user: ``` change_linear_weights_to_int8_dqtensors change_linear_weights_to_int8_woqtensors change_linear_weights_to_int4_woqtensors ``` **Test Plan:** CI [ghstack-poisoned] * Update base for Update on "Remove old `change_linear_weights_to_*` APIs" **Summary:** This commit removes these super old quantization APIs that aren't even accessible by the user: ``` change_linear_weights_to_int8_dqtensors change_linear_weights_to_int8_woqtensors change_linear_weights_to_int4_woqtensors ``` **Test Plan:** CI [ghstack-poisoned] * Update base for Update on "Remove old `change_linear_weights_to_*` APIs" **Summary:** This commit removes these super old quantization APIs that aren't even accessible by the user: ``` change_linear_weights_to_int8_dqtensors change_linear_weights_to_int8_woqtensors change_linear_weights_to_int4_woqtensors ``` **Test Plan:** CI [ghstack-poisoned] * Update base for Update on "Remove old `change_linear_weights_to_*` APIs" **Summary:** This commit removes these super old quantization APIs that aren't even accessible by the user: ``` change_linear_weights_to_int8_dqtensors change_linear_weights_to_int8_woqtensors change_linear_weights_to_int4_woqtensors ``` **Test Plan:** CI [ghstack-poisoned] * Update base for Update on "Remove old `change_linear_weights_to_*` APIs" **Summary:** This commit removes these super old quantization APIs that aren't even accessible by the user: ``` change_linear_weights_to_int8_dqtensors change_linear_weights_to_int8_woqtensors change_linear_weights_to_int4_woqtensors ``` **Test Plan:** CI [ghstack-poisoned] * Replace `export_for_training` with `torch.export.export` **Summary:** Bypasses the following deprecation warning: ``` `torch.export.export_for_training` is deprecated and will be removed in PyTorch 2.10. Please use `torch.export.export` instead, which is functionally equivalent. ``` Bonus: remove some references to `capture_pre_autograd_graph`, which is even older. **Test Plan:** CI [ghstack-poisoned] * Update base for Update on "Replace `export_for_training` with `torch.export.export`" **Summary:** Bypasses the following deprecation warning: ``` `torch.export.export_for_training` is deprecated and will be removed in PyTorch 2.10. Please use `torch.export.export` instead, which is functionally equivalent. ``` Bonus: remove some references to `capture_pre_autograd_graph`, which is even older. **Test Plan:** CI [ghstack-poisoned] * Update base for Update on "Replace `export_for_training` with `torch.export.export`" **Summary:** Bypasses the following deprecation warning: ``` `torch.export.export_for_training` is deprecated and will be removed in PyTorch 2.10. Please use `torch.export.export` instead, which is functionally equivalent. ``` Bonus: remove some references to `capture_pre_autograd_graph`, which is even older. **Test Plan:** CI [ghstack-poisoned] * Update base for Update on "Replace `export_for_training` with `torch.export.export`" **Summary:** Bypasses the following deprecation warning: ``` `torch.export.export_for_training` is deprecated and will be removed in PyTorch 2.10. Please use `torch.export.export` instead, which is functionally equivalent. ``` Bonus: remove some references to `capture_pre_autograd_graph`, which is even older. **Test Plan:** CI [ghstack-poisoned] * Update base for Update on "Replace `export_for_training` with `torch.export.export`" **Summary:** Bypasses the following deprecation warning: ``` `torch.export.export_for_training` is deprecated and will be removed in PyTorch 2.10. Please use `torch.export.export` instead, which is functionally equivalent. ``` Bonus: remove some references to `capture_pre_autograd_graph`, which is even older. **Test Plan:** CI [ghstack-poisoned] * Update base for Update on "Replace `export_for_training` with `torch.export.export`" **Summary:** Bypasses the following deprecation warning: ``` `torch.export.export_for_training` is deprecated and will be removed in PyTorch 2.10. Please use `torch.export.export` instead, which is functionally equivalent. ``` Bonus: remove some references to `capture_pre_autograd_graph`, which is even older. **Test Plan:** CI [ghstack-poisoned]
A
andrewor14 committed
615877df69eeeed8d2b87cf7bbb1a05305912cbc
Parent: e79208c
Committed by GitHub <noreply@github.com>
on 8/13/2025, 2:14:17 PM