8000 [pytree] simplify public API exposition with `__module__` by XuehaiPan · Pull Request #148328 · pytorch/pytorch · GitHub
[go: up one dir, main page]
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 70 commits into
base: gh/XuehaiPan/253/base
Choose a base branch
from
Open

[pytree] simplify public API exposition with __module__ #148328

wants to merge 70 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.

✅ No Failures

As of commit 7e34dae with merge base cb5f31a (image):
💚 Looks good so far! There are no failures yet. 💚

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

This was referenced Mar 3, 2025
Open
Open
XuehaiPan added a commit that referenced this pull request Mar 3, 2025
@XuehaiPan
[pytree] simplify public API exposition with __module__
58d24cd 
ghstack-source-id: bfd087a
Pull Request resolved: #148328
@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 added a commit that referenced this pull request Mar 3, 2025
@XuehaiPan
[pytree] simplify public API exposition with __module__
23a52e0 
ghstack-source-id: c42fd59
Pull Request resolved: #148328
@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 that referenced this pull request Mar 3, 2025
@XuehaiPan
[pytree] simplify public API exposition with __module__
32ef5dc 
ghstack-source-id: 35e6bba
Pull Request resolved: #148328
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 added a commit that referenced this pull request Mar 3, 2025
@XuehaiPan
[pytree] simplify public API exposition with __module__
5f47472 
ghstack-source-id: ad8946c
Pull Request resolved: #148328
8000
@XuehaiPan
Update
085e755 
[ghstack-poisoned]
XuehaiPan added a commit that referenced this pull request Mar 3, 2025
@XuehaiPan
[pytree] simplify public API exposition with __module__
6080b70 
ghstack-source-id: 9b6b12d
Pull Request resolved: #148328
@XuehaiPan
Update
f1b7f65 
[ghstack-poisoned]
XuehaiPan added a commit that referenced this pull request Mar 3, 2025
@XuehaiPan
[pytree] simplify public API exposition with __module__
3dec46e 
ghstack-source-id: 9a99e58
Pull Request resolved: #148328
XuehaiPan added a commit that referenced this pull request May 1, 2025
@XuehaiPan
[pytree] simplify public API exposition with __module__
e2e71d3 
ghstack-source-id: 49a0b47
Pull Request resolved: #148328
@XuehaiPan
Update
c86c47f 
[ghstack-poisoned]
XuehaiPan added a commit that referenced this pull request May 1, 2025
@XuehaiPan
[pytree] simplify public API exposition with __module__
97ee705 
ghstack-source-id: 292b135
Pull Request resolved: #148328
@XuehaiPan
Update
5ece761 
[ghstack-poisoned]
XuehaiPan added a commit that referenced this pull request May 1, 2025
@XuehaiPan
[pytree] simplify public API exposition with __module__
8e40a8e 
ghstack-source-id: 1614e50
Pull Request resolved: #148328
@XuehaiPan XuehaiPan mentioned this pull request May 1, 2025
Open
XuehaiPan added a commit that referenced this pull request May 1, 2025
@XuehaiPan
[pytree] simplify public API exposition with __module__
f6c0590 
ghstack-source-id: 1614e50
Pull Request resolved: #148328
@XuehaiPan
Update
eb075c0 
[ghstack-poisoned]
XuehaiPan added a commit that referenced this pull request May 1, 2025
@XuehaiPan
[pytree] simplify public API exposition with __module__
3190325 
ghstack-source-id: 71a0900
Pull Request resolved: #148328
XuehaiPan added a commit to XuehaiPan/pytorch that referenced this pull request May 2, 2025
@XuehaiPan
[pytree] simplify public API exposition with __module__
1a1f5c3 
ghstack-source-id: 71a0900
Pull Request resolved: pytorch#148328
@XuehaiPan
Update
66ef529 
[ghstack-poisoned]
XuehaiPan added a commit that referenced this pull request May 2, 2025
@XuehaiPan
[pytree] simplify public API exposition with __module__
c3f5875 
ghstack-source-id: 6d1e47c
Pull Request resolved: #148328
@XuehaiPan
Update
11501ce 
[ghstack-poisoned]
XuehaiPan added a commit that referenced this pull request May 2, 2025
@XuehaiPan
[pytree] simplify public API exposition with __module__
0c73f77 
ghstack-source-id: 722fcd3
Pull Request resolved: #148328
@XuehaiPan
Update
ece5408 
[ghstack-poisoned]
XuehaiPan added a commit that referenced this pull request May 2, 2025
@XuehaiPan
[pytree] simplify public API 8000 exposition with __module__
068bcc2 
ghstack-source-id: a4bfc6a
Pull Request resolved: #148328
XuehaiPan added a commit to XuehaiPan/pytorch that referenced this pull request May 2, 2025
@XuehaiPan
[pytree] simplify public API exposition with __module__
0286730 
ghstack-source-id: a4bfc6a
Pull Request resolved: pytorch#148328
@XuehaiPan
Update
4eaaa56 
[ghstack-poisoned]
XuehaiPan added a commit that referenced this pull request May 2, 2025
@XuehaiPan
[pytree] simplify public API exposition with __module__
f21b20b 
ghstack-source-id: beeb916
Pull Request resolved: #148328
@XuehaiPan
Update
0c1cda4 
[ghstack-poisoned]
XuehaiPan added a commit that referenced this pull request May 3, 2025
@XuehaiPan
[pytree] simplify public API exposition with __module__
2c4583c 
ghstack-source-id: 7c68572
Pull Request resolved: #148328
XuehaiPan added a commit to XuehaiPan/pytorch that referenced this pull request May 3, 2025
@XuehaiPan
[pytree] simplify public API exposition with __module__
db80b95 
ghstack-source-id: 7c68572
Pull Request resolved: pytorch#148328
XuehaiPan added a commit to XuehaiPan/pytorch that referenced this pull request May 5, 2025
@XuehaiPan
[pytree] simplify public API exposition with __module__
b680bfd 
ghstack-source-id: 7c68572
Pull Request resolved: pytorch#148328
@XuehaiPan
Update
e45acce 
[ghstack-poisoned]
XuehaiPan added a commit that referenced this pull request May 8, 2025
@XuehaiPan
[pytree] simplify public API exposition with __module__
b7bf879 
ghstack-source-id: 9e5ba4a
Pull Request resolved: #148328
XuehaiPan added a commit to XuehaiPan/pytorch that referenced this pull request May 11, 2025
@XuehaiPan
[pytree] simplify public API exposition with __module__
87bcb73 
ghstack-source-id: 9e5ba4a
Pull Request resolved: pytorch#148328
@XuehaiPan
Update
c06df34 
[ghstack-poisoned]
9E88
XuehaiPan added a commit that referenced this pull request May 14, 2025
@XuehaiPan
[pytree] simplify public API exposition with __module__
d4c4b84 
ghstack-source-id: 031e10f
Pull Request resolved: #148328
@XuehaiPan
Update
7e34dae 
[ghstack-poisoned]
XuehaiPan added a commit that referenced this pull request May 16, 2025
@XuehaiPan
[pytree] simplify public API exposition with __module__
732678b 
ghstack-source-id: ab77d21
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
0