1. 22 8月, 2023 6 次提交
  2. 21 8月, 2023 31 次提交
  3. 19 8月, 2023 2 次提交
  4. 18 8月, 2023 1 次提交
    • 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