Use cpython declaration of _PyWeakref_ClearRef (pytorch#136300)

cyyever · pytorchmergebot · commit d2455b99fb4b · 2024-09-20T18:58:58.000Z
To avoid the DLL inconsistency warning by MSVC: ``` torch/csrc/utils/python_compat.h(38): warning C4273: '_PyWeakref_ClearRef': inconsistent dll linkage ``` Pull Request resolved: pytorch#136300 Approved by: https://github.com/Skylion007
diff --git a/torch/csrc/utils/python_compat.h b/torch/csrc/utils/python_compat.h
@@ -35,7 +35,11 @@ PyCode_GetNFreevars(PyCodeObject* code) {
 }
 
 // Provided by CPython but getting the header for them is very hard
+#if IS_PYTHON_3_11_PLUS
+PyAPI_FUNC(void) _PyWeakref_ClearRef(PyWeakReference* self);
+#else
 extern void _PyWeakref_ClearRef(PyWeakReference* self);
+#endif
 
 #ifdef __cplusplus
 }

Original file line number	Diff line number	Diff line change
`@@ -35,7 +35,11 @@ PyCode_GetNFreevars(PyCodeObject* code) {`
`35`	`35`	`}`
`36`	`36`
`37`	`37`	`// Provided by CPython but getting the header for them is very hard`
	`38`	`+#if IS_PYTHON_3_11_PLUS`
	`39`	`+PyAPI_FUNC(void) _PyWeakref_ClearRef(PyWeakReference* self);`
	`40`	`+#else`
`38`	`41`	`extern void _PyWeakref_ClearRef(PyWeakReference* self);`
	`42`	`+#endif`
`39`	`43`
`40`	`44`	`#ifdef __cplusplus`
`41`	`45`	`}`