Fix incorrect max boost when enabling threads

This commit is contained in:
loki-47-6F-64 2022-05-12 22:05:12 +02:00
parent 1b94b83dc3
commit a0d5f42baf
2 changed files with 131 additions and 78 deletions

195
main.py
View file

@ -1,16 +1,87 @@
import time import time
#import subprocess
import os import os
VERSION = "0.4.1" VERSION = "0.4.2"
class CPU:
SCALING_FREQUENCIES = [1700000, 2400000, 2800000]
def __init__(self, number):
self.number = number
if(self.status()):
self.max_boost = self._get_max_boost()
else:
self.max_boost = CPU.SCALING_FREQUENCIES[-1]
def enable(self):
# CPU number 0 is special
if(self.number == 0):
return
filepath = cpu_online_path(self.number)
write_to_sys(filepath, 1)
# The user might have changed the maximum cpu clock while the cpu was offline
self._set_max_boost(self.max_boost)
def disable(self):
# CPU number 0 is special
if(self.number == 0):
return
filepath = cpu_online_path(self.number)
write_to_sys(filepath, 0)
def set_max_boost(self, frequency):
self.max_boost = frequency
if(self.status()):
self._set_max_boost(frequency)
def status(self) -> bool:
# cpu number 0 is always online
if(self.number == 0):
return True
filepath = cpu_online_path(self.number)
return read_from_sys(filepath) == "1"
def _read_scaling_governor(self) -> str:
filepath = cpu_governor_scaling_path(self.number)
return read_from_sys(filepath, amount=-1).strip()
def _write_scaling_governor(self, governor: str):
filepath = cpu_governor_scaling_path(self.number)
with open(filepath, mode="w") as f:
f.write(governor)
def _set_max_boost(self, frequency):
if(frequency == CPU.SCALING_FREQUENCIES[-1]):
self._write_scaling_governor("schedutil")
return
if(self._read_scaling_governor() != "userspace"):
self._write_scaling_governor("userspace")
else:
filepath = cpu_freq_scaling_path(self.number)
write_to_sys(filepath, frequency)
def _get_max_boost(self) -> int:
filepath = cpu_freq_scaling_path(self.number)
freq_maybe = read_from_sys(filepath, amount=-1).strip()
if(freq_maybe is None or len(freq_maybe) == 0 or freq_maybe == "<unsupported>"):
return CPU.SCALING_FREQUENCIES[-1]
freq = int(freq_maybe)
return freq
class Plugin: class Plugin:
CPU_COUNT = 8 CPU_COUNT = 8
SCALING_FREQUENCIES = [1700000, 2400000, 2800000]
FAN_SPEEDS = [0, 1000, 2000, 3000, 4000, 5000, 6000] FAN_SPEEDS = [0, 1000, 2000, 3000, 4000, 5000, 6000]
auto_fan = True auto_fan = True
smt = None
async def get_version(self) -> str: async def get_version(self) -> str:
return VERSION return VERSION
@ -18,43 +89,36 @@ class Plugin:
# CPU stuff # CPU stuff
# call from main_view.html with setCPUs(count, smt) # call from main_view.html with setCPUs(count, smt)
async def set_cpus(self, count, smt=True) -> int: async def set_cpus(self, count, smt=True):
cpu_count = len(self.cpus)
self.smt = smt self.smt = smt
# print("Setting CPUs") # print("Setting CPUs")
if smt: if smt:
count = min(int(count), self.CPU_COUNT) count = min(int(count), cpu_count)
# never touch cpu0, since it's special for cpu in self.cpus[: count]:
for cpu in range(1, count): cpu.enable()
enable_cpu(cpu) for cpu in self.cpus[count :: 1]:
for cpu in range(count, self.CPU_COUNT): cpu.disable()
disable_cpu(cpu)
return self.CPU_COUNT
else: else:
count = min(int(count), self.CPU_COUNT / 2) count = min(int(count), cpu_count / 2)
for cpu in range(1, self.CPU_COUNT, 2): # never touch cpu0, since it's special
disable_cpu(cpu) for cpu in self.cpus[1 : cpu_count : 2]:
for cpu in range(2, self.CPU_COUNT, 2): cpu.disable()
if (cpu / 2) + 1 > count: for cpu in self.cpus[2 : cpu_count : 2]:
disable_cpu(cpu) if(cpu.number / 2 + 1 > count):
cpu.disable()
else: else:
enable_cpu(cpu) cpu.enable()
async def get_cpus(self) -> int: async def get_cpus(self) -> int:
online_count = 1 # cpu0 is always online online_count = 0
for cpu in range(1, self.CPU_COUNT): for cpu in self.cpus:
online_count += int(status_cpu(cpu)) if(cpu.status()):
online_count += 1
return online_count return online_count
async def get_smt(self) -> bool: async def get_smt(self) -> bool:
if self.smt is None: return self.smt
for cpu in range(1, self.CPU_COUNT, 2):
if (not status_cpu(cpu)) and status_cpu(cpu+1):
self.smt = False
return False
self.smt = True
return True
else:
return self.smt
async def set_boost(self, enabled: bool) -> bool: async def set_boost(self, enabled: bool) -> bool:
write_to_sys("/sys/devices/system/cpu/cpufreq/boost", int(enabled)) write_to_sys("/sys/devices/system/cpu/cpufreq/boost", int(enabled))
@ -63,30 +127,19 @@ class Plugin:
async def get_boost(self) -> bool: async def get_boost(self) -> bool:
return read_from_sys("/sys/devices/system/cpu/cpufreq/boost") == "1" return read_from_sys("/sys/devices/system/cpu/cpufreq/boost") == "1"
async def set_max_boost(self, index) -> int: async def set_max_boost(self, index):
if index >= len(self.SCALING_FREQUENCIES): if index < 0 or index >= len(CPU.SCALING_FREQUENCIES):
return False return 0
selected_freq = self.SCALING_FREQUENCIES[index]
updated = 0 selected_freq = CPU.SCALING_FREQUENCIES[index]
for cpu in range(0, self.CPU_COUNT):
if cpu == 0 or status_cpu(cpu): for cpu in self.cpus:
if read_scaling_governor_cpu(cpu) != "userspace": cpu.set_max_boost(selected_freq)
write_scaling_governor_cpu(cpu, "userspace")
path = cpu_freq_scaling_path(cpu) return len(self.cpus)
if index == len(self.SCALING_FREQUENCIES) - 1:
write_scaling_governor_cpu(cpu, "schedutil")
else:
write_to_sys(path, selected_freq)
updated += 1
return updated
async def get_max_boost(self) -> int: async def get_max_boost(self) -> int:
path = cpu_freq_scaling_path(0) return CPU.SCALING_FREQUENCIES.index(self.cpus[0].max_boost)
freq_maybe = read_from_sys(path, amount=-1).strip()
if freq_maybe is None or len(freq_maybe) == 0 or freq_maybe == "<unsupported>":
return len(self.SCALING_FREQUENCIES) - 1
freq = int(freq_maybe)
return self.SCALING_FREQUENCIES.index(freq)
# GPU stuff # GPU stuff
@ -150,6 +203,21 @@ class Plugin:
async def _main(self): async def _main(self):
pass pass
# called from main_view::onViewReady
async def on_ready(self):
self.cpus = []
for cpu_number in range(0, Plugin.CPU_COUNT):
self.cpus.append(CPU(cpu_number))
# If any core has two threads, smt is True
self.smt = self.cpus[1].status()
if(not self.smt):
for cpu_number in range(2, len(self.cpus), 2):
if(self.cpus[cpu_number].status()):
self.smt = True
break
# these are stateless (well, the state is not saved internally) functions, so there's no need for these to be called like a class method # these are stateless (well, the state is not saved internally) functions, so there's no need for these to be called like a class method
@ -172,24 +240,3 @@ def write_to_sys(path, value: int):
def read_from_sys(path, amount=1): def read_from_sys(path, amount=1):
with open(path, mode="r") as f: with open(path, mode="r") as f:
return f.read(amount) return f.read(amount)
def enable_cpu(cpu_number: int):
filepath = cpu_online_path(cpu_number)
write_to_sys(filepath, 1)
def disable_cpu(cpu_number: int):
filepath = cpu_online_path(cpu_number)
write_to_sys(filepath, 0)
def status_cpu(cpu_number: int) -> bool:
filepath = cpu_online_path(cpu_number)
return read_from_sys(filepath) == "1"
def read_scaling_governor_cpu(cpu_number: int) -> str:
filepath = cpu_governor_scaling_path(cpu_number)
return read_from_sys(filepath, amount=-1).strip()
def write_scaling_governor_cpu(cpu_number: int, governor: str):
filepath = cpu_governor_scaling_path(cpu_number)
with open(filepath, mode="w") as f:
f.write(governor)

View file

@ -10,6 +10,10 @@
return call_plugin_method("get_version", {}); return call_plugin_method("get_version", {});
} }
function onViewReady() {
return call_plugin_method("on_ready", {});
}
function setCPUs(value, smt) { function setCPUs(value, smt) {
return call_plugin_method("set_cpus", {"count":value, "smt": smt}); return call_plugin_method("set_cpus", {"count":value, "smt": smt});
} }
@ -73,6 +77,8 @@
// other logic // other logic
async function onReady() { async function onReady() {
await onViewReady();
let boostToggle = document.getElementById("boostToggle"); let boostToggle = document.getElementById("boostToggle");
setToggleState(boostToggle, await getCPUBoost()); setToggleState(boostToggle, await getCPUBoost());
setToggleState(document.getElementById("smtToggle"), await getSMT()); setToggleState(document.getElementById("smtToggle"), await getSMT());