- 08 9月, 2023 1 次提交
-
-
由 cyber-pioneer 提交于
* test prim custom vjp in New IR * polish gelu_grad
-
- 07 9月, 2023 1 次提交
-
-
由 Chen Zhiyang 提交于
* fix StrAttribute vjp gen bug * add dropout to list * fix bug * fix bug
-
- 06 9月, 2023 1 次提交
-
-
由 Xianduo Li 提交于
* support new ir primitive operator in static operants * support more vjp code gen * support more vjp code gen * support more vjp code gen * use code gen * fix operants codegen * support more vjp code gen * Fix ci build error * set FLAGS_tensor_operants_mode to static in generated_vjp for testing * fix bugs * change the order of ops_name of divide_grad * replace FLAGS_enable_new_ir_in_executor by FLAGS_enable_new_ir_api in codegen and test_vjp_prim --------- Co-authored-by: NCharles-hit <wanghao107@baidu.com> Co-authored-by: N0x45f <wangzhen45@baidu.com>
-
- 05 9月, 2023 1 次提交
-
-
由 Charles-hit 提交于
-
- 01 9月, 2023 1 次提交
-
-
由 Charles-hit 提交于
* Fix attr type error like concat axis * Fix None input error * Fix intermediate output * support vjp code gen --------- Co-authored-by: N0x45f <wangzhen45@baidu.com>
-
- 23 8月, 2023 1 次提交
-
-
由 Charles-hit 提交于
-
- 22 8月, 2023 1 次提交
-
-
由 Charles-hit 提交于
* support ir api form prim * support ir api for prim * support vjp prim mode in new ir * remove useless code * remove useless code * auto code generator for primitive vjp methods * add vjp and backend manual and fix segment fault --------- Co-authored-by: Ncyber-pioneer <chenzhuo@tju.edu.cn> Co-authored-by: Ncxxly <chenxx_id@163.com>
-