pythongh-101469: Optimise get_io_state() by using _PyModule_GetState()

erlend-aasland · Jan 31, 2023 · bd4f58a · bd4f58a
1 parent 666c084
commit bd4f58a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/Modules/_io/_iomodule.c b/Modules/_io/_iomodule.c
@@ -11,6 +11,7 @@
 #include "Python.h"
 #include "_iomodule.h"
 #include "pycore_pystate.h"       // _PyInterpreterState_GET()
+#include "pycore_moduleobject.h"  // _PyModule_GetState()
 
 #ifdef HAVE_SYS_TYPES_H
 #include <sys/types.h>
@@ -560,7 +561,7 @@ PyNumber_AsOff_t(PyObject *item, PyObject *err)
 static inline _PyIO_State*
 get_io_state(PyObject *module)
 {
-    void *state = PyModule_GetState(module);
+    void *state = _PyModule_GetState(module);
     assert(state != NULL);
     return (_PyIO_State *)state;
 }