提交 e89c6e68 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Dump THEANO_INIT_FUNC

上级 75c785da
......@@ -171,14 +171,14 @@ static struct PyModuleDef moduledef = {{
MyMethods,
}};
""".format(name=self.hash_placeholder), file=stream)
print(("THEANO_INIT_FUNC PyInit_%s(void) {" %
print(("PyMODINIT_FUNC PyInit_%s(void) {" %
self.hash_placeholder), file=stream)
for block in self.init_blocks:
print(' ', block, file=stream)
print(" PyObject *m = PyModule_Create(&moduledef);", file=stream)
print(" return m;", file=stream)
else:
print(("THEANO_INIT_FUNC init%s(void){" %
print(("PyMODINIT_FUNC init%s(void){" %
self.hash_placeholder), file=stream)
for block in self.init_blocks:
print(' ', block, file=stream)
......
......@@ -237,7 +237,7 @@ def compile_cutils():
CutilsExtMethods,
};
THEANO_INIT_FUNC
PyMODINIT_FUNC
PyInit_cutils_ext(void) {
import_array();
return PyModule_Create(&moduledef);
......@@ -246,7 +246,7 @@ def compile_cutils():
"""
else:
code += """
THEANO_INIT_FUNC
PyMODINIT_FUNC
initcutils_ext(void)
{
import_array();
......
......@@ -1038,11 +1038,11 @@ static struct PyModuleDef moduledef = {
#endif
#if defined(NPY_PY3K)
#define RETVAL m
THEANO_INIT_FUNC
PyMODINIT_FUNC
PyInit_lazylinker_ext(void) {
#else
#define RETVAL
THEANO_INIT_FUNC
PyMODINIT_FUNC
initlazylinker_ext(void)
{
#endif
......
......@@ -21,7 +21,8 @@
#define THEANO_RTYPE PyObject *
#endif
/* Can't use PyMODINIT_FUNC since we need to place MOD_PUBLIC in the middle */
#define THEANO_INIT_FUNC THEANO_EXTERN MOD_PUBLIC THEANO_RTYPE
/* We need to redefine PyMODINIT_FUNC to add MOD_PUBLIC in the middle */
#undef PyMODINIT_FUNC
#define PyMODINIT_FUNC THEANO_EXTERN MOD_PUBLIC THEANO_RTYPE
#endif
......@@ -3490,10 +3490,10 @@ static struct PyModuleDef cuda_ndarray_moduledef =
module_methods
};
THEANO_INIT_FUNC
PyMODINIT_FUNC
PyInit_cuda_ndarray(void)
#else
THEANO_INIT_FUNC
PyMODINIT_FUNC
initcuda_ndarray(void)
#endif
{
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论