From 1d4baa428478dcd6b1f3f200357c512ae5d39784 Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Tue, 2 Mar 2021 17:24:42 +0800 Subject: [PATCH] fix(imperative/dtype): fix is_dtype_equal GitOrigin-RevId: b097c9c3ca55806f5149d2ff3f2c444ba4a32849 --- imperative/python/src/numpy_dtypes.cpp | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/imperative/python/src/numpy_dtypes.cpp b/imperative/python/src/numpy_dtypes.cpp index c4d62a83..5543aa4f 100644 --- a/imperative/python/src/numpy_dtypes.cpp +++ b/imperative/python/src/numpy_dtypes.cpp @@ -91,9 +91,8 @@ bool _is_dtype_equal(PyArray_Descr* dt1, PyArray_Descr* dt2) { PyDict_GetItemString(dt1->metadata, "mgb_dtype"), "zero_point"); PyObject* zp2 = PyDict_GetItemString( PyDict_GetItemString(dt2->metadata, "mgb_dtype"), "zero_point"); - if (!zp1 || !zp2) { - throw py::key_error("zero_point"); - } + if (!zp1 && !zp2) return true; + if (!zp1 || !zp2) return false; return PyLong_AsLong(zp1) == PyLong_AsLong(zp2); } if (!q1 && !q2) { -- GitLab