Function ThrowExceptionToPython

paddle/fluid/pybind/exception.cc:91–140

}

void ThrowExceptionToPython(std::exception_ptr p) {
  static PyObject* EOFExceptionException =
      PyErr_NewException("paddle.EOFException", PyExc_Exception, nullptr);
  static PyObject* EnforceNotMetException =
      PyErr_NewException("paddle.EnforceNotMet", PyExc_Exception, nullptr);
  try {
    if (p) std::rethrow_exception(p);
  } catch (const platform::EOFException& e) {
    PyErr_SetString(EOFExceptionException, e.what());
  } catch (const memory::allocation::BadAlloc& e) {
    PyErr_SetString(PyExc_MemoryError, e.what());
  } catch (const platform::EnforceNotMet& e) {
    switch (e.code()) {
      case common::ErrorCode::INVALID_ARGUMENT:
        PyErr_SetString(PyExc_ValueError, e.what());
        break;
      case common::ErrorCode::NOT_FOUND:
      case common::ErrorCode::ALREADY_EXISTS:
      case common::ErrorCode::PRECONDITION_NOT_MET:
      case common::ErrorCode::PERMISSION_DENIED:
      case common::ErrorCode::EXECUTION_TIMEOUT:
      case common::ErrorCode::UNAVAILABLE:
        PyErr_SetString(PyExc_RuntimeError, e.what());
        break;
      case common::ErrorCode::OUT_OF_RANGE:
        PyErr_SetString(PyExc_IndexError, e.what());
        break;
      case common::ErrorCode::RESOURCE_EXHAUSTED:
        PyErr_SetString(PyExc_MemoryError, e.what());
        break;
      case common::ErrorCode::UNIMPLEMENTED:
        PyErr_SetString(PyExc_NotImplementedError, e.what());
        break;
      case common::ErrorCode::FATAL:
        PyErr_SetString(PyExc_SystemError, e.what());
        break;
      case common::ErrorCode::EXTERNAL:
        PyErr_SetString(PyExc_OSError, e.what());
        break;
      case common::ErrorCode::INVALID_TYPE:
        PyErr_SetString(PyExc_TypeError, e.what());
        break;
      default:
        PyErr_SetString(EnforceNotMetException, e.what());
        break;
    }
  } catch (const common::PD_Exception& e) {
    PyErr_SetString(PyExc_OSError, e.what());
  }
}
}  // namespace paddle::pybind