• Mohit Sharma's avatar
    Update ROCM libs and improvements (#2579) · f9e561ec
    Mohit Sharma authored
    * style
    
    * update torch
    
    * ix issues
    
    * fix clone
    
    * revert mkl
    
    * added custom PA
    
    * style
    
    * fix style
    
    * style
    
    * hide env vart
    
    * fix mixtral model
    
    * add skinny kernel and merge fixes
    
    * fixed style
    
    * fix issue for sliding window models
    
    * addressed review comments
    
    * fix import
    
    * improved error messag
    
    * updated default value
    
    * remove import
    
    * fix imports after rebase
    
    * float16 dep
    
    * improve dockerfile
    
    * cleaned dockerfile
    f9e561ec
linear.py 3.71 KB