We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent c5e972a commit ad61f63Copy full SHA for ad61f63
torchvision/csrc/vision.cpp
@@ -14,18 +14,18 @@
14
#include "nms.h"
15
16
// If we are in a Windows environment, we need to define
17
-// initialization functions for the _custom_ops extension
+// initialization functions for the _C extension
18
#ifdef _WIN32
19
#if PY_MAJOR_VERSION < 3
20
-PyMODINIT_FUNC init_custom_ops(void) {
+PyMODINIT_FUNC init_C(void) {
21
// No need to do anything.
22
- // _custom_ops.py will run on load
+ // extension.py will run on load
23
return NULL;
24
}
25
#else
26
-PyMODINIT_FUNC PyInit__custom_ops(void) {
+PyMODINIT_FUNC PyInit__C(void) {
27
28
29
30
31
#endif
0 commit comments