Commit Graph

3 Commits

Author SHA1 Message Date
heliqi
4d1f264d01 FDTensor support GPU device (#190)
* fdtensor support GPU

* TRT backend support GPU FDTensor

* FDHostAllocator add FASTDEPLOY_DECL

* fix FDTensor Data

* fix FDTensor dtype

Co-authored-by: Jason <jiangjiajun@baidu.com>
2022-09-08 16:53:08 +08:00
Jason
969531dcc8 Optimize TensorRT backend to support rebuild engine (#189)
* optimize tensorrt usage

* format code

* fix input shape error for onnx model

Co-authored-by: root <root@bjyz-sys-gpu-kongming3.bjyz.baidu.com>
2022-09-06 10:53:05 +08:00
Jason
737b62a2ba Remove tensorrt/common codes (#171) 2022-08-29 19:21:18 +08:00