1. 09 3月, 2023 5 次提交
    • 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
    • G
      Fix hybrid parallel training strategy using bf16 (#51103) · 8db15a42
      Ghost Screaming 提交于
      * Fix bug of reduce_sum op. When input.numel() > INT32_MAX, its result
      is wrong.
      
      * Remove climits.
      
      * Fix bug of hybrid parallel strategy with recompute using bf16.
      
      * Fix bug of recompute_hybrid ctx.amp_dtype
      
      * Fix bug of amp_dtype.
      
      * Fix bug of auto_cast.
      8db15a42
    • TaoTao Li's avatar
      Add comm context manager, add phi broadcast op (#51072) · c191b707
      TaoTao Li 提交于
      * * add comm context for device context
      
      * add broadcast phi operator kernel and api
      
      * add broadcast support dtype, update ut
      
      * fix broadcast bfloat16 type
      
      * fix ut
      
      * update test_collective_broadcast_api timeout to 300
      c191b707
    • fix adamax test (#51366) · f759cf0f
      姜永久 提交于
      f759cf0f
    • W
      [prim] add elementwise_pow backward (#51230) · d9de3ef6
      wangzhen38 提交于
      * [cinn] add elementwise_pow backward
      
      * [cinn] update unnitest
      
      * [cinn] update by comments
      
      * [cinn] for ci
      
      * [cinn] for ci
      
      * [cinn] for ci
      
      * [cinn] for ci
      
      * [cinn] for ci
      d9de3ef6
  2. 08 3月, 2023 16 次提交
  3. 07 3月, 2023 8 次提交
  4. 06 3月, 2023 11 次提交