• X
    [NewIR]Add builtin.split op (#56431) · 14393611
    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
    
    * add builtin split op
    
    ---------
    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>
    14393611
builtin_op.cc 9.8 KB