1. 01 3月, 2022 1 次提交
    • Z
      [bf16] add bf16 kernel: layer_norm p_norm reduce_sum (#39843) · ce8ed978
      zhangbo9674 提交于
      * add layer norm
      
      * add p norm
      
      * add reduce sum
      
      * refine layer norm register bf16 for cudnn811
      
      * add bf16 cast for hip
      
      * add unittest
      
      * refine rocm
      
      * refine layer_norm unittest
      
      * refine reduce op
      
      * refine unittest
      
      * enhance atol for reduce unittest
      ce8ed978
  2. 09 2月, 2022 1 次提交
  3. 25 1月, 2022 2 次提交
  4. 17 12月, 2021 1 次提交
  5. 15 6月, 2021 1 次提交
    • J
      Support reduce_sum_op float16 (#32966) · 606939de
      jiangcheng 提交于
      * add reduce_sum_op by add self-kernel
      
      * set all ReduceKernel MPType for accuracy
      
      * add float16 test script which input is integer number
      
      * solve reduce sum float16 check_grad problem
      
      * solve conflict and change test script for CI
      
      * change kernel register for CI
      
      * remove all useless template
      606939de
  6. 28 5月, 2021 1 次提交
  7. 18 5月, 2021 1 次提交
  8. 25 12月, 2020 1 次提交
  9. 05 9月, 2019 1 次提交
  10. 16 11月, 2018 1 次提交
    • W
      Refine operator cmake (#14413) · a2d9b344
      Wu Yi 提交于
      * wip simplify operator framework
      
      * wip
      
      * wip
      
      * done test=develop
      
      * clean test=develop
      
      * fix test=develop
      
      * fix deps test=develop
      
      * fix cpu build test=develop
      
      * fix tensorrt build test=develop
      
      * fix tests test=develop
      
      * fix test=develop
      
      * fix cpu build test=develop
      a2d9b344
  11. 12 11月, 2018 1 次提交
  12. 06 6月, 2018 3 次提交