• 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
test_ir_backward.py 3.9 KB