saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-29 20:14:01 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix dangling pointer to local string variable in onnxruntime_pybind_state.cc.

Browse source
This commit is contained in:
Edward Chen 2020-06-12 00:04:06 +00:00 committed by edgchen1
parent 6b4f652017
commit f74861841e
1 changed files with 4 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
onnxruntime/python/onnxruntime_pybind_state.cc
View file

@ -209,7 +209,7 @@ void GetPyObjFromTensor(const Tensor& rtensor, py::object& obj, const DataTransf
}
}
static std::string GetDeviceName(const OrtDevice& device) {
static const char* GetDeviceName(const OrtDevice& device) {
switch (device.Type()) {
case OrtDevice::CPU:
return CPU;
@ -218,7 +218,7 @@ static std::string GetDeviceName(const OrtDevice& device) {
case OrtDevice::FPGA:
return "FPGA";
default:
throw std::runtime_error("Unknow device type:" + std::to_string(device.Type()));
ORT_THROW("Unknown device type: ", device.Type());
}
}
@ -624,9 +624,7 @@ void addObjectMethods(py::module& m, Environment& env) {
int type_num = dtype->type_num;
Py_DECREF(dtype);
std::string device_name = GetDeviceName(device);
OrtMemoryInfo info(device_name.c_str(), OrtDeviceAllocator, device);
OrtMemoryInfo info(GetDeviceName(device), OrtDeviceAllocator, device);
std::unique_ptr<Tensor> p_tensor = onnxruntime::make_unique<Tensor>(NumpyTypeToOnnxRuntimeType(type_num), shape, (void*)data_ptr, info);
OrtValue mlvalue;
@ -644,9 +642,7 @@ void addObjectMethods(py::module& m, Environment& env) {
int type_num = dtype->type_num;
Py_DECREF(dtype);
std::string device_name = GetDeviceName(device);
OrtMemoryInfo info(device_name.c_str(), OrtDeviceAllocator, device);
OrtMemoryInfo info(GetDeviceName(device), OrtDeviceAllocator, device);
std::unique_ptr<Tensor> p_tensor = onnxruntime::make_unique<Tensor>(NumpyTypeToOnnxRuntimeType(type_num), shape, (void*)data_ptr, info);
OrtValue mlvalue;