1. 09 8月, 2023 1 次提交
    • X
      【New IR】backward code of new ir (#55957) · 0399b39f
      xiaoguoguo626807 提交于
      * [prim][newir] add basic framework for primitive
      
      * support desctensor in new ir
      
      * add vjp interface
      
      * support vjp in new ir
      
      * support vjp in new ir
      
      * polish vjp interface
      
      * fix stop_gradients set
      
      * fix vjp dispatch
      
      * add comment
      
      * add vjp test for new ir
      
      * add test for tanh vjp
      
      * [prim][newir] add basic framework for primitive
      
      * support desctensor in new ir
      
      * support vjp in new ir
      
      * support vjp in new ir
      
      * polish vjp interface
      
      * fix stop_gradients set
      
      * fix vjp dispatch
      
      * add comment
      
      * add vjp test for new ir
      
      * add test for tanh vjp
      
      * add eager and static backend for warp lower level api
      
      * support call_vjp pybind
      
      * polish code and add test for vjp
      
      * remove useless code
      
      * polish code
      
      * remove useless code
      
      * support mean vjp
      
      * backward origin code
      
      * add test for mean vjp and support has_vjp function
      
      * fix call_vjp
      
      * polish code
      
      * add attrs and dtype interface
      
      * add primitive ops set for backend
      
      * fix compile bugs
      
      * fix some bugs
      
      * fix windows bugs
      
      * add vjp test for tanh_
      
      * fix inference CI
      
      * fix inference ci
      
      * modify fluid cmake
      
      * origin test of tanh and mean passed
      
      * fix conflict
      
      * modify stop_gradient
      
      * modify block.ops
      
      * modify test
      
      * fix conflict
      
      * reply review comments
      
      * reply review comments
      
      * pulish code
      
      ---------
      Co-authored-by: Ncxxly <chenxx_id@163.com>
      Co-authored-by: NCharles-hit <wanghao107@baidu.com>
      Co-authored-by: Nzhangbo9674 <zhangbo54@baidu.com>
      Co-authored-by: NYuanRisheng <yuanrisheng@baidu.com>
      0399b39f