Skip to content

[inductor] More fixes on the keys of constants and signature dictionaries#135406

Closed
Jokeren wants to merge 8 commits intopytorch:mainfrom
Jokeren:keren/fix-constants
Closed

[inductor] More fixes on the keys of constants and signature dictionaries#135406
Jokeren wants to merge 8 commits intopytorch:mainfrom
Jokeren:keren/fix-constants

Conversation

@Jokeren
Copy link
Contributor

@Jokeren Jokeren commented Sep 7, 2024

@pytorch-bot
Copy link

pytorch-bot bot commented Sep 7, 2024

🔗 Helpful Links

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

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

✅ No Failures

As of commit a0527af with merge base 21a64d5 (image):
💚 Looks good so far! There are no failures yet. 💚

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

@Jokeren Jokeren changed the title [inductor] More fixes on [inductor] More fixes on the keys of constants and signature dictionaries Sep 7, 2024
@Jokeren
Copy link
Contributor Author

Jokeren commented Sep 7, 2024

@pytorchbot label "module: inductor"

@Jokeren
Copy link
Contributor Author

Jokeren commented Sep 7, 2024

@pytorchbot label "topic: not user facing"

@pytorch-bot pytorch-bot bot added the topic: not user facing topic category label Sep 7, 2024
@ezyang ezyang added the triaged This issue has been looked at a team member, and triaged and prioritized into an appropriate module label Sep 9, 2024
Copy link
Contributor

@shunting314 shunting314 left a comment

Choose a reason for hiding this comment

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

Would there be any backward compatibility issue?
i.e., if PyTorch is running with an older version of triton, with triton still handle the new metadata format properly?

@Jokeren
Copy link
Contributor Author

Jokeren commented Sep 9, 2024

Would there be any backward compatibility issue?
i.e., if PyTorch is running with an older version of triton, with triton still handle the new metadata format properly?

Are you asking about whether old triton could handle the new format? I think it happened about 6 months ago.

https://github.com/triton-lang/triton/blame/5aaddb8b497181744992face03a825173da4806c/python/triton/runtime/jit.py

jansel
jansel previously approved these changes Sep 10, 2024
@jansel
Copy link
Contributor

jansel commented Sep 10, 2024

Seems fine, though maybe we need some sort of triton version check.

@Jokeren
Copy link
Contributor Author

Jokeren commented Sep 10, 2024

I think there also need to be some tests on the triton side to check the key type of the dictionaries. I'll go add them soon.

@Jokeren
Copy link
Contributor Author

Jokeren commented Sep 11, 2024

Triton PR: triton-lang/triton#4693

@Jokeren
Copy link
Contributor Author

Jokeren commented Sep 11, 2024

Can someone help me kick off the CI again? Thanks

@Jokeren
Copy link
Contributor Author

Jokeren commented Sep 12, 2024

Is it good to merge? I probably don't have the permission to ask the pytorchbot to merge

@jansel
Copy link
Contributor

jansel commented Sep 13, 2024

@pytorchbot merge

@pytorch-bot pytorch-bot bot added the ciflow/trunk Trigger trunk jobs on your pull request label Sep 13, 2024
@pytorchmergebot
Copy link
Collaborator

Merge started

Your change will be merged once all checks pass (ETA 0-4 Hours).

Learn more about merging in the wiki.

Questions? Feedback? Please reach out to the PyTorch DevX Team

Advanced Debugging
Check the merge workflow status
here

@jeanschmidt
Copy link
Contributor

@pytorchbot revert -m "Reverting as it is breaking triton_mtia internal signals @jansel could you have a look and help get those changes merged?" -c ghfirst

@pytorchmergebot
Copy link
Collaborator

@pytorchbot successfully started a revert job. Check the current status here.
Questions? Feedback? Please reach out to the PyTorch DevX Team

pytorchmergebot added a commit that referenced this pull request Sep 16, 2024
…re` dictionaries (#135406)"

This reverts commit e54b559.

Reverted #135406 on behalf of https://github.com/jeanschmidt due to Reverting as it is breaking triton_mtia internal signals @jansel could you have a look and help get those changes merged? ([comment](#135406 (comment)))
@pytorchmergebot
Copy link
Collaborator

@Jokeren your PR has been successfully reverted.

@pytorch-bot pytorch-bot bot dismissed jansel’s stale review September 16, 2024 17:58

This PR was reopened (likely due to being reverted), so your approval was removed. Please request another review.

@jansel
Copy link
Contributor

jansel commented Sep 17, 2024

@jfix71 how hard would it be to update MTIA Triton to use this new API?

@jfix71
Copy link
Contributor

jfix71 commented Sep 17, 2024

@jansel I think it should be a pretty reasonable change, but we'd want to import the PR and update the diff internally to tag on our changes before landing, if that's a possibility. What's the standard workflow expected for these sorts of changes, so we don't have to fix forward?

@Myrthan
Copy link

Myrthan commented Sep 18, 2024

@jfix71 how hard would it be to update MTIA Triton to use this new API?

We fixed MTIA yesterday check D62887579and D62896747. Changes should be landed in fresh master. Could you please try recreate fbsource Phabricator diff out of this pr? It should be all green for you, I tested it on your changes from this Github PR D62888678.

@facebook-github-bot
Copy link
Contributor

@Myrthan has imported this pull request. If you are a Meta employee, you can view this diff on Phabricator.

Chao1Han pushed a commit to Chao1Han/pytorch that referenced this pull request Sep 20, 2024
…ionaries (pytorch#135406)

Previous PR forgets to change two other places that also create `constants` and `signature`. pytorch#135170

Pull Request resolved: pytorch#135406
Approved by: https://github.com/jansel
Chao1Han pushed a commit to Chao1Han/pytorch that referenced this pull request Sep 20, 2024
…re` dictionaries (pytorch#135406)"

This reverts commit e54b559.

Reverted pytorch#135406 on behalf of https://github.com/jeanschmidt due to Reverting as it is breaking triton_mtia internal signals @jansel could you have a look and help get those changes merged? ([comment](pytorch#135406 (comment)))
@Myrthan
Copy link

Myrthan commented Sep 20, 2024

@jfix71 cause there was no update for some time, do you think it is wise to use "@ pytorchbot merge"? will the bot merge it on top of fresh changes? Or this PR has to be rebased first? Otherwise I have to recreate the new github PR, but I don't want to take the ownership from @Jokeren

@Jokeren
Copy link
Contributor Author

Jokeren commented Sep 20, 2024

@jfix71 cause there was no update for some time, do you think it is wise to use "@ pytorchbot merge"? will the bot merge it on top of fresh changes? Or this PR has to be rebased first? Otherwise I have to recreate the new github PR, but I don't want to take the ownership from @Jokeren

Thanks. It's fine to create a new PR. In terms of tracing issues in the future if needed, you can just add me as a co-author in the PR so that I will be notified.

@Myrthan
Copy link

Myrthan commented Sep 23, 2024

@jfix71 cause there was no update for some time, do you think it is wise to use "@ pytorchbot merge"? will the bot merge it on top of fresh changes? Or this PR has to be rebased first? Otherwise I have to recreate the new github PR, but I don't want to take the ownership from @Jokeren

Thanks. It's fine to create a new PR. In terms of tracing issues in the future if needed, you can just add me as a co-author in the PR so that I will be notified.

Great, let me do this right now.

@Myrthan
Copy link

Myrthan commented Sep 24, 2024

@Jokeren I recreated it on fresh main branch here #136514 I tried adding you as coauthor, not sure if that worked, let me know. I will need approve and we can merge it.

@Jokeren
Copy link
Contributor Author

Jokeren commented Sep 24, 2024

Thanks. I think it should work

@Myrthan
Copy link

Myrthan commented Sep 25, 2024

#136514 has been merged, we can abandon this PR I believe. Thank you!

@Jokeren
Copy link
Contributor Author

Jokeren commented Sep 25, 2024

Great. Thanks!

@Jokeren Jokeren closed this Sep 25, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

ciflow/trunk Trigger trunk jobs on your pull request Merged module: inductor open source Reverted topic: not user facing topic category triaged This issue has been looked at a team member, and triaged and prioritized into an appropriate module

Projects

None yet

Development

Successfully merging this pull request may close these issues.

10 participants