1. 11 4月, 2023 1 次提交
  2. 10 4月, 2023 1 次提交
  3. 07 4月, 2023 1 次提交
  4. 06 4月, 2023 2 次提交
    • J
      support more custom vjp (#52533) · 29c28e2f
      Jiabin Yang 提交于
      29c28e2f
    • K
      feat: add composite rule of roll grad (#52532) · 348a36b5
      Kang Zhao 提交于
      * feat: add relu composite rule
      
      * feat: add relu composite rule, maximum op
      
      * feat: add relu composite rule, maximum op
      
      * feat: add relu composite rule, polish comments
      
      * feat: add relu composite rule, polish comments
      
      * feat: add relu composite rule, add python api of relu
      
      * feat: add relu composite rule, commit hook
      
      * fix: maximum type error & ban cinn test
      
      * fix: maximum input sequence bugs
      
      * resolve conflicts
      
      * fix: code style bugs
      
      * add: relu fp16 test
      
      * feat: add rsqrt composite rule
      
      * feat: add rsqrt composite rule
      
      * resolve conflicts of composite rule
      
      * fix: delete check eager
      
      * feat: add roll grad composite rule
      
      * fix minus shift
      
      * fix test roll op
      348a36b5
  5. 03 4月, 2023 1 次提交
  6. 31 3月, 2023 2 次提交
  7. 30 3月, 2023 3 次提交
  8. 29 3月, 2023 1 次提交
  9. 28 3月, 2023 1 次提交
    • X
      【prim】change layernorm_grad rules (#51879) · 789aac8a
      xiaoguoguo626807 提交于
      * support layer_norm prim and cinn test
      
      * enable cinn test
      
      * fix merge conflict
      
      * polish input for check_output_with_place
      
      * fix merge conflict
      
      * add more test case
      
      * fix merge conflict
      
      * polish test case
      
      * polish op_test
      
      * change ln_g rules
      
      * modify scale is none case
      
      * modify scale is none case
      
      * add public_python_api for check prim
      
      * modify setoutputgrad and fp64bug
      
      * add todo & delete log
      
      * delete Single***varname
      
      * delete get varname
      
      * modify FP64 bug
      
      * delete op test
      
      * recover
      
      * fix conflict
      
      ---------
      Co-authored-by: NWeilong Wu <veyron_wu@163.com>
      789aac8a
  10. 23 3月, 2023 2 次提交
  11. 22 3月, 2023 2 次提交
  12. 21 3月, 2023 4 次提交
  13. 20 3月, 2023 4 次提交
    • X
      【prim】New layer_norm grad (#51750) · 802a81d0
      xiaoguoguo626807 提交于
      * Add flatten composite rule
      
      * get the right xshape and pass func test
      
      * add cinn unit test
      
      * Remove cinn test, wait for it to be added after repair
      
      * add comp test to test_flatten_contiguous_range_op.py
      
      * remove func test on composite_ops
      
      * Add comments to maybe_wrap_dim func
      
      * remove commented code
      
      * fix the problem with 0D tensor case
      
      * add flatten split rule comment
      
      * fix syntax issues
      
      * block flatten on resnet_prim_cinn
      
      * init change
      
      * tmp commit
      
      * add layer_norm InferMeta check
      
      * cast type modify
      
      * [CINN]Enhance CacheKey hash logic by considering input dtypes (#50557)
      
      * [CINN]Enhance CacheKey hash logic by considering input dtypes
      
      * add unittest
      
      * fix typo
      
      * fix typo
      
      * fix map.at
      
      * fix find
      
      * fix test
      
      * fix cinn cache key structure realize
      
      * using ordered map for attributes
      
      * add test by review advice
      
      ---------
      Co-authored-by: Njiangcheng <thisjiang@qq.com>
      
      * [prim] enable dygraph_to_static to support custom_vjp
      
      * Pr 50885 (#7)
      
      * [CINN]Enhance CacheKey hash logic by considering input dtypes (#50557)
      
      * [CINN]Enhance CacheKey hash logic by considering input dtypes
      
      * add unittest
      
      * fix typo
      
      * fix typo
      
      * fix map.at
      
      * fix find
      
      * fix test
      
      * fix cinn cache key structure realize
      
      * using ordered map for attributes
      
      * add test by review advice
      
      ---------
      Co-authored-by: Njiangcheng <thisjiang@qq.com>
      
      * [prim] enable dygraph_to_static to support custom_vjp
      
      * fix code in a dy2static-friendly way.
      
      * [dystatic] add hooker for prim
      
      ---------
      Co-authored-by: NAurelius84 <zhangliujie@baidu.com>
      Co-authored-by: Njiangcheng <thisjiang@qq.com>
      Co-authored-by: Ncxxly <chenxx_id@163.com>
      
      * [prim] enable dygraph_to_static to support custom_vjp
      
      * fix cast prim and vjp dtype mapping error bug
      
      * recover
      
      * big tol
      
      * [CINN]Enhance CacheKey hash logic by considering input dtypes (#50557)
      
      * [CINN]Enhance CacheKey hash logic by considering input dtypes
      
      * add unittest
      
      * fix typo
      
      * fix typo
      
      * fix map.at
      
      * fix find
      
      * fix test
      
      * fix cinn cache key structure realize
      
      * using ordered map for attributes
      
      * add test by review advice
      
      ---------
      Co-authored-by: Njiangcheng <thisjiang@qq.com>
      
      * [prim] enable dygraph_to_static to support custom_vjp
      
      * Pr 50885 (#7)
      
      * [CINN]Enhance CacheKey hash logic by considering input dtypes (#50557)
      
      * [CINN]Enhance CacheKey hash logic by considering input dtypes
      
      * add unittest
      
      * fix typo
      
      * fix typo
      
      * fix map.at
      
      * fix find
      
      * fix test
      
      * fix cinn cache key structure realize
      
      * using ordered map for attributes
      
      * add test by review advice
      
      ---------
      Co-authored-by: Njiangcheng <thisjiang@qq.com>
      
      * [prim] enable dygraph_to_static to support custom_vjp
      
      * fix code in a dy2static-friendly way.
      
      * [dystatic] add hooker for prim
      
      ---------
      Co-authored-by: NAurelius84 <zhangliujie@baidu.com>
      Co-authored-by: Njiangcheng <thisjiang@qq.com>
      Co-authored-by: Ncxxly <chenxx_id@163.com>
      
      * [prim] enable dygraph_to_static to support custom_vjp
      
      * fix cast prim and vjp dtype mapping error bug
      
      * Cxx prim custom vjp (#8)
      
      * [CINN]Enhance CacheKey hash logic by considering input dtypes (#50557)
      
      ---------
      Co-authored-by: Njiangcheng <thisjiang@qq.com>
      
      * [prim] enable dygraph_to_static to support custom_vjp
      
      * Pr 50885 (#7)
      
      * [CINN]Enhance CacheKey hash logic by considering input dtypes (#50557)
      
      * [CINN]Enhance CacheKey hash logic by considering input dtypes
      
      ---------
      Co-authored-by: Njiangcheng <thisjiang@qq.com>
      
      * [prim] enable dygraph_to_static to support custom_vjp
      
      * fix code in a dy2static-friendly way.
      
      * [dystatic] add hooker for prim
      
      ---------
      Co-authored-by: NAurelius84 <zhangliujie@baidu.com>
      Co-authored-by: Njiangcheng <thisjiang@qq.com>
      Co-authored-by: Ncxxly <chenxx_id@163.com>
      
      * [prim] enable dygraph_to_static to support custom_vjp
      
      * fix cast prim and vjp dtype mapping error bug
      
      * [dy2static-ci] fix dy2static ci errors.
      
      ---------
      Co-authored-by: NAurelius84 <zhangliujie@baidu.com>
      Co-authored-by: Njiangcheng <thisjiang@qq.com>
      Co-authored-by: Ncxxly <chenxx_id@163.com>
      
      * [Prim] enable whitelist and blacklist for custom_vjp
      
      * debug log
      
      * clear log
      
      * fix
      
      * nothing
      
      * less memory
      
      * recover utils
      
      * fix
      
      * modify threshold value
      
      * skip layer_norm for test_bert
      
      * back to bert success state
      
      * add epsion
      
      * delete unnecessary compute
      
      * modify amp dtype
      
      * modify * order
      
      * delete sqrt check and fp16
      
      ---------
      Co-authored-by: Nxuyongsheng <xuyongsheng@baidu.com>
      Co-authored-by: Nxysheng-baidu <121540080+xysheng-baidu@users.noreply.github.com>
      Co-authored-by: NAurelius84 <zhangliujie@baidu.com>
      Co-authored-by: Njiangcheng <thisjiang@qq.com>
      Co-authored-by: Ncxxly <chenxx_id@163.com>
      Co-authored-by: Nxiongkun <807377414@qq.com>
      802a81d0
    • W
      add sigmoid custom grad for prim (#51768) · ac47d003
      Weilong Wu 提交于
      ac47d003
    • J
      support relue custom vjp (#51742) · 604b7a53
      Jiabin Yang 提交于
      604b7a53
    • H
      [Tensor Operants & Prim-Relevant] Tensor supports compare operants (#51713) · 6bfb8152
      HongyuJia 提交于
      * [Tensor Operants & Prim-Relevant] Tensor supports compare operants
      
      * fix dependence of test_comp_static
      
      * fix unit test
      6bfb8152
  14. 19 3月, 2023 1 次提交
  15. 17 3月, 2023 1 次提交
  16. 15 3月, 2023 3 次提交
  17. 14 3月, 2023 5 次提交
  18. 13 3月, 2023 2 次提交
    • H
      【prim】Maximum grad (#51006) · 4a484973
      heyanru 提交于
      * refresh
      
      * compat
      
      * register
      
      * testop
      
      * fix
      
      * fix
      
      * fox
      
      * cast
      
      * cast
      
      * fix
      
      * type
      
      * fix
      
      * out
      
      * cast
      
      * fix
      
      * fix
      
      * fix
      
      * broad
      
      * broad
      
      * broad
      
      * fix
      
      * fix
      
      * fix
      
      * fix
      
      * fix
      
      * broad
      
      * broad
      
      * numel
      
      * fix
      
      * fix
      
      * fix
      
      * fix
      
      * cinn
      
      * fix
      
      * fix
      
      * fix
      
      * fix
      4a484973
    • C
      fix static prim api code gen (#51445) · f2fe7c6e
      Charles-hit 提交于
      * fix static prim api code gen
      
      * fix static prim api gen
      f2fe7c6e
  19. 10 3月, 2023 1 次提交
  20. 09 3月, 2023 2 次提交
    • G
      add prim erf grad (#50436) · b7e4d974
      GGBond8488 提交于
      * add prim erf grad
      
      * add yaml config for prim erf grad
      
      * add math.h
      
      * add cmath
      
      * add math  defines
      
      * use define math
      
      * use define math
      
      * define M_2_SQRTPI
      
      * M_2_SQRTPI math
      
      * try math.h
      
      * fix typro
      
      * remove pow in erf grad
      
      * use new optest
      
      * add fp16 fp32 test
      
      * remove fp16 test
      b7e4d974
    • S
      add abs composite backward op (#50963) · d0d739ca
      SylarTiaNII 提交于
      * add abs composite backward op
      
      * add missing changes during merge
      
      * modify according to new rules
      
      * local UT OK
      
      * fix typo
      
      * codestyle
      
      * register composite operator
      
      * add fp16 test for abs
      
      * replace experimenta::tensor
      d0d739ca