- 08 9月, 2023 1 次提交
-
-
由 cyber-pioneer 提交于
* test prim custom vjp in New IR * polish gelu_grad
-
- 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>
-
- 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>
-
- 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>
-