ao icon indicating copy to clipboard operation
ao copied to clipboard

[sparse] Migrate Float8SemiSparseTensor off of AQT

Open jcaip opened this issue 1 month ago • 2 comments

This PR migrates Float8DynamicActivationFloat8SemiSparseWeighConfig off of using the AQT CutlassSemiSparseLayout subclass.

The old AQT flow can still be used by passing version=1 into the config

Testing:

pytest test/quantization/quantize_/workflows/float8/test_float8_semi_sparse_tensor.py

jcaip avatar Nov 20 '25 18:11 jcaip

:link: Helpful Links

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

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

:x: 1 New Failure, 1 Unrelated Failure

As of commit 2298c3c0bcdb533006bf995162cfcd65aec48bad with merge base 1f9bfd7ebf054c616543052c5767a8e6972cbdf8 (image):

NEW FAILURE - The following job has failed:

BROKEN TRUNK - The following job failed but were present on the merge base:

👉 Rebase onto the `viable/strict` branch to avoid these failures

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

pytorch-bot[bot] avatar Nov 20 '25 18:11 pytorch-bot[bot]

@jcaip has imported this pull request. If you are a Meta employee, you can view this in D87560869.

meta-codesync[bot] avatar Nov 20 '25 18:11 meta-codesync[bot]

cc @RandySheriff if you want to review the changes, couldn't add you as a reviewer for some reason.

jcaip avatar Dec 18 '25 05:12 jcaip

I think we should split the tensor into 2, one for each packing format

jerryzh168 avatar Dec 18 '25 22:12 jerryzh168