Enable FSDP Test in CI - Githubissues

pytorch / ao

Native PyTorch library for quantization and sparsity

https://pytorch.org/ao

BSD 3-Clause "New" or "Revised" License

307 stars 49 forks source link

Closed msaroufim closed 1 month ago

msaroufim commented 1 month ago

Repeat of https://github.com/pytorch/ao/pull/206 but against main instead of my own branch 🤦

facebook-github-bot commented 1 month ago

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

weifengpy commented 1 month ago

i had the same issue of not finding hqq before. curious of the solution

msaroufim commented 1 month ago

Taking a look

msaroufim commented 1 month ago

Ok it worked just needed better import skips - waiting for @jerryzh168's ship it work to land before landing this