Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "70d7486de5e7a52a4744244d023b866d46252889"
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_autograd.py 9.81 KB