• X
    【new ir】modify builtin_split bug (#56463) · 6de9a8d3
    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
    
    * remove useless deps
    
    * add cmake
    
    * modify block.ops
    
    * modify test
    
    * fix conflict
    
    * reply review comments
    
    * reply review comments
    
    * pulish code
    
    * fix comment
    
    * fix test
    
    * polish code
    
    * modify backward stop_gradients
    
    * modify static_backend.cc
    
    * refactor grad_op
    
    * support add and add_inplace vjp
    
    * remove useless code
    
    * remove useless code
    
    * remove cout
    
    * modify add_n
    
    * modify add_n with add_vjp test
    
    * modify add_n with add_vjp test
    
    * fix conflict and concat call_vjp
    
    * modify backward test
    
    * Add more gen api
    
    * modify split kernel pass
    
    * modify concat api
    
    * modify builtin split bug
    
    * delete vlog
    
    ---------
    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>
    Co-authored-by: N0x45f <wangzhen45@baidu.com>
    6de9a8d3
builtin_op.cc 10.2 KB