• 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_functional.py 55.7 KB