mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2025-10-24 00:53:22 +08:00
[Serving]ppcls preprocessor support gpu (#615)
* serving ppcls support gpu * serving ppcls preprocessor use cpu
This commit is contained in:
@@ -151,8 +151,7 @@ pybind11::capsule FDTensorToDLPack(FDTensor& fd_tensor) {
|
||||
|
||||
dlpack_tensor->dl_tensor.dtype = FDToDlpackType(fd_tensor.dtype);
|
||||
|
||||
// TODO(liqi): FDTensor add device_id
|
||||
dlpack_tensor->dl_tensor.device.device_id = 0;
|
||||
dlpack_tensor->dl_tensor.device.device_id = fd_tensor.device_id;
|
||||
if(fd_tensor.device == Device::GPU) {
|
||||
if (fd_tensor.is_pinned_memory) {
|
||||
dlpack_tensor->dl_tensor.device.device_type = DLDeviceType::kDLCUDAHost;
|
||||
|
Reference in New Issue
Block a user