1. 05 3月, 2023 1 次提交
  2. 03 3月, 2023 11 次提交
  3. 02 3月, 2023 20 次提交
  4. 01 3月, 2023 8 次提交
    • C
      Integration flash attention (#49869) · 61611786
      Chitsing KUI 提交于
      * flash attn
      
      * seed
      
      * almost
      
      * softmax
      
      * fix workspace
      
      * add unitest; linux only
      
      * fix setup
      
      * fix datatype include
      
      * fix setup typo
      
      * fix def scope
      
      * new error api
      
      * use paddle fork
      
      * fix attr bug; complete ut
      
      * update flash hash
      
      * fix rng reset
      
      * fix offset
      
      * fix comments
      61611786
    • J
      【Prim】Fix sqrt grad (#51045) · 5751b7f4
      Jiabin Yang 提交于
      * fix sqrt grad
      
      * fix sqrt grad
      5751b7f4
    • H
      [Tensor Operants & Prim-Relevant] Tensor supports logical operants (#50983) · 1794927b
      HongyuJia 提交于
      * Add comments for #50886
      
      * [Tensor Operants & Prim-Relevant] Tensor supports logical operants
      
      * add prim dynamic unit test
      
      * add prim static unit test
      1794927b
    • 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
    • Y
      [Zero-Dim] Add Expand/Expand_as/Top_k for XPU to support Zero Dim Input. (#50947) · 226b4a95
      yunyaoXYY 提交于
      * Add unitest from shilong
      
      * Add kernel code from shilong
      
      * fix codestyle
      
      * add broadcast_shape test
      
      * fix unitest
      
      * fix unitests
      
      * fix unitest
      
      * add 0D grad support
      
      * add 0D grad support
      
      * add 0D grad support
      
      * fix 0D tensor
      
      * fix 0D
      
      * fix xpu 0D
      
      * fix expand kernel
      
      * fix xpu expand
      
      * Fix 0D kernel
      
      * fix 0D
      
      * fix 0D
      
      * fix 0D
      
      * fix 0D
      
      * fix XPU top_k
      
      * cancel the modify of xpu
      
      * add XPU 0D tensor
      
      * fix 0D
      226b4a95
    • W
      fix the backward bug of cumsum (#50997) · 934934d8
      wawltor 提交于
      934934d8
    • M
    • TaoTao Li's avatar
      a348a423