Skip to content

GitLab

"docs/vscode:/vscode.git/clone" did not exist on "d220ee2f66659c828422cb55344973e64fc33a55"
Switch branch/tag
Find file BlameHistoryPermalink
  • Matthew Douglas's avatar
    Updates for device agnosticism (#1601) · 1088ec52
    Matthew Douglas authored 
    * Include device support tags for transformers multi-backend compatability; add xpu() and cpu() to Params4bit

* Make test suite more device-agnostic

* Additional device agnostic tests

* Additional device agnosticism for tests

* Add BNB_TEST_DEVICE env var to manually select device for unit tests

* Include device support tags for transformers multi-backend compatability; add xpu() and cpu() to Params4bit

* Make test suite more device-agnostic

* Additional device agnostic tests

* Additional device agnosticism for tests

* Add BNB_TEST_DEVICE env var to manually select device for unit tests

* Small bugfix for int8 test

* Exclude backward() from code coverage reports

* Params4bit: don't try to quantize when moving to meta device
    1088ec52
test_modules.py 18.8 KB