Skip to content

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

Merged
merged 5 commits into from
Feb 6, 2021
Merged

Added NNC lowering example #884

merged 5 commits into from
Feb 6, 2021

Conversation

Chillee
Copy link
Contributor

@Chillee Chillee commented Feb 4, 2021

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.

@jamesr66a jamesr66a self-requested a review February 5, 2021 18:36
Copy link
Contributor

@jamesr66a jamesr66a left a 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
Copy link
Contributor

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

Copy link
Contributor

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

Copy link
Contributor Author

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.

Copy link
Contributor Author

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

@Chillee Chillee requested a review from jamesr66a February 6, 2021 00:48
@jamesr66a jamesr66a merged commit 1973c3c into pytorch:master Feb 6, 2021
YinZhengxun pushed a commit to YinZhengxun/mt-exercise-02 that referenced this pull request Mar 30, 2025
Added NNC lowering example
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants