-
Notifications
You must be signed in to change notification settings - Fork 9.7k
Added NNC lowering example #884
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
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.
Looks pretty good
# | ||
# NNC time: 0.0066373348236083984 | ||
# PyTorch time 0.025979042053222656 | ||
# Static Runtime time 0.011004209518432617 |
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.
Maybe put your lscpu
in a gist so people know what machine these results come from
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.
Also, is Static Runtime public at all? If not, remove references to it
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.
Bram has made references to it on the PyTorch dev discussion forum, so I assume it's public.
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.
I put my CPU version
Added NNC lowering example
Taken from pytorch/pytorch#51130
Probably still need to add comments, but this is an annoying amount of code for a meaningfully commented example. Most of the lowering rules are obvious (about what they do), but it's still structured strangely.