fairseq/clib/libbleu/module.cpp

/**
 * Copyright 2017-present, Facebook, Inc.
 * All rights reserved.
 *
 * This source code is licensed under the license found in the
 * LICENSE file in the root directory of this source tree.
 */

#include <Python.h>

static PyMethodDef method_def[] = {{NULL, NULL, 0, NULL}}; // NOLINT

static struct PyModuleDef module_def = {
    PyModuleDef_HEAD_INIT,
    "libbleu", /* name of module */
    // NOLINTNEXTLINE
    NULL, /* module documentation, may be NULL */
    -1, /* size of per-interpreter state of the module,
           or -1 if the module keeps state in global variables. */
    method_def}; // NOLINT

#if PY_MAJOR_VERSION == 2
PyMODINIT_FUNC init_libbleu()
#else
PyMODINIT_FUNC PyInit_libbleu()
#endif
{
  PyObject* m = PyModule_Create(&module_def);
  if (!m) {
    return NULL;
  }
  return m;
}