Skip to content
This repository was archived by the owner on Aug 7, 2024. It is now read-only.

[8/x] make single linear profiling script work with Float8 scaling type#299

Closed
vkuzo wants to merge 1 commit intogh/vkuzo/18/basefrom
gh/vkuzo/18/head
Closed

[8/x] make single linear profiling script work with Float8 scaling type#299
vkuzo wants to merge 1 commit intogh/vkuzo/18/basefrom
gh/vkuzo/18/head

Conversation

@vkuzo
Copy link
Contributor

@vkuzo vkuzo commented Jul 2, 2024

Summary:

Makes `benchmarks/bench_linear_float8.py` support per-tensor scaling
configurations.

Verified that performance is as we expect

Test Plan:

paste of testing for delayed -> dynamic, changing the tensors one by
one:
https://gist.github.com/vkuzo/9e8f995e51ef16f483347c0f86bb0ac3

Reviewers:

Subscribers:

Tasks:

Tags:

[ghstack-poisoned]
@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 Jul 2, 2024
vkuzo added a commit that referenced this pull request Jul 2, 2024
Summary:

Makes `benchmarks/bench_linear_float8.py` support per-tensor scaling
configurations.

Verified that performance is as we expect

Test Plan:

paste of testing for delayed -> dynamic, changing the tensors one by
one:
https://gist.github.com/vkuzo/9e8f995e51ef16f483347c0f86bb0ac3

Reviewers:

Subscribers:

Tasks:

Tags:

ghstack-source-id: bb0c5ac
Pull Request resolved: #299
@vkuzo vkuzo requested a review from drisspg July 2, 2024 23:02
@vkuzo
Copy link
Contributor Author

vkuzo commented Jul 2, 2024

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

@facebook-github-bot
Copy link
Contributor

This pull request has been merged in 4fb0ada.

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.

Labels

CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. Merged

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants