- 30 3月, 2021 1 次提交
-
-
由 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>
-
- 24 3月, 2021 1 次提交
-
-
由 Shang Zhizhou 提交于
* fix conflict * fix compile error * cherry-pick #31316 * Refine cudnn softmax (#25757) * refine cudnn softmax * Trt elementwise plugin serialize (#31587) * add serialize unittest * fix element_op trt plugin serialize bug * remove PassVersionChecker.IsCompatible * fix unittest Co-authored-by: NPei Yang <peiyang@baidu.com> Co-authored-by: NGaoWei8 <53294385+GaoWei8@users.noreply.github.com>
-
- 11 3月, 2021 1 次提交
-
-
由 Shang Zhizhou 提交于
-
- 04 1月, 2021 1 次提交
-
-
由 Wilber 提交于
-
- 07 12月, 2020 1 次提交
-
-
由 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
-
- 03 12月, 2020 2 次提交
-
-
由 Shang Zhizhou 提交于
* fix tensorrt output shape error * fix unittest tensorrt_engine_op_test * fix code style for unitest
-
由 Shang Zhizhou 提交于
* cherry-pick a22ea652 * fix analysis_config bug. (#29304) * fix code format Co-authored-by: NWilber <jiweibo@baidu.com>
-
- 01 12月, 2020 2 次提交
-
-
由 yaoxuefeng 提交于
-
由 Wilber 提交于
-
- 25 11月, 2020 1 次提交
-
-
由 Wilber 提交于
-
- 24 11月, 2020 1 次提交
-
-
由 yaoxuefeng 提交于
-
- 20 11月, 2020 1 次提交
-
-
由 Wilber 提交于
* update. * fix trt error in ssd model.
-
- 13 11月, 2020 1 次提交
-
-
由 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
-
- 09 11月, 2020 1 次提交
-
-
由 Wilber 提交于
-
- 05 11月, 2020 1 次提交
-
-
由 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>
-
- 12 10月, 2020 1 次提交
-
-
由 TeslaZhao 提交于
* cherry-pick:Add double grad in Squeeze and Unsqueeze to release/1.8, test=develop Signed-off-by: NTeslaZhao <zhaolisoftware@163.com> * cherry-pick:Add double grad in Squeeze and Unsqueeze to release/1.8, test=develop
-
- 10 10月, 2020 2 次提交
-
-
由 wanghuancoder 提交于
-
由 wangxinxin08 提交于
* add matmul doublegrad op * fix compile errors * modify code according to review * delete float16 * delete GetDimForInput to be consitent with release/1.8
-
- 28 9月, 2020 1 次提交
-
-
由 Jack Zhou 提交于
[Cherry-Pick] Add Log double grad kernel (#27604)
-
- 27 9月, 2020 2 次提交
- 23 9月, 2020 1 次提交
-
-
由 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>
-
- 21 9月, 2020 2 次提交
-
-
由 Pei Yang 提交于
-
由 yaoxuefeng 提交于
-
- 18 9月, 2020 1 次提交
-
-
由 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>
-
- 16 9月, 2020 1 次提交
-
-
由 Pei Yang 提交于
* fix multihead matmul shared params (#27121) * fix multihead matmul shared params
-
- 14 9月, 2020 1 次提交
-
-
由 Jacek Czaja 提交于
test=develop
-
- 11 9月, 2020 1 次提交
-
-
由 lilong12 提交于
* add double grad for expand, test=develop
-
- 10 9月, 2020 1 次提交
-
-
由 Qinghe JING 提交于
* add double grad to reduce sum
-
- 03 9月, 2020 1 次提交
-
-
由 yaoxuefeng 提交于
-
- 02 9月, 2020 2 次提交
-
-
由 Thunderbrook 提交于
cherry-pick fix cvm check test=develop Co-authored-by: N123malin <malin10@baidu.com>
-
由 Thunderbrook 提交于
* fix eigen in push sparse; fix hadoop command test=develop * add log in load_combine_op test=develop
-
- 01 9月, 2020 1 次提交
-
-
由 Pei Yang 提交于
This commit fixs the compiling bug regarding unique_ptr of IOptimizationProfile. IOptimizationProfile has protected dtor and is controlled by TensorRT internally. Application shouldn't delete the pointer of IOptimizationProfile. See TensorRT document: https://docs.nvidia.com/deeplearning/sdk/tensorrt-api/c_api/classnvinfer1_1_1_i_builder.html#a9ac47e100454151d8206ac91d543299a test=develop Co-authored-by: NJeng Bai-Cheng <jeng1220@users.noreply.github.com>
-
- 24 8月, 2020 1 次提交
-
-
由 yaoxuefeng 提交于
* mod cvm test=develop * mod code format test=develop
-
- 19 8月, 2020 1 次提交
-
-
由 Wilber 提交于
-
- 18 8月, 2020 1 次提交
-
-
由 Thunderbrook 提交于
* add mock barrier all (#24786) * add mock barrier all test=develop * fix test=develop * fix test=develop * fix test=develop * fix gloo error test=develop Co-authored-by: Nxujiaqi01 <173596896@qq.com>
-
- 12 8月, 2020 1 次提交
-
-
由 Thunderbrook 提交于
* fix dataset py3 (#25012) * fix dataset py3 error * test=develop * fix logger (#24682) * fix logger of FetchHandler,which may print log twice * test=develop * add timeout and http store in communication (#23436) * add timeout and http store in communication, add revert and confirm in fleet * test=develop * modify datanorm op test=develop (#23030) Co-authored-by: Nxujiaqi01 <173596896@qq.com> Co-authored-by: Nyaoxuefeng <yaoxuefeng@baidu.com>
-
- 11 8月, 2020 1 次提交
-
-
由 Pei Yang 提交于
* add macro check for using TRT api dynamicRangeIsSet() (#25694) * adjust minimum trt version for hard_sigmoid converter to 5130. test=develop (#24746)
-
- 10 8月, 2020 1 次提交
-
-
由 MRXLT 提交于
* fix bug bug fix
-
- 07 8月, 2020 1 次提交
-
-
由 iducn 提交于
Co-authored-by: NTao Luo <luotao02@baidu.com>
-