Commit e2a8b418 authored by peterjc123's avatar peterjc123 Committed by Francisco Massa
Browse files

Fix Windows build by renaming Python init functions (#1779)

parent 8f4f8d89
...@@ -14,18 +14,18 @@ ...@@ -14,18 +14,18 @@
#include "nms.h" #include "nms.h"
// If we are in a Windows environment, we need to define // If we are in a Windows environment, we need to define
// initialization functions for the _custom_ops extension // initialization functions for the _C extension
#ifdef _WIN32 #ifdef _WIN32
#if PY_MAJOR_VERSION < 3 #if PY_MAJOR_VERSION < 3
PyMODINIT_FUNC init_custom_ops(void) { PyMODINIT_FUNC init_C(void) {
// No need to do anything. // No need to do anything.
// _custom_ops.py will run on load // extension.py will run on load
return NULL; return NULL;
} }
#else #else
PyMODINIT_FUNC PyInit__custom_ops(void) { PyMODINIT_FUNC PyInit__C(void) {
// No need to do anything. // No need to do anything.
// _custom_ops.py will run on load // extension.py will run on load
return NULL; return NULL;
} }
#endif #endif
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment