Function dtype_mgb2np

imperative/python/src/helper.cpp:665–691

}

PyObject* dtype_mgb2np(mgb::DType dtype) {
    PYTHON_GIL;
    // According to
    // https://docs.scipy.org/doc/numpy/reference/c-api.array.html#c.PyArray_TypeObjectFromType
    // the following is equivalent to PyArray_TypeObjectFromType for built-in
    // types.
    if (!dtype.valid()) {
        Py_XINCREF(Py_None);
        return Py_None;
    }
    auto descr = dtype_mgb2np_descr(dtype);
    if (descr == nullptr) {
        Py_XINCREF(Py_None);
        return Py_None;
    }
    static bool use_typeobj_as_dtype = MGB_GETENV("MGE_USE_TYPEOBJ_AS_DTYPE");
    if (use_typeobj_as_dtype) {
        if (dtype.has_param()) {
            return reinterpret_cast<PyObject*>(descr.release());
        }
        PyObject* typeobj = reinterpret_cast<PyObject*>(descr->typeobj);
        Py_XINCREF(typeobj);
        return typeobj;
    } else {
        return reinterpret_cast<PyObject*>(descr.release());
    }
}

mgb::DType dtype_np2mgb(PyObject* obj) {
    mgb_assert(obj && obj != Py_None, "can not convert null PyObject to numpy dtype");