1. 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
  2. 22 9月, 2020 3 次提交
  3. 21 9月, 2020 2 次提交
  4. 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
  5. 17 9月, 2020 1 次提交
  6. 16 9月, 2020 1 次提交
  7. 15 9月, 2020 1 次提交
  8. 14 9月, 2020 1 次提交
  9. 11 9月, 2020 1 次提交
  10. 10 9月, 2020 1 次提交
  11. 08 9月, 2020 1 次提交
  12. 03 9月, 2020 1 次提交
  13. 02 9月, 2020 2 次提交
  14. 01 9月, 2020 1 次提交
  15. 28 8月, 2020 1 次提交
  16. 27 8月, 2020 1 次提交
  17. 24 8月, 2020 1 次提交
  18. 21 8月, 2020 1 次提交
  19. 20 8月, 2020 1 次提交
  20. 19 8月, 2020 1 次提交
  21. 18 8月, 2020 1 次提交
  22. 17 8月, 2020 5 次提交
  23. 12 8月, 2020 2 次提交
  24. 11 8月, 2020 1 次提交
  25. 10 8月, 2020 1 次提交
  26. 07 8月, 2020 4 次提交
  27. 06 8月, 2020 2 次提交