- 18 8月, 2023 1 次提交
-
-
由 Charles-hit 提交于
* support ir api form prim * convert vector of int to intarray * support ir api for prim * support vjp prim mode in new ir * remove useless code * add test for prim * modify utils * remove useless code --------- Co-authored-by: Ncyber-pioneer <chenzhuo@tju.edu.cn>
-
- 16 8月, 2023 1 次提交
-
-
由 kangguangli 提交于
* fix inset get_parameter op bug * fix bug: insert only for parameters * fix bug: wrong idx in vector --------- Co-authored-by: Nzhangbo9674 <zhangbo54@baidu.com>
-
- 09 8月, 2023 1 次提交
-
-
由 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>
-
- 08 8月, 2023 1 次提交
-
-
由 cyber-pioneer 提交于
* Support Prim Forward in New IR * Fix test case * polish code * fix code * polish format * format code
-