Skip to content

[pytree] simplify public API exposition with __module__ #148328

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.

Sign up for GitHub

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

Jump to bottom
Open
wants to merge 83 commits into
base: gh/XuehaiPan/253/base
Choose a base branch
from
Open

[pytree] simplify public API exposition with __module__ #148328

wants to merge 83 commits into from

Conversation

XuehaiPan
Copy link
Collaborator

@XuehaiPan XuehaiPan commented Mar 3, 2025 •
edited
Loading

Stack from ghstack (oldest at bottom):

Before this PR, the following statements are already available:

import torch

import torch.utils.pytree.python
from torch.utils.pytree import python
from torch.utils.pytree import cxx
from torch.utils.pytree.python import tree_map

torch.utils.pytree.python
torch.utils.pytree.python.tree_map
torch.utils.pytree.cxx
torch.utils.pytree.cxx.tree_map

This PR makes the following extra statements available:

import torch.utils.pytree.cxx
from torch.utils.pytree.cxx import tree_map

cc @zou3519 @voznesenskym @penguinwu @EikanWang @jgong5 @Guobing-Chen @XiaobingSuper @zhuhaozhe @blzheng @wenzhe-nrv @jiayisunx @chenyang78 @kadeng @chauhang @amjames

@XuehaiPan
Update
5869417 
[ghstack-poisoned]
@pytorch-bot PyTorch Bot
Copy link

pytorch-bot bot commented Mar 3, 2025 •
edited
Loading

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/148328

Note: Links to docs will display an error until the docs builds have been completed.

❌ 2 New Failures

As of commit ac991d3 with merge base 556e2a7 (image):

NEW FAILURES - The following jobs have failed:

This comment was automatically generated by Dr. CI and updates every 15 minutes.

This was referenced Mar 3, 2025
Open
Open
@XuehaiPan XuehaiPan added the topic: not user facing topic category label Mar 3, 2025
@XuehaiPan XuehaiPan requested review from zou3519 and albanD March 3, 2025 14:40
@XuehaiPan
Update
c45807e 
[ghstack-poisoned]
@XuehaiPan XuehaiPan added the ci-test-showlocals Show local variables on test failures label Mar 3, 2025
@XuehaiPan
Update
ae1f74d 
[ghstack-poisoned]
XuehaiPan added a commit that referenced this pull request Mar 3, 2025
@XuehaiPan
[pytree] simplify public API exposition with __module__
67aaa6a 
ghstack-source-id: 3f6962f
Pull Request resolved: #148328
@XuehaiPan
Update
15f6dfa 
[ghstack-poisoned]
XuehaiPan added a commit to XuehaiPan/pytorch that referenced this pull request Mar 3, 2025
@XuehaiPan
[pytree] simplify public API exposition with __module__
aa01744 
ghstack-source-id: 35e6bba
Pull Request resolved: pytorch#148328
@XuehaiPan
Update
c76deaa 
[ghstack-poisoned]
@XuehaiPan
Update
085e755 
[ghstack-poisoned]
@XuehaiPan
Update
f1b7f65 
[ghstack-poisoned]
@XuehaiPan
Update
84e3de9 
[ghstack-poisoned]
XuehaiPan added a commit that referenced this pull request Jun 6, 2025
@XuehaiPan
[pytree] simplify public API exposition with __module__
6e3a58c 
ghstack-source-id: 12fe7e8
Pull Request resolved: #148328
XuehaiPan added a commit to XuehaiPan/pytorch that referenced this pull request Jun 9, 2025
@XuehaiPan
[pytree] simplify public API exposition with __module__
f5487e1 
ghstack-source-id: 12fe7e8
Pull Request resolved: pytorch#148328
XuehaiPan added a commit to XuehaiPan/pytorch that referenced this pull request Jun 11, 2025
@XuehaiPan
[pytree] simplify public API exposition with __module__
f2e7edd 
ghstack-source-id: 12fe7e8
Pull Request resolved: pytorch#148328
XuehaiPan added a commit to XuehaiPan/pytorch that referenced this pull request Jun 16, 2025
@XuehaiPan
[pytree] simplify public API exposition with __module__
9097728 
ghstack-source-id: 12fe7e8
Pull Request resolved: pytorch#148328
@XuehaiPan
Update
9743e29 
[ghstack-poisoned]
XuehaiPan added a commit that referenced this pull request Jun 18, 2025
@XuehaiPan
[pytree] simplify public API exposition with __module__
39baabf 
ghstack-source-id: 6d95de7
Pull Request resolved: #148328
@XuehaiPan
Update
1841fda 
[ghstack-poisoned]
XuehaiPan added a commit that referenced this pull request Jun 23, 2025
@XuehaiPan
[pytree] simplify public API exposition with __module__
92f7e7a 
ghstack-source-id: a168be7
Pull Request resolved: #148328
XuehaiPan added a commit to XuehaiPan/pytorch that referenced this pull request Jun 24, 2025
@XuehaiPan
[pytree] simplify public API exposition with __module__
064e1b2 
ghstack-source-id: a168be7
Pull Request resolved: pytorch#148328
XuehaiPan added a commit to XuehaiPan/pytorch that referenced this pull request Jun 24, 2025
@XuehaiPan
[pytree] simplify public API exposition with __module__
f249242 
ghstack-source-id: a168be7
Pull Request resolved: pytorch#148328
@XuehaiPan
Update
ab06a3a 
[ghstack-poisoned]
XuehaiPan added a commit that referenced this pull request Jun 27, 2025
@XuehaiPan
[pytree] simplify public API exposition with __module__
0db2a8f 
ghstack-source-id: 21a72dd
Pull Request resolved: #148328
XuehaiPan added a commit that referenced this pull request Jun 27, 2025
@XuehaiPan
[pytree] simplify public API exposition with __module__
30dc1dd 
ghstack-source-id: 21a72dd
Pull Request resolved: #148328
@XuehaiPan
Update
00bf718 
[ghstack-poisoned]
XuehaiPan added a commit that referenced this pull request Jun 28, 2025
@XuehaiPan
[pytree] simplify public API exposition with __module__
41c35be 
ghstack-source-id: c32653c
Pull Request resolved: #148328
@XuehaiPan
Update
26cff99 
[ghstack-poisoned]
XuehaiPan added a commit that referenced this pull request Jul 3, 2025
@XuehaiPan
[pytree] simplify public API exposition with __module__
de6a8ca 
ghstack-source-id: 4deb7da
Pull Request resolved: #148328
XuehaiPan added a commit to XuehaiPan/pytorch that referenced this pull request Jul 5, 2025
@XuehaiPan
[pytree] simplify public API exposition with __module__
c67e943 
ghstack-source-id: 4deb7da
Pull Request resolved: pytorch#148328
@XuehaiPan
Update
72fa391 
[ghstack-poisoned]
XuehaiPan added a commit that referenced this pull request Jul 9, 2025
@XuehaiPan
[pytree] simplify public API exposition with __module__
b46f356 
ghstack-source-id: f642999
Pull Request resolved: #148328
@XuehaiPan
Update
80a7d88 
[ghstack-poisoned]
XuehaiPan added a commit that referenced this pull request Jul 17, 2025
@XuehaiPan
[pytree] simplify public API exposition with __module__
21a43a7 
ghstack-source-id: 0429d56
Pull Request resolved: #148328
@XuehaiPan
Update
99edc1d 
[ghstack-poisoned]
XuehaiPan added a commit that referenced this pull request Jul 25, 2025
@XuehaiPan
[pytree] simplify public API exposition with __module__
fcbf8b0 
ghstack-source-id: ad88449
Pull Request resolved: #148328
@XuehaiPan
Update
e3129b8 
[ghstack-poisoned]
XuehaiPan added a commit that referenced this pull request Jul 31, 2025
@XuehaiPan
[pytree] simplify public API exposition with __module__
afe513d 
ghstack-source-id: 17d0670
Pull Request resolved: #148328
@XuehaiPan
Update
ac991d3 
[ghstack-poisoned]
XuehaiPan added a commit that referenced this pull request Aug 8, 2025
@XuehaiPan
[pytree] simplify public API exposition with __module__
4b6e418 
ghstack-source-id: 744658b
Pull Request resolved: #148328
XuehaiPan added a commit that referenced this pull request Aug 8, 2025
@XuehaiPan
[pytree] simplify public API exposition with __module__
ea0679d 
ghstack-source-id: 744658b
Pull Request resolved: #148328
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@zou3519 zou3519 Awaiting requested review from zou3519

@avikchaudhuri avikchaudhuri Awaiting requested review from avikchaudhuri avikchaudhuri is a code owner

@tugsbayasgalan tugsbayasgalan Awaiting requested review from tugsbayasgalan tugsbayasgalan is a code owner

@zhxchen17 zhxchen17 Awaiting requested review from zhxchen17 zhxchen17 is a code owner

@ydwu4 ydwu4 Awaiting requested review from ydwu4 ydwu4 is a code owner

@angelayi angelayi Awaiting requested review from angelayi angelayi is a code owner

@albanD albanD Awaiting requested review from albanD

Assignees
No one assigned
Labels
ci-test-showlocals Show local variables on test failures ciflow/inductor ciflow/trunk Trigger trunk jobs on your pull request module: dynamo module: pytree open source topic: not user facing topic category
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
3 participants
@XuehaiPan @albanD @pytorchbot