• J
    [Dist Pass] Amp Pass (#38764) · cc24427e
    JZ-LIANG 提交于
    * auto parallel sharding base
    
    * chmod
    
    * add unitest
    
    * set unitest cmake dist label
    
    * revise code according to rewiew
    
    * chmod
    
    * bugfix for grad_clip and param broadcast
    
    * chmod
    
    * update unitest
    
    * chmod
    
    * add clip
    
    * chmod
    
    * add amp pass
    
    * chmod
    
    * add unitest
    
    * remove grad update
    
    * fixed bug
    
    * fixed bug
    
    * fixed typose
    
    * fixed typoes
    cc24427e
auto_parallel_amp.py 32.3 KB