1. 01 3月, 2023 1 次提交
    • Z
      add topk prim backward (#50679) · 296b3ff0
      zqw_1997 提交于
      * tmp gather vjp
      
      * support gather
      
      * remove useless code
      
      * fix compiling error
      
      * fix ut
      
      * add eager test
      
      * add eager test
      
      * add seed
      
      * small change
      
      * fix cpu error
      
      * fix transpose op compat
      
      * remove tensor index case
      
      * fix prim_cinn
      
      * small commit
      
      * add cumsum prim backward
      
      * small commit
      
      * skip aixs=None test case
      
      * fix op generante eror
      
      * fix static test error
      
      * remove unused code
      
      * fix static test error
      
      * small commit
      
      * skip cpu float16 test case
      
      * skip eager cpu cumsum float16 test case
      
      * add eager and static UT
      
      * fix ut
      
      * add composite backward rule
      
      * fix error
      
      * fix type error and format error
      
      * add try cpu+float16 test
      
      * fix test bugs
      
      * remove test for cpu+float16 and make y[0] be the grad arg
      
      * add cinn test
      
      * fix UT
      
      * fix the wrong dim of v in test cases
      
      * change y[0] to y[1] for grad in UT
      
      * reshape flatten out
      
      * Disable cinn single test
      
      * use scatter_nd_add
      
      * modify the reshape part of topk_grad
      
      * delete useless build file
      
      * to make the syntax right
      
      * modify bug
      
      * try use of put_along_axis
      
      * remove cinn test
      
      * reformat todo
      
      * add silu composite rule
      
      * fix code style.
      
      * add cinn test
      
      * fix composite grad maker code gen
      
      * add prim in cumsum op test
      
      * remove old test
      
      * fix typro
      
      * pass the static test
      
      * fix typro
      
      * modify optest and delete old test files
      
      * remove normal test_top_k_op test
      
      * fix typro
      
      * pass axis=None test case
      
      * buffer comment
      
      * for debug
      
      * add silu fp16 unit test.
      
      * add static guard
      
      * remove forward prim test
      
      * remove same name axis
      
      * modify the test_top_v2_op.py to pass all local tests
      
      * delete the useless testcase
      
      * fix mistake
      
      * add more testcases to test dtype16 and dtype32
      
      ---------
      Co-authored-by: NJiabinYang <360788950@qq.com>
      Co-authored-by: NGGBond8488 <857631483@qq.com>
      Co-authored-by: Nzxcd <228587199@qq.com>
      Co-authored-by: NCharles-hit <wanghao107@baidu.com>
      296b3ff0
  2. 28 2月, 2023 1 次提交
    • G
      add cumsum prim backward (#50565) · ca2b6095
      GGBond8488 提交于
      * add cumsum prim backward
      
      * skip aixs=None test case
      
      * fix op generante eror
      
      * fix static test error
      
      * remove unused code
      
      * fix static test error
      
      * skip cpu float16 test case
      
      * skip eager cpu cumsum float16 test case
      
      * add cinn test
      
      * reshape flatten out
      
      * Disable cinn single test
      
      * remove cinn test
      
      * reformat todo
      
      * add prim in cumsum op test
      
      * remove old test
      
      * fix typro
      
      * fix typro
      
      * fix typro
      
      * pass axis=None test case
      
      * remove forward prim test
      
      * remove same name axis
      ca2b6095
  3. 27 2月, 2023 1 次提交
  4. 24 2月, 2023 1 次提交
    • X
      【prim】Slice grad (#50771) · f6dea800
      xiaoguoguo626807 提交于
      * support prim test in OpTest
      
      * fix cmake
      
      * fix op test
      
      * fix test_input_spec
      
      * disable cinn in reduce_sum unit test
      
      * add bfloat16 dtype for sum
      
      * add approve rules
      
      * polish code
      
      * add clear jit program function
      
      * convert grad out from tensor to numpy
      
      * remove unnecessary code
      
      * add only_prim flag
      
      * fix flag
      
      * fix op test
      
      * add attr
      
      * fix optest comp inplace error
      
      * fix op test
      
      * fix op test with guard
      
      * add initialization of check_comp flag
      
      * fix comp inplace error in op test
      
      * rename check_comp with check_prim and add bfloat16 dtype convert
      
      * rename comp_op_type to prim_op_type
      
      * rename comp to prim
      
      * remove useless code
      
      * skip ci check for only prim
      
      * add no_grad_vars and grad_outputs in prim test
      
      * fix var_dict
      
      * fix op test for only_prim
      
      * fix dy2static bugs
      
      * polish some code
      
      * temp
      
      * modify op test
      
      * except cinn test
      
      * modify bfp16
      
      * modify pad grad
      
      * add pad_grad dtype
      
      * start cinn part
      
      ---------
      Co-authored-by: NCharles-hit <wanghao107@baidu.com>
      f6dea800
  5. 23 2月, 2023 1 次提交
    • H
      [Tensor Operants & Prim] Tensor arithmetic operants support right scalar type (#50563) · 5f5a2082
      HongyuJia 提交于
      * polish namespace
      
      * change static_tensor_operants
      
      * polish namespace
      
      * support add, subtract, divide
      
      * add unit test
      
      * polish unittest
      
      * fix cmake error
      
      * solve conflicts, merge auto code-gen
      
      * add scalar operator in tensor.h
      
      * tensorbase
      
      * static prim full support more datatype
      
      * fix prim unittest
      
      * polish codes
      
      * fix cmake error
      5f5a2082
  6. 22 2月, 2023 1 次提交
    • J
      【Prim】Add gather vjp (#50305) · 4db8e5c7
      Jiabin Yang 提交于
      * tmp gather vjp
      
      * support gather
      
      * remove useless code
      
      * fix compiling error
      
      * fix ut
      
      * add eager test
      
      * add eager test
      
      * add seed
      
      * fix cpu error
      
      * fix transpose op compat
      
      * remove tensor index case
      
      * fix prim_cinn
      
      * fix ut
      4db8e5c7
  7. 18 2月, 2023 1 次提交
  8. 12 2月, 2023 1 次提交