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.

Sign up for GitHub

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

[tensor wrapper subclass] Add support for torchao.float8 mlp #1585

Draft
wants to merge 1 commit into
base: tensor_subclass_2
Choose a base branch
from

updates for MLP with `torchao.float8`

8435406
Select commit
Loading
Failed to load commit list.
Draft

[tensor wrapper subclass] Add support for torchao.float8 mlp #1585

updates for MLP with `torchao.float8`
8435406
Select commit
Loading
Failed to load commit list.
Azure Pipelines / lightning-thunder (GPUs) (testing ubuntu22.04 | cuda 12.1 | python 3.10 | torch 2.5.1 | regular) succeeded Jan 2, 2025 in 31m 42s

testing ubuntu22.04 | cuda 12.1 | python 3.10 | torch 2.5.1 | regular succeeded

0 errors / 0 warnings

View more details on Azure Pipelines