psfgh-174: Support free-threading CPython by disabling psutil related features

Author: corona10

doc/changelog.rst

@@ -1,6 +1,14 @@
 Changelog
 =========
 
+Version 2.6.3 (2024-03-05)
+---------------------------
+
+* Support Free-threading CPython (PEP-703) by disabling psutil related features.
+  Patch by Donghee Na.
+* Fix mem_max_rss measurement on macOS.
+  Patch by Mike Droettboom.
+
 Version 2.6.2 (2023-11-02)
 ---------------------------
 

pyperf/_collect_metadata.py

@@ -11,8 +11,12 @@
     resource = None
 
 try:
+    from pyperf._utils import IS_FREE_THREADING
     # Optional dependency
-    import psutil
+    if IS_FREE_THREADING:
+        psutil = None
+    else:
+        import psutil
 except ImportError:
     psutil = None
 

pyperf/_cpu_utils.py

@@ -2,11 +2,14 @@
 import os
 import re
 
-from pyperf._utils import sysfs_path, proc_path, read_first_line
+from pyperf._utils import sysfs_path, proc_path, read_first_line, IS_FREE_THREADING
 
 try:
     # Optional dependency
-    import psutil
+    if IS_FREE_THREADING:
+        psutil = None
+    else:
+        import psutil
 except ImportError:
     psutil = None
 
@@ -152,7 +155,10 @@ def set_cpu_affinity(cpus):
         return True
 
     try:
-        import psutil
+        if IS_FREE_THREADING:
+            return
+        else:
+            import psutil
     except ImportError:
         return
 
@@ -166,7 +172,10 @@ def set_cpu_affinity(cpus):
 
 def set_highest_priority():
     try:
-        import psutil
+        if IS_FREE_THREADING:
+            return
+        else:
+            import psutil
     except ImportError:
         return
 

pyperf/_psutil_memory.py

@@ -1,6 +1,10 @@
 import os
 try:
-    import psutil
+    from pyperf._utils import IS_FREE_THREADING
+    if IS_FREE_THREADING:
+        raise ImportError
+    else:
+        import psutil
 except ImportError:
     raise ImportError('psutil is not installed')
 import threading

pyperf/_utils.py

@@ -3,10 +3,11 @@
 import os
 import statistics
 import sys
+import sysconfig
 from shlex import quote as shell_quote   # noqa
 from shutil import which
 
-
+IS_FREE_THREADING = bool(sysconfig.get_config_var('Py_GIL_DISABLED'))
 MS_WINDOWS = (sys.platform == 'win32')
 MAC_OS = (sys.platform == 'darwin')