Function dtype_np2mgb

imperative/python/src/helper.cpp:693–708

}

mgb::DType dtype_np2mgb(PyObject* obj) {
    mgb_assert(obj && obj != Py_None, "can not convert null PyObject to numpy dtype");
    // see
    // http://stackoverflow.com/questions/8477122/numpy-c-api-convert-type-object-to-type-number
    PYTHON_GIL;

    PyArray_Descr* dtype;
    if (!PyArray_DescrConverter(obj, &dtype)) {
        throw ConversionError(ssprintf(
                "can not convert to np.dtype from %s", repr_pyobj(obj).c_str()));
    }

    mgb::DType result = dtype_np2mgb_descr(dtype);
    Py_DECREF(dtype);
    return result;
}

PyObject* to_mgb_supported_dtype(PyObject* dtype) {
    PYTHON_GIL;