Browse Source

fix(mge/functional): fix illegal memory access

GitOrigin-RevId: 8a965ddfbf
master
Megvii Engine Team 2 years ago
parent
commit
f01b1255a5
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      imperative/python/src/tensor_utils.cpp

+ 2
- 2
imperative/python/src/tensor_utils.cpp View File

@@ -427,8 +427,8 @@ py::object _astype_cpp(py::handle tensor, py::handle dtype_hdl) {
"can not convert to numpy.dtype from %s",
dtype_hdl.ptr()->ob_type->tp_name));
}
PyArray_Descr* cur = npy::dtype_mgb2np_descr(_get_dtype(tensor)).get();
if (!dtype_equal(cur, descr)) {
auto&& cur = npy::dtype_mgb2np_descr(_get_dtype(tensor));
if (!dtype_equal(cur.get(), descr)) {
std::shared_ptr<OpDef> op = TypeCvt::make(npy::dtype_np2mgb_descr(descr));
py::object Op = py::cast(op);
PyObject* p[2] = {Op.ptr(), tensor.ptr()};


Loading…
Cancel
Save