[mlir][python] fix PyThreadState_GetFrame

[makslevental]

PyThreadState_GetFrame wasn't added until 3.9 (fixes currently failing rocm builder)

[llvmbot]

@llvm/pr-subscribers-mlir

Author: Maksim Levental (makslevental)

Changes

PyThreadState_GetFrame wasn't added until 3.9 (fixes currently failing rocm builder)

---

Full diff: https://github.com/llvm/llvm-project/pull/153325.diff

1 Files Affected:

• (modified) mlir/lib/Bindings/Python/IRCore.cpp (+8)

diff --git a/mlir/lib/Bindings/Python/IRCore.cpp b/mlir/lib/Bindings/Python/IRCore.cpp
index 1aec7ffe8e083..03b04ffbe9749 100644
--- a/mlir/lib/Bindings/Python/IRCore.cpp
+++ b/mlir/lib/Bindings/Python/IRCore.cpp
@@ -2786,6 +2786,14 @@ class PyOpAttributeMap {
   PyOperationRef operation;
 };
 
+// bpo-40429 added PyThreadState_GetFrame() to Python 3.9.0b1
+#if PY_VERSION_HEX < 0x030900B1 && !defined(PYPY_VERSION)
+static inline PyFrameObject *PyThreadState_GetFrame(PyThreadState *tstate) {
+  assert(tstate != _Py_NULL);
+  return _Py_CAST(PyFrameObject *, Py_XNewRef(tstate->frame));
+}
+#endif
+
 MlirLocation tracebackToLocation(MlirContext ctx) {
   size_t framesLimit =
       PyGlobals::get().getTracebackLoc().locTracebackFramesLimit();