1. 09 8月, 2022 1 次提交
    • W
      fix auc instag (#40672) · b042c226
      wangzhen38 提交于
      * fix auc instag
      
      * update ins_tag_w default
      
      * update ins_tag_weight default style
      
      * update ins_tag_weight default style
      
      * optimize
      
      * update by reviews of zwh
      
      * optmize default value
      
      * fix auc_op order
      b042c226
  2. 28 1月, 2022 1 次提交
    • F
      [PSLIB] Add Metrics Module, Support User-defined Add Metric (#38230) · 7460a891
      Fan Zhang 提交于
      * 12.3 first add metrics module
      
      * add Mask/MultiTask
      
      * add WuAUC
      
      * [PSLIB] Update WuAUC Compute
      
      * [PSLIB] Change WuAUC Compute Mehod
      
      * [PSLIB] Clean WuAUC Compute
      
      * [PSLIB] Clean Metric Module Unused Code
      
      * mv metric instance
      
      * [PSLIB] Add Metrics Module, Support User-defined Add Metric (#38789)
      
      * [PSLIB] Add Metrics Module, Support User-defined Add Metric
      
      * [PSLIB] Modify According to CI
      
      * [PSLIB] Modify According to CI
      
      * [PSLIB] Modify According to CI
      
      * [PSLIB] Modify According to CI Coverage
      
      * [PSLIB] Modify According to CI
      
      * [PSLIB] Modify According to CI
      
      * [PSLIB] Modify According to CI
      
      * [PSLIB] Modify According to CI
      
      * [PSLIB] Modify According to CI
      
      * [PSLIB] Modify According to CI Coverage
      
      * [PSLIB] Modify According to CI Coverage
      
      * [PSLIB] Modify According to CI Coverage
      
      * modify role_maker
      
      * update CMakeLists.txt
      7460a891
  3. 17 1月, 2022 1 次提交
  4. 08 11月, 2021 1 次提交
  5. 07 9月, 2021 1 次提交
  6. 17 8月, 2021 1 次提交
  7. 16 7月, 2021 1 次提交
  8. 18 5月, 2021 1 次提交
  9. 09 4月, 2021 1 次提交
  10. 30 3月, 2021 1 次提交
    • S
      Scale 1.8 (#31940) · c36c22fe
      Shang Zhizhou 提交于
      * add n-d input support for trt scale converter (#31316)
      
      * add n-d input support for trt scale converter
      
      * add flatten for ut
      
      * fix dims
      
      * fix batchnorm when inpu dims < 3 (#31933)
      
      * fix batchnorm when inpu dims < 3
      
      * add unittest for batchnorm dims = 2
      
      * fix unittest
      Co-authored-by: NPei Yang <peiyang@baidu.com>
      c36c22fe
  11. 24 3月, 2021 1 次提交
  12. 11 3月, 2021 1 次提交
  13. 04 1月, 2021 1 次提交
  14. 07 12月, 2020 1 次提交
    • S
      cherry-pick PR #27933 (#29377) · 9a6ecb03
      Shang Zhizhou 提交于
      * cherry-pick PR #27933
      
      * fix: cuda version is in varibale CUDA_VERSION in 1.8 cuda.cmake
      
      * close unittest failed temporarily
      
      * cherry-pick PR #27544, fix layer_norm and softmax bug in tensorRT
      9a6ecb03
  15. 03 12月, 2020 2 次提交
  16. 01 12月, 2020 2 次提交
  17. 25 11月, 2020 1 次提交
  18. 24 11月, 2020 1 次提交
  19. 20 11月, 2020 1 次提交
  20. 13 11月, 2020 1 次提交
    • S
      Skip layernorm to 1.8 (#28583) · ec672e88
      Shang Zhizhou 提交于
      * 裁剪transformer模型trt支持;修复tensorRT不支持DeletePass的bug (#28517)
      
      * skip_layernorm_op done
      
      * add unittest
      
      * slice op convertor support trt < 6
      
      * skip_layernorm only work in ernie
      
      * fix unittest
      
      * fix unittest
      ec672e88
  21. 09 11月, 2020 1 次提交
  22. 05 11月, 2020 1 次提交
    • S
      Ernie varlen to 1.8 (#28400) · 78d68d59
      Shang Zhizhou 提交于
      * Fix TRT plugin registry without TRT lib (#25982)
      
      * fix trt plugin registry without trt lib
      
      * support trt4
      
      * refine code style
      
      * pick ea851796 from develop
      
      * cherry-pick develop PR  #26273 && #27796
      
      * fix unittest error
      
      * fix unittest error
      
      * remove const_cast
      Co-authored-by: NPei Yang <peiyang@baidu.com>
      78d68d59
  23. 12 10月, 2020 1 次提交
  24. 10 10月, 2020 2 次提交
  25. 28 9月, 2020 1 次提交
  26. 27 9月, 2020 2 次提交
  27. 23 9月, 2020 1 次提交
    • P
      Optimize slice trt plugin (#26970) (#27456) · 8e1712a7
      Pei Yang 提交于
      * optimize slice TRT plugin
      
      This patch removes unnecessary barrier for data transfer of needed offset,
      so data transfer can be overlap with GPU kernel execution.
      
      This patch also fixes incorrect name of slice plugin. That is, replaces
      "layernorm" with "slice"
      
      test=develop
      
      * add serialize/deserialize to slice plugin
      
      * add static shape slice trt plugin
      
      * fix slice trt op convertor dynamic shape bug
      
      * fix format by clang-format
      
      * fix pylint format error
      
      * fix problems commented by peiyang
      Co-authored-by: NRyan Jeng <rjeng@nvidia.com>
      Co-authored-by: NShang Zhizhou <shangzhizhou@baidu.com>
      Co-authored-by: NRyan Jeng <rjeng@nvidia.com>
      8e1712a7
  28. 22 9月, 2020 2 次提交
    • T
      update python 2.7.15 (#27457) · 2747adff
      tianshuo78520a 提交于
      2747adff
    • G
      [cherry-pick] support python3.8 (#27450) · 5260e473
      guofei 提交于
      * Fix test_gast_with_compatibility.py due to the problem of gast in python3.8 (#27433)
      
      test=develop
      
      * fix dll load bug on windows from python3.8 (#27324)
      
      * Support python3.8 (#26850)
      
      * Support python3.8
      
      test=notest
      
      * Replace the 'spawn' start method with 'fork' start method for multiprocessing, on MacOS with python>=3.8 (#27317)
      
      * Replace the 'spawn' start method with 'fork' start method for multiprocessing, on MacOs when python>=3.8
      
      test=develop
      
      * Correct the error in decorator.py (#27409)
      
      test=develop
      Co-authored-by: NZhou Wei <52485244+zhouwei25@users.noreply.github.com>
      5260e473
  29. 21 9月, 2020 2 次提交
  30. 18 9月, 2020 1 次提交
    • P
      [cherry-pick][Paddle-TRT] Stack op plugin (#25605) (#27365) · 4283be52
      Pei Yang 提交于
      * [Paddle-TRT] Stack op plugin (#25605)
      
      * add stack_op to CMakeLists
      
      * add dim=3 support for scale op
      
      * add trt stack op, test=develop
      
      * remove debug message
      
      * add stack plugin serialize
      
      * remove slice, scale op, will add later
      
      * enhence error message
      
      * revise trt ernie test to conver the stack op CI testi, test=develop
      
      * add stack op serialization
      
      * fix test shape after adding stack op
      
      * remove slice op, will add after implementing serialization
      
      * roll back to min_graph=5 to avoid using slice op
      
      * fix scale op output layer
      
      * implement stack op createPlugin
      
      * use workspace and move the defination to .cu
      
      * move stack plugin creator definition to .cu, test=develop
      
      * sync ut with develop
      Co-authored-by: Nzlsh80826 <zlsh80826@gmail.com>
      4283be52
  31. 16 9月, 2020 1 次提交
  32. 14 9月, 2020 1 次提交
  33. 11 9月, 2020 1 次提交
  34. 10 9月, 2020 1 次提交