- 27 11月, 2020 1 次提交
-
-
由 Shang Zhizhou 提交于
* remove -DSUPPORTS_CUDA_FP16 in cuda.cmake * comile with cuda9 * add some unittest * notest;test=coverage * add unittest for trt plugin swish && split * update ernie unittest * fix some error message * remove repeated judgement of CUDA version in mbEltwiseLayerNormOpConverter * fix comile errror when CUDA_ARCH_NAME < Pascal" * fix comile error * update unittest timeout * compile with cuda9 * update error msg * fix code style * add some comments * add define IF_CUDA_ARCH_SUPPORT_FP16 * rename IF_CUDA_ARCH_SUPPORT_FP16 to CUDA_ARCH_FP16_SUPPORTED
-
- 01 4月, 2020 1 次提交
-
-
由 Zhaolong Xing 提交于
test=develop
-
- 24 7月, 2019 1 次提交
-
-
由 Zhaolong Xing 提交于
* update paddle-trt for: 1. fix bug: when batch > 2, core in split plugin. 2. add leaky_relu trt5.0 support (yolov3 from 65ms to 42ms.) 3. add new attr to dropout. 4. shuffle channel, swish, relu6 support test=develop * 1. fix ci test=develop
-
- 08 3月, 2019 2 次提交
- 27 2月, 2019 1 次提交
-
-
由 nhzlx 提交于
test=develop
-
- 22 2月, 2019 1 次提交
-
-
由 nhzlx 提交于
1). add static trt load model 2). fix bug: when device_id is not 0, the trt will have a bug test=develop
-
- 21 11月, 2018 1 次提交
-
-
由 hjchen2 提交于
-
- 20 11月, 2018 1 次提交
-
-
由 Yiqun Liu 提交于
* Initialize the elementwise plugin. * Implement the basic CUDA kernel of elementwise plugin. test=develop
-
- 16 11月, 2018 1 次提交
-
-
由 nhzlx 提交于
-
- 13 11月, 2018 3 次提交