apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-12-24 13:28:13 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

[Pybind] Try to fix xpu with Set External Data support copy (#2022)

Browse Source 
Set External Data support copy
This commit is contained in:
Jason
2023-06-09 15:47:40 +08:00
committed by GitHub
parent 5f9e8b6e08
commit 73f42e36be
1 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
fastdeploy/pybind/fd_tensor.cc
View File

@@ -266,9 +266,13 @@ void BindFDTensor(pybind11::module& m) {
}
Device fd_data_place;
bool copy = false;
if (data_place.find("gpu") != data_place.npos) {
fd_data_place = Device::GPU;
} else {
} else if (data_place.find("cpu") != data_place.npos) {
copy = true;
fd_data_place = Device::CPU;
} else {
FDASSERT(false,
("Device type " + data_place +
" is not support by FDTensor.from_external_data.")
@@ -277,9 +281,7 @@ void BindFDTensor(pybind11::module& m) {
void* data_ptr = nullptr;
data_ptr = reinterpret_cast<void*>(data_addr);
FDTensor fd_tensor(name);
fd_tensor.SetExternalData(shape, fd_data_type,
static_cast<void*>(data_ptr),
fd_data_place, device_id);
fd_tensor.SetData(shape, fd_data_type, static_cast<void*>(data_ptr), copy, fd_data_place, device_id);
return fd_tensor;
})
.def("to_dlpack", &FDTensorToDLPack)