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 from capture_pre_autograd_graph to export_for_training #8398

Merged
merged 1 commit into from
Dec 17, 2024

Conversation

yushangdi
Copy link
Contributor

capture_pre_autograd_graph API is deprecated and will be deleted.

capture_pre_autograd_graph API is deprecated and will be deleted.
@yushangdi
Copy link
Contributor Author

yushangdi commented Nov 19, 2024

@lsy323 @qihqi Hi, the capture_pre_autograd_graph API is deprecated and will be deleted soon. Please use export_for_training instead. Thanks!

@JackCaoG JackCaoG requested a review from qihqi November 19, 2024 23:39
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
@huydhn
Copy link
Contributor

huydhn commented Dec 17, 2024

Ping for review as this is blocking a change on PyTorch side at pytorch/pytorch#143224 (comment)

@miladm
Copy link
Collaborator

miladm commented Dec 17, 2024

cc @lsy323 to help look review this PR from the TorchXLA side.
(cc @tengyifei for viz)

@miladm miladm requested a review from lsy323 December 17, 2024 19:55
@lsy323 lsy323 merged commit b2b890e into pytorch:master Dec 17, 2024
12 checks passed
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.

4 participants