Skip to content

Revert "pin nightlies to deal with std::badalloc" #1283

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.

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

Merged
merged 25 commits into from
Nov 20, 2024
Merged

Conversation

jcaip
Copy link
Contributor

@jcaip jcaip commented Nov 14, 2024

Reverts #1256

Copy link

pytorch-bot bot commented Nov 14, 2024

🔗 Helpful Links

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

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

❗ 1 Active SEVs

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

✅ No Failures

As of commit d722cf2 with merge base 26648c2 (image):
💚 Looks good so far! There are no failures yet. 💚

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 Nov 14, 2024
@jerryzh168
Copy link
Contributor

@jcaip I think these changes can be included in #1278

@jcaip jcaip added the topic: not user facing Use this tag if you don't want this PR to show up in release notes label Nov 18, 2024
@jcaip jcaip force-pushed the revert-1256-jcaip-patch-5 branch from b40c08c to 66e11fc Compare November 18, 2024 16:24
@@ -70,7 +71,7 @@ def test_awq_loading(device, qdtype):

model_save_path = "awq_model.pth"
torch.save(m, model_save_path)
loaded_model = torch.load(model_save_path)
loaded_model = torch.load(model_save_path, assign=True)
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

assign=True is an arg for model.load_state_dict(...) I think, see https://pytorch.org/ao/stable/serialization.html

@malfet
Copy link
Contributor

malfet commented Nov 19, 2024

@jcaip do you know what caused it and what fixed it?

@jcaip
Copy link
Contributor Author

jcaip commented Nov 19, 2024

@malfet we switched over to the v2 jobs #1302 in this PR which fixed the badalloc error. Not sure about the root cause of the std::badalloc job but it was being triggerd on a jit test.

@jcaip jcaip merged commit 8bc9046 into main Nov 20, 2024
18 checks passed
sunjiweiswift pushed a commit to sunjiweiswift/ao that referenced this pull request Nov 25, 2024
* Revert "pin nightlies to deal with std::badalloc (pytorch#1256)"

This reverts commit 0e854ec.

* Update regression_test.yml

* Update regression_test.yml

* skip tests

* update

* work

* fix

* fix lint

* Update test_awq.py

* Update test_awq.py

* Update regression_test.yml

* Update regression_test.yml

* Update regression_test.yml

* Update regression_test.yml

* Update regression_test.yml

* Update test_awq.py

* Update test_awq.py

* Update test_awq.py

* Update test_awq.py

* Update test_awq.py

* Update test_awq.py

* Update test_awq.py

* Update test_awq.py
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: not user facing Use this tag if you don't want this PR to show up in release notes
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants