• Ella Charlaix's avatar
    Add deepspeed test to amd scheduled CI (#27633) · 39acfe84
    Ella Charlaix authored
    
    
    * add deepspeed scheduled test for amd
    
    * fix image
    
    * add dockerfile
    
    * add comment
    
    * enable tests
    
    * trigger
    
    * remove trigger for this branch
    
    * trigger
    
    * change runner env to trigger the docker build image test
    
    * use new docker image
    
    * remove test suffix from docker image tag
    
    * replace test docker image with original image
    
    * push new image
    
    * Trigger
    
    * add back amd tests
    
    * fix typo
    
    * add amd tests back
    
    * fix
    
    * comment until docker image build scheduled test fix
    
    * remove deprecated deepspeed build option
    
    * upgrade torch
    
    * update docker & make tests pass
    
    * Update docker/transformers-pytorch-deepspeed-amd-gpu/Dockerfile
    
    * fix
    
    * tmp disable test
    
    * precompile deepspeed to avoid timeout during tests
    
    * fix comment
    
    * trigger deepspeed tests with new image
    
    * comment tests
    
    * trigger
    
    * add sklearn dependency to fix slow tests
    
    * enable back other tests
    
    * final update
    
    ---------
    
    Co-authored-by: default avatarFelix Marty <felix@hf.co>
    Co-authored-by: default avatarFélix Marty <9808326+fxmarty@users.noreply.github.com>
    Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
    39acfe84