From f01b1255a58a1d6cb1dcd5cc05ba4cfc82d1ab3f Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Mon, 26 Sep 2022 13:50:27 +0800 Subject: [PATCH] fix(mge/functional): fix illegal memory access GitOrigin-RevId: 8a965ddfbf4c51c96f10414e727f1d78b07bf20d --- imperative/python/src/tensor_utils.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/imperative/python/src/tensor_utils.cpp b/imperative/python/src/tensor_utils.cpp index e8f75b34e..906615f5d 100644 --- a/imperative/python/src/tensor_utils.cpp +++ b/imperative/python/src/tensor_utils.cpp @@ -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 op = TypeCvt::make(npy::dtype_np2mgb_descr(descr)); py::object Op = py::cast(op); PyObject* p[2] = {Op.ptr(), tensor.ptr()}; -- GitLab