• H
    Move bn to pten (#39347) · ebd0f512
    hong 提交于
    * add bn cpu version; test=develop
    
    * move batch norm to pten
    
    * move batch norm to pten; test=develop
    
    * fix bug; test=develop
    
    * fix func::tranpose depend bug; test=develop
    
    * fix compile bugs; test=develop
    
    * fix use_op batch_norm bug; test=develop
    
    * fix cudnn bn add relu test; test=develop
    
    * fix pten context build and double grad bug; test= develop
    
    * remve useless code; test=develop
    
    * add batch norm gpu fp16 support; test=develop
    
    * fix test bn op bug; test=develop
    
    * remove output dtype set; test=develop
    
    * fix bug; test=develop
    
    * fix bug; test=develop
    
    * fix applay pass to program bug; test=develop
    
    * revert to develop; test=develop
    
    * fix rocm bug; test=develop
    
    * revert operator to develop; test=develop
    
    * fix pre_commit; test=develop
    
    * fix statci check error; test=develop
    
    * resolve conflict; test=develop
    
    * ana batch norm bug;
    
    * revert batch norm op
    
    * resolve conlict
    
    * fix nan inf and speed bug; test=develop
    
    * fix bug; test=develop
    
    * fix error; test=develop
    
    * test expand op; test=develop
    
    * fix bug; test=develop
    
    * resolve confilct
    
    * resolve confilct; test=develop
    
    * polish code; test=develop
    
    * polish code; test=develop
    
    * change mutable data to ctx alloc; test=develop
    
    * make format same with ci; test=develop
    
    * fix format error with ci; test=develop
    ebd0f512
batch_norm_op.cu 1.4 KB