Skip to content
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

Add pplbench beanmachine model #869

Closed
wants to merge 17 commits into from

Conversation

agunapal
Copy link
Contributor

@agunapal agunapal commented Apr 15, 2022

Code for integrating beanchmachine pplbench with torchbench

Outputs:

python run.py pplbench_beanmachine -d cpu -t eval

Running eval method from pplbench_beanmachine on cpu in eager mode.
CPU Total Wall Time: 98.262 milliseconds

python run.py pplbench_beanmachine -d cpu -t train

Running train method from pplbench_beanmachine on cpu in eager mode.
CPU Total Wall Time: 16629.886 milliseconds

@agunapal agunapal changed the title Agunapal/add beanmachine pplbench Add pplbench beanmachine model Apr 18, 2022
Copy link
Contributor

@xuzhao9 xuzhao9 left a comment

Choose a reason for hiding this comment

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

Please fix the gen_inputs() function to comply with the interface.

torchbenchmark/__init__.py Outdated Show resolved Hide resolved
Copy link
Contributor

@xuzhao9 xuzhao9 left a comment

Choose a reason for hiding this comment

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

LGTM

Copy link
Contributor

@xuzhao9 xuzhao9 left a comment

Choose a reason for hiding this comment

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

LGTM. Thanks!

@xuzhao9 xuzhao9 requested review from anijain2305 and Chillee April 19, 2022 20:56
@facebook-github-bot
Copy link
Contributor

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

@facebook-github-bot
Copy link
Contributor

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

@agunapal agunapal force-pushed the agunapal/add_beanmachine_pplbench branch from e783c86 to 41b18a7 Compare April 21, 2022 00:52
@facebook-github-bot
Copy link
Contributor

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

1 similar comment
@facebook-github-bot
Copy link
Contributor

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

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