- 11 8月, 2023 1 次提交
-
-
由 Charles-hit 提交于
* [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 * add test for mean vjp and support has_vjp function * fix call_vjp * polish code * add primitive ops set for backend * add vjp test for tanh_ * fix inference CI * fix inference ci * modify fluid cmake * remove useless deps * add cmake * fix comment * fix test * polish code * modify backward stop_gradients * modify static_backend.cc * remove useless code --------- Co-authored-by: Ncxxly <chenxx_id@163.com> Co-authored-by: Nzhangbo9674 <zhangbo54@baidu.com>
-
- 08 8月, 2023 1 次提交
-
-
由 Charles-hit 提交于
* [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 * add test for mean vjp and support has_vjp function * fix call_vjp * polish code * add primitive ops set for backend * add vjp test for tanh_ * fix inference CI * fix inference ci * modify fluid cmake * remove useless deps * add cmake --------- Co-authored-by: Ncxxly <chenxx_id@163.com> Co-authored-by: Nzhangbo9674 <zhangbo54@baidu.com>
-