Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Migrate away from deprecated PyTorch API #2400

Open
wants to merge 1 commit into
base: main
Choose a base branch
from

Conversation

yushangdi
Copy link

@yushangdi yushangdi commented Nov 20, 2024

Hi CoreML group,

The capture_pre_autograd_graph API is deprecated and will be deleted soon. Please use the export_for_training API instead. The export_for_training API is available since PyTorch 2.5.0.

Thanks!

cc @shoumikhin

@jakesabathia2
Copy link
Collaborator

Thank you @yushangdi for this great fix.
@YifanShenSZ I think we might need to first upgrade our supported torch version to 2.5.0.

@YifanShenSZ YifanShenSZ self-assigned this Nov 25, 2024
@YifanShenSZ
Copy link
Collaborator

Yes, we will need some work to upgrade our torch support to 2.5, due to numpy conflict from latest executorch 0.4

  • We support numpy 2.0+
  • However, executorch 0.4 requires numpy 1.21.3

Will come back to this PR as soon as we resolve that

yushangdi added a commit to yushangdi/pytorch that referenced this pull request Dec 13, 2024
Summary:
Delete the following API:

- capture_pre_autograd_graph()
- capture_pre_autograd_graph_using_training_ir()
- gm_using_training_ir()

There's no more call sites to `capture_pre_autograd_graph`.

Except
1) two test cases in coreml, PR to remove: apple/coremltools#2400
2) XLA: one test case in pytorch/xla, PR to remove: pytorch/xla#8398
3) a few call sites guarded by version guard (< 2.5.0)

Test Plan: CI

Reviewed By: tugsbayasgalan

Differential Revision: D64056353
yushangdi added a commit to yushangdi/pytorch that referenced this pull request Dec 13, 2024
Summary:
as title

There's no more call sites to `capture_pre_autograd_graph`.

Except
1) two test cases in coreml, PR to remove: apple/coremltools#2400
2) XLA: one test case in pytorch/xla, PR to remove: pytorch/xla#8398
3) a few call sites guarded by version guard (< 2.5.0)

Test Plan: CI

Differential Revision: D64056353
yushangdi added a commit to yushangdi/pytorch that referenced this pull request Dec 16, 2024
Summary:

Delete the following API:

- capture_pre_autograd_graph()
- capture_pre_autograd_graph_using_training_ir()
- gm_using_training_ir()

There's no more call sites to `capture_pre_autograd_graph`.

Except
1) two test cases in coreml, PR to remove: apple/coremltools#2400
2) XLA: one test case in pytorch/xla, PR to remove: pytorch/xla#8398
3) a few call sites guarded by version guard (< 2.5.0)

Test Plan: CI

Reviewed By: tugsbayasgalan

Differential Revision: D64056353
pytorchmergebot pushed a commit to pytorch/pytorch that referenced this pull request Dec 16, 2024
Summary:
Delete the following API:

- capture_pre_autograd_graph()
- capture_pre_autograd_graph_using_training_ir()
- gm_using_training_ir()

There's no more call sites to `capture_pre_autograd_graph`.

Except
1) two test cases in coreml, PR to remove: apple/coremltools#2400
2) XLA: one test case in pytorch/xla, PR to remove: pytorch/xla#8398
3) a few call sites guarded by version guard (< 2.5.0)

Test Plan: CI

Reviewed By: tugsbayasgalan

Differential Revision: D64056353

Pull Request resolved: #143224
Approved by: https://github.com/tugsbayasgalan
yushangdi added a commit to yushangdi/pytorch that referenced this pull request Dec 17, 2024
Summary:
Delete the following API:

- capture_pre_autograd_graph()
- capture_pre_autograd_graph_using_training_ir()
- gm_using_training_ir()

There's no more call sites to `capture_pre_autograd_graph`.

Update XLA pin to include pytorch/xla#8398

Except
1) two test cases in coreml, PR to remove: apple/coremltools#2400
2) a few call sites guarded by version guard (< 2.5.0)

Test Plan: CI

Differential Revision: D67354440
pytorchmergebot pushed a commit to pytorch/pytorch that referenced this pull request Dec 18, 2024
Summary:
Delete the following API:

- capture_pre_autograd_graph()
- capture_pre_autograd_graph_using_training_ir()
- gm_using_training_ir()

Update XLA pin to include pytorch/xla#8398

There's no more call sites to `capture_pre_autograd_graph`.

Except
1) two test cases in coreml, guarded by version guard, PR to remove: apple/coremltools#2400
2) a few call sites guarded by version guard (< 2.5.0)

Test Plan: CI

Differential Revision: D67354440

Pull Request resolved: #143426
Approved by: https://github.com/gmagogsfm
aditew01 pushed a commit to aditew01/pytorch that referenced this pull request Dec 18, 2024
Summary:
Delete the following API:

- capture_pre_autograd_graph()
- capture_pre_autograd_graph_using_training_ir()
- gm_using_training_ir()

There's no more call sites to `capture_pre_autograd_graph`.

Except
1) two test cases in coreml, PR to remove: apple/coremltools#2400
2) XLA: one test case in pytorch/xla, PR to remove: pytorch/xla#8398
3) a few call sites guarded by version guard (< 2.5.0)

Test Plan: CI

Reviewed By: tugsbayasgalan

Differential Revision: D64056353

Pull Request resolved: pytorch#143224
Approved by: https://github.com/tugsbayasgalan
aditew01 pushed a commit to aditew01/pytorch that referenced this pull request Dec 18, 2024
Summary:
Delete the following API:

- capture_pre_autograd_graph()
- capture_pre_autograd_graph_using_training_ir()
- gm_using_training_ir()

Update XLA pin to include pytorch/xla#8398

There's no more call sites to `capture_pre_autograd_graph`.

Except
1) two test cases in coreml, guarded by version guard, PR to remove: apple/coremltools#2400
2) a few call sites guarded by version guard (< 2.5.0)

Test Plan: CI

Differential Revision: D67354440

Pull Request resolved: pytorch#143426
Approved by: https://github.com/gmagogsfm
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants