-
Notifications
You must be signed in to change notification settings - Fork 25.7k
[TensorExpr] Kernel: make prim::ConstantChunk handled like other ops. #65549
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
Conversation
Previously it had a special handling, with this change it follows the same mechanism as other ops. [ghstack-poisoned]
🔗 Helpful links
💊 CI failures summary and remediationsAs of commit 599606c (more details on the Dr. CI page):
🕵️ 2 new failures recognized by patternsThe following CI failures do not appear to be due to upstream breakages:
|
|
@ZolotukhinM has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator. |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
torch/csrc/jit/tensorexpr/kernel.cpp
Outdated
| if (op == prim::ConstantChunk) { | ||
| auto const& n = v->node(); | ||
| argInputs.push_back(toArg(inputs[0])); | ||
| argInputs.push_back((int64_t)v->offset()); |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
"static_cast<int64_t>" instead of the traditional cast.
… other ops." Previously it had a special handling, with this change it follows the same mechanism as other ops. Differential Revision: [D31148924](https://our.internmc.facebook.com/intern/diff/D31148924) [ghstack-poisoned]
|
@ZolotukhinM has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator. |
… other ops." Previously it had a special handling, with this change it follows the same mechanism as other ops. Differential Revision: [D31148924](https://our.internmc.facebook.com/intern/diff/D31148924) [ghstack-poisoned]
|
@ZolotukhinM has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator. |
|
@ZolotukhinM has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator. |
… other ops." Previously it had a special handling, with this change it follows the same mechanism as other ops. Differential Revision: [D31148924](https://our.internmc.facebook.com/intern/diff/D31148924) [ghstack-poisoned]
|
@ZolotukhinM has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator. |
… other ops." Previously it had a special handling, with this change it follows the same mechanism as other ops. Differential Revision: [D31148924](https://our.internmc.facebook.com/intern/diff/D31148924) [ghstack-poisoned]
|
@ZolotukhinM has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator. |
… other ops." Previously it had a special handling, with this change it follows the same mechanism as other ops. Differential Revision: [D31148924](https://our.internmc.facebook.com/intern/diff/D31148924) [ghstack-poisoned]
|
@ZolotukhinM has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator. |
Stack from ghstack:
Previously it had a special handling, with this change it follows the
same mechanism as other ops.
Differential Revision: D31148924