pytorch / ao

PyTorch native quantization and sparsity for training and inference
BSD 3-Clause "New" or "Revised" License
1.6k stars 179 forks source link

[float8] Allow specifying arbitrary dtype for each tensor #1326

Open lw opened 3 days ago

lw commented 3 days ago

Stack from ghstack (oldest at bottom):

pytorch-bot[bot] commented 3 days ago

:link: Helpful Links

:test_tube: See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/ao/1326

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

:heavy_exclamation_mark: 1 Active SEVs

There are 1 currently active SEVs. If your PR is affected, please view them below:

:white_check_mark: No Failures

As of commit b9672f55eedf0ad65d55289690ee9ade4381fe7f with merge base 06ad55acb0d034a4e98e82a9eeddbd41d4d94b31 (image): :green_heart: Looks good so far! There are no failures yet. :green_heart:

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