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

[2/x] float8 cleanup: move roofline utils to testing #1780

Merged
merged 3 commits into from
Feb 26, 2025
Merged

Conversation

vkuzo
Copy link
Contributor

@vkuzo vkuzo commented Feb 25, 2025

Summary:

Moves torchao/float8/roofline_utils.py to torchao/testing/float8, as
this is only needed for tests/benchmarks.

Test Plan:

python benchmarks/float8/float8_roofline.py ~/local/tmp/20250223_test

Reviewers:

Subscribers:

Tasks:

Tags:

[ghstack-poisoned]
[ghstack-poisoned]
@vkuzo
Copy link
Contributor Author

vkuzo commented Feb 25, 2025

Stack from ghstack (oldest at bottom):

Copy link

pytorch-bot bot commented Feb 25, 2025

🔗 Helpful Links

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

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

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

vkuzo added a commit that referenced this pull request Feb 25, 2025
Summary:

Moves `torchao/float8/roofline_utils.py` to `torchao/testing/float8`, as
this is only needed for tests/benchmarks.

Test Plan:

```
python benchmarks/float8/float8_roofline.py ~/local/tmp/20250223_test
```

Reviewers:

Subscribers:

Tasks:

Tags:

ghstack-source-id: 92b3a33aa8936bafed303bd954408c5e8ee6f632
ghstack-comment-id: 2683402019
Pull Request resolved: #1780
@facebook-github-bot facebook-github-bot added the CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. label Feb 25, 2025
@vkuzo vkuzo added the topic: not user facing Use this tag if you don't want this PR to show up in release notes label Feb 25, 2025
[ghstack-poisoned]
@vkuzo vkuzo changed the base branch from gh/vkuzo/41/head to main February 26, 2025 17:28
vkuzo added a commit that referenced this pull request Feb 26, 2025
Summary:

Moves `torchao/float8/roofline_utils.py` to `torchao/testing/float8`, as
this is only needed for tests/benchmarks.

Test Plan:

```
python benchmarks/float8/float8_roofline.py ~/local/tmp/20250223_test
```

Reviewers:

Subscribers:

Tasks:

Tags:

ghstack-source-id: 8bfadd4a6a6e96d0d2c3175aa94279ac015fe18e
ghstack-comment-id: 2683402019
Pull Request resolved: #1780
@vkuzo vkuzo merged commit d00ee41 into main Feb 26, 2025
23 of 24 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. topic: not user facing Use this tag if you don't want this PR to show up in release notes
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants