Skip to content

Conversation

@alanwaketan
Copy link
Collaborator

@alanwaketan alanwaketan commented Dec 12, 2022

Summary:
This patch overrides LazyGraphExecutor::RunPostOrder() and adds a few comments on why some methods are different from the upstream.

This depends on pytorch/pytorch#90680.

Test Plan:
CI

Copy link
Collaborator

@wonjoo-wj wonjoo-wj left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Nice, LGTM pending CI completion!

@alanwaketan
Copy link
Collaborator Author

Thanks Wonjoo and Jack.

@alanwaketan alanwaketan force-pushed the alanwaketan/lazy_graph_5 branch from 5718904 to adab74e Compare December 13, 2022 09:57
pytorchmergebot pushed a commit to pytorch/pytorch that referenced this pull request Dec 13, 2022
Summary:
This patch makes LazyGraphExecutor::RunPostOrder() virtual such that XLA can reuse it.

It's related to pytorch/xla#4315.

Test Plan:
CI.

Pull Request resolved: #90680
Approved by: https://github.com/antoniojkim, https://github.com/JackCaoG
This reverts commit adab74e.
@alanwaketan alanwaketan merged commit 4ffdfe2 into master Dec 13, 2022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

tracing Lazy Tensor tracing

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants