scripts/RAPL_PKG_Throttle.py

@@ -0,0 +1,53 @@
+#!/usr/bin/python3
+# Copyright (C) 2018 spdfg
+# 
+# This file is part of Elektron.
+# 
+# Elektron is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+# 
+# Elektron is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+# 
+# You should have received a copy of the GNU General Public License
+# along with Elektron.  If not, see <http://www.gnu.org/licenses/>.
+# 
+
+import sys
+import os
+import math
+
+def main():
+   POWER_CAP_DIR = "/sys/class/powercap/"
+
+   if not len(sys.argv) == 2:
+      print(sys.argv[0] + " <throttle percent>")
+      exit(-1)
+
+   if not os.path.exists(POWER_CAP_DIR):
+      print("Powercap framework not installed exist")
+      exit(-1)
+
+   throttle_percent = float(sys.argv[1])/100.0
+
+   if throttle_percent > 1.0 or throttle_percent < 0:
+      print("Percent must be between 0 and 100")
+      exit(-1)
+
+#   print(throttle_percent)
+
+
+
+   for directory in os.listdir(POWER_CAP_DIR):
+      if len(directory.split(':')) == 2:
+         max_watts = open(POWER_CAP_DIR + directory + '/constraint_0_max_power_uw', 'r')
+         rapl_cap_watts = open(POWER_CAP_DIR + directory + '/constraint_0_power_limit_uw', 'w') #0=longer window, 1=shorter window
+
+         rapl_cap_watts.write(str(math.ceil(float(max_watts.read())*throttle_percent)))
+   
+
+main()