adjustor 3.2.1__tar.gz → 3.3.0__tar.gz

{adjustor-3.2.1/src/adjustor.egg-info → adjustor-3.3.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: adjustor
-Version: 3.2.1
+Version: 3.3.0
 Summary: Adjustor, a userspace program for managing the TDP of handheld devices.
 Author-email: Kapenekakis Antheas <pypi@antheas.dev>
 Project-URL: Homepage, https://github.com/hhd-dev/adjustor

{adjustor-3.2.1 → adjustor-3.3.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "adjustor"
-version = "3.2.1"
+version = "3.3.0"
 authors = [
   { name="Kapenekakis Antheas", email="pypi@antheas.dev" },
 ]

{adjustor-3.2.1 → adjustor-3.3.0}/src/adjustor/core/const.py

@@ -41,12 +41,24 @@ DEV_PARAMS_7040: dict[str, DeviceParams] = {
     "temp_target": D(60, 70, 85, 90, 100),
 }
 
+DEV_PARAMS_NEO: dict[str, DeviceParams] = {
+    "stapm_limit": D(0, 4, 15, 28, 35),
+    "skin_limit": D(0, 4, 15, 28, 35),
+    "slow_limit": D(0, 4, 20, 32, 37),
+    "fast_limit": D(0, 4, 25, 35, 40),
+    # Times
+    "slow_time": D(5, 5, 10, 10, 10),
+    "stapm_time": D(100, 100, 100, 200, 200),
+    # Temp
+    "temp_target": D(60, 70, 85, 90, 100),
+}
+
 DEV_PARAMS_6040: dict[str, DeviceParams] = DEV_PARAMS_7040
 DEV_PARAMS_8040: dict[str, DeviceParams] = DEV_PARAMS_7040
 DEV_PARAMS_LEGO = DEV_PARAMS_7040
 
 DEV_DATA: dict[str, tuple[dict[str, DeviceParams], dict[str, AlibParams], bool]] = {
-    "NEO-01": (DEV_PARAMS_7040, ALIB_PARAMS_7040, False),
+    "NEO-01": (DEV_PARAMS_NEO, ALIB_PARAMS_7040, False),
     "83E1": (DEV_PARAMS_LEGO, ALIB_PARAMS_7040, False),
 }
 

{adjustor-3.2.1 → adjustor-3.3.0}/src/adjustor/drivers/amd/__init__.py

@@ -2,10 +2,11 @@ import logging
 import os
 import subprocess
 import sys
-from threading import Event as TEvent
 from threading import Thread
 from typing import Literal
+import shutil
 import time
+import signal
 
 from hhd.plugins import Context, HHDPlugin, load_relative_yaml
 from hhd.plugins.conf import Config
@@ -67,14 +68,18 @@ class AmdGPUPlugin(HHDPlugin):
         self.ppd_conflict = False
         self.initialized = False
         self.supports_boost = False
+        self.supports_sched = False
+        self.avail_scheds = {}
 
         self.proc = None
         self.t = None
 
         self.queue = None
+        self.sched_proc = None
         self.old_ppd = False
         self.old_gpu = None
         self.old_freq = None
+        self.old_sched = "disabled"
         self.old_boost = None
         self.old_epp = None
         self.old_target = None
@@ -174,6 +179,31 @@ class AmdGPUPlugin(HHDPlugin):
                 "cpu_pref"
             ]
 
+        self.avail_scheds = {}
+        avail_pretty = {}
+        kernel_supports = os.path.isfile("/sys/kernel/sched_ext/state")
+        if kernel_supports:
+            for sched, pretty in sets["enabled"]["children"]["mode"]["modes"]["manual"][
+                "children"
+            ]["sched"]["options"].items():
+                if sched == "disabled":
+                    avail_pretty[sched] = pretty
+                    continue
+
+                exe = shutil.which(sched)
+                if exe:
+                    self.avail_scheds[sched] = exe
+                    avail_pretty[sched] = pretty
+
+        if self.avail_scheds:
+            sets["enabled"]["children"]["mode"]["modes"]["manual"]["children"]["sched"][
+                "options"
+            ] = avail_pretty
+        else:
+            del sets["enabled"]["children"]["mode"]["modes"]["manual"]["children"][
+                "sched"
+            ]
+
         self.logged_boost = True
         return {
             "tdp": {"amd_energy": sets["enabled"]},
@@ -197,7 +227,7 @@ class AmdGPUPlugin(HHDPlugin):
                         self.proc.stdin.flush()
                 except Exception as e:
                     logger.error(f"Failed to send PPD mode:\n{e}")
-                    self.close()
+                    self.close_ppd()
 
     def update(self, conf: Config):
         self.core_enabled = conf["hhd.settings.tdp_enable"].to(bool)
@@ -224,9 +254,9 @@ class AmdGPUPlugin(HHDPlugin):
                     self.proc, self.t = _open_ppd_server(self.emit)
                 except Exception as e:
                     logger.error(f"Failed to open PPD server:\n{e}")
-                    self.close()
+                    self.close_ppd()
             else:
-                self.close()
+                self.close_ppd()
 
         if conf["tdp.amd_energy.mode.mode"].to(str) == "auto":
             curr = time.perf_counter()
@@ -239,6 +269,8 @@ class AmdGPUPlugin(HHDPlugin):
                 logger.info(
                     f"Handling energy settings for power profile '{self.target}'."
                 )
+                # Unless it is set manually, use the default scheduler.
+                self.close_sched()
                 try:
                     match self.target:
                         case "balanced":
@@ -329,10 +361,43 @@ class AmdGPUPlugin(HHDPlugin):
                     except Exception as e:
                         logger.error(f"Failed to set minimum CPU frequency:\n{e}")
 
-    def close(self):
+            if self.avail_scheds:
+                # Check health and print error
+                if self.sched_proc and self.sched_proc.poll():
+                    err = self.sched_proc.poll()
+                    self.sched_proc = None
+                    logger.error(
+                        f"Scheduler from sched_ext '{self.old_sched}' closed with error code: {err}"
+                    )
+
+                new_sched = conf.get("tdp.amd_energy.mode.manual.sched", "disabled")
+                if new_sched != self.old_sched:
+                    self.close_sched()
+                    self.old_sched = new_sched
+                    if new_sched and new_sched != "disabled":
+                        logger.info(f"Starting sched_ext scheduler '{new_sched}'")
+                        self.sched_proc = subprocess.Popen(
+                            self.avail_scheds[new_sched],
+                            stderr=subprocess.DEVNULL,
+                            stdout=subprocess.DEVNULL,
+                        )
+
+    def close_ppd(self):
         if self.proc is not None:
-            self.proc.terminate()
+            self.proc.send_signal(signal.SIGINT)
             self.proc.wait()
+            self.proc = None
         if self.t is not None:
             self.t.join()
             self.t = None
+
+    def close_sched(self):
+        if self.sched_proc is not None:
+            logger.info(f"Closing sched_ext scheduler '{self.old_sched}'.")
+            self.sched_proc.send_signal(signal.SIGINT)
+            self.sched_proc.wait()
+            self.sched_proc = None
+
+    def close(self):
+        self.close_ppd()
+        self.close_sched()

{adjustor-3.2.1 → adjustor-3.3.0}/src/adjustor/drivers/amd/settings.yml

@@ -87,6 +87,19 @@ enabled:
                       max: 2000
                       step: 100
                       default: 1000
+            sched:
+              type: multiple
+              title: Custom Scheduler
+              hint: >-
+                Allows attaching a scheduler to the kernel sched_ext.
+                Schedulers need to be installed.
+                "Kernel" .
+              options:
+                disabled: Disabled
+                scx_lavd: LAVD
+                scx_bpfland: bpfland 
+                scx_rusty: rusty
+              default: disabled
 
 conflict:
   title: Energy Management

{adjustor-3.2.1 → adjustor-3.3.0}/src/adjustor/hhd.py

@@ -274,6 +274,18 @@ def autodetect(existing: Sequence[HHDPlugin]) -> Sequence[HHDPlugin]:
 
     if not drivers_matched and prod in DEV_DATA:
         dev, cpu, pp_enable = DEV_DATA[prod]
+
+        try:
+            # Set values for the steam slider
+            if dev["skin_limit"].smin:
+                min_tdp = dev["skin_limit"].smin
+            if dev["skin_limit"].default:
+                default_tdp = dev["skin_limit"].default
+            if dev["skin_limit"].smax:
+                max_tdp = dev["skin_limit"].smax
+        except Exception as e:
+            logger.error(f"Failed to get TDP limits for {prod}:\n{e}")
+
         pp_enable |= bool(os.environ.get("HHD_ADJ_DEBUG"))
         drivers.append(
             SmuDriverPlugin(

{adjustor-3.2.1 → adjustor-3.3.0/src/adjustor.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: adjustor
-Version: 3.2.1
+Version: 3.3.0
 Summary: Adjustor, a userspace program for managing the TDP of handheld devices.
 Author-email: Kapenekakis Antheas <pypi@antheas.dev>
 Project-URL: Homepage, https://github.com/hhd-dev/adjustor