Skip to content

P
Paddle

PaddlePaddle / Paddle
1 年多 前同步成功

通知 2302
Star 20931
Fork 5422
P
Paddle

refine EngineIOConverter, and use io_convert in test_trt_activation_op !10495

!10495 已合并 由 saxon_zh@saxon_zh 创建

Created by: luotao1

  1. change EngineInputConverter to EngineIOConverter, which has two function: ConvertInput(LoDTensor->ITensor) and ConvertOutput(ITensor -> LoDTensor)
  2. improve the unit-test test_io_converter.cc
  3. use EngineIOConverter in unit-test test_activation_op.cc
  4. remove duplicated cudaMemcpyAsync in SetInputFromCPU