• Anton Lozhkov's avatar
    Add Apple M1 tests (#796) · cca59ce3
    Anton Lozhkov authored
    
    
    * [CI] Add Apple M1 tests
    
    * setup-python
    
    * python build
    
    * conda install
    
    * remove branch
    
    * only 3.8 is built for osx-arm
    
    * try fetching prebuilt tokenizers
    
    * use user cache
    
    * update shells
    
    * Reports and cleanup
    
    * -> MPS
    
    * Disable parallel tests
    
    * Better naming
    
    * investigate worker crash
    
    * return xdist
    
    * restart
    
    * num_workers=2
    
    * still crashing?
    
    * faulthandler for segfaults
    
    * faulthandler for segfaults
    
    * remove restarts, stop on segfault
    
    * torch version
    
    * change installation order
    
    * Use pre-RC version of PyTorch.
    
    To be updated when it is released.
    
    * Skip crashing test on MPS, add new one that works.
    
    * Skip cuda tests in mps device.
    
    * Actually use generator in test.
    
    I think this was a typo.
    
    * make style
    Co-authored-by: default avatarPedro Cuenca <pedro@huggingface.co>
    cca59ce3
test_layers_utils.py 13.5 KB