From 470a7db9341f76e4fcf36c0766ddef3ad8c205c1 Mon Sep 17 00:00:00 2001 From: HexToString <506181616@qq.com> Date: Tue, 1 Jun 2021 14:54:20 +0000 Subject: [PATCH] fix pybind bug --- core/general-client/src/general_model.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/core/general-client/src/general_model.cpp b/core/general-client/src/general_model.cpp index 855e79b4..532b7862 100755 --- a/core/general-client/src/general_model.cpp +++ b/core/general-client/src/general_model.cpp @@ -271,7 +271,7 @@ int PredictorClient::numpy_predict( return -1; } int nbytes = float_feed[vec_idx].nbytes(); - void *rawdata_ptr = reinterpret_cast float_feed[vec_idx].data(0); + void *rawdata_ptr = (void*)(float_feed[vec_idx].data(0)); int total_number = float_feed[vec_idx].size(); Tensor *tensor = tensor_vec[idx]; @@ -413,7 +413,7 @@ int PredictorClient::numpy_predict( } Tensor *tensor = tensor_vec[idx]; int nbytes = int_feed[vec_idx].nbytes(); - void *rawdata_ptr = reinterpret_cast int_feed[vec_idx].data(0); + void *rawdata_ptr = (void *)(int_feed[vec_idx].data(0)); int total_number = int_feed[vec_idx].size(); for (uint32_t j = 0; j < int_shape[vec_idx].size(); ++j) { -- GitLab