-
Notifications
You must be signed in to change notification settings - Fork 6k
enable deterministic in bnb 4 bit tests #11738
New issue
Have a question about this project? # for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “#”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? # to your account
base: main
Are you sure you want to change the base?
Conversation
Signed-off-by: jiqing-feng <jiqing.feng@intel.com>
Hi @sayakpaul , could you please review this PR? Thanks! |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Should it not be propagated to test_mixed_int8.py
as well?
Right, fixed. |
Hold on for a while, need to recheck if it works. |
Hi @sayakpaul . I have verified this PR on both |
Signed-off-by: jiqing-feng <jiqing.feng@intel.com>
The motivation is that I can pass all tests when running
pytest -rA tests/quantization/bnb/test_4bit.py
, but failed onpytest -rA tests/quantization/bnb/test_4bit.py::ExtendedSerializationTest
. That's because running all tests will trigger torch.use_deterministic_algorithms(True), and it will have impact on all the following tests. So the correct way should be calling this function at the setup stage.