Skip to content

Conversation

@vkuzo
Copy link
Contributor

@vkuzo vkuzo commented Jun 3, 2025

Summary:

Update the float8 training example code snippet to include time measurement that properly excludes torch.compile one-time warmup. Also, use larger shapes to demonstrate speedup from float8.

Test Plan:

copy-paste the snippet and run it, it works. Commenting out float8 shows a slowdown, as expected.

Reviewers:

Subscribers:

Tasks:

Tags:

@pytorch-bot
Copy link

pytorch-bot bot commented Jun 3, 2025

🔗 Helpful Links

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

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

❌ 1 New Failure

As of commit 0eb0616 with merge base dd43f16 (image):

NEW FAILURE - The following job has failed:

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

@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 Jun 3, 2025
Summary:

Update the float8 training example code snippet to include time
measurement that properly excludes torch.compile one-time warmup. Also,
use larger shapes to demonstrate speedup from float8.

Test Plan:

copy-paste the snippet and run it, it works. Commenting out float8 shows
a slowdown, as expected.

Reviewers:

Subscribers:

Tasks:

Tags:
@vkuzo vkuzo added the topic: documentation Use this tag if this PR adds or improves documentation label Jun 4, 2025
@vkuzo vkuzo merged commit 152a8e3 into main Jun 4, 2025
19 of 20 checks passed
liangel-02 pushed a commit that referenced this pull request Aug 25, 2025
Summary:

Update the float8 training example code snippet to include time
measurement that properly excludes torch.compile one-time warmup. Also,
use larger shapes to demonstrate speedup from float8.

Test Plan:

copy-paste the snippet and run it, it works. Commenting out float8 shows
a slowdown, as expected.

Reviewers:

Subscribers:

Tasks:

Tags:
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: documentation Use this tag if this PR adds or improves documentation

Projects

None yet

Development

Successfully merging this pull request may close these issues.

5 participants