mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
7ddc6f84c4
Summary: How did we get so many uses of `NULL` again? ezyang Pull Request resolved: https://github.com/pytorch/pytorch/pull/11047 Differential Revision: D9566799 Pulled By: goldsborough fbshipit-source-id: 83469f352ac69aa65bdaf1a1a21f922d892e0db3
45 lines
891 B
C++
45 lines
891 B
C++
#include "torch/csrc/python_headers.h"
|
|
|
|
static PyObject* module;
|
|
|
|
static PyMethodDef TorchNvrtcMethods[] = {
|
|
{nullptr, nullptr, 0, nullptr}
|
|
};
|
|
|
|
#if PY_MAJOR_VERSION != 2
|
|
static struct PyModuleDef torchnvrtcmodule = {
|
|
PyModuleDef_HEAD_INIT,
|
|
"torch._nvrtc",
|
|
nullptr,
|
|
-1,
|
|
TorchNvrtcMethods
|
|
};
|
|
#endif
|
|
|
|
#if PY_MAJOR_VERSION == 2
|
|
PyMODINIT_FUNC init_nvrtc(void)
|
|
#else
|
|
PyMODINIT_FUNC PyInit__nvrtc(void)
|
|
#endif
|
|
{
|
|
|
|
#if PY_MAJOR_VERSION == 2
|
|
#define ASSERT_TRUE(cmd) if (!(cmd)) {PyErr_SetString(PyExc_ImportError, "initialization error in torch._nvrtc"); return;}
|
|
#else
|
|
#define ASSERT_TRUE(cmd) if (!(cmd)) return nullptr
|
|
#endif
|
|
|
|
#if PY_MAJOR_VERSION == 2
|
|
ASSERT_TRUE(module = Py_InitModule("torch._nvrtc", TorchNvrtcMethods));
|
|
#else
|
|
ASSERT_TRUE(module = PyModule_Create(&torchnvrtcmodule));
|
|
#endif
|
|
|
|
#if PY_MAJOR_VERSION == 2
|
|
#else
|
|
return module;
|
|
#endif
|
|
|
|
#undef ASSERT_TRUE
|
|
}
|