Fix regex pattern for OD_SCLK and OD_RANGE

mengmeet · Feb 19, 2024 · bf97d1e · bf97d1e
1 parent 647cfed
commit bf97d1e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/backend/gpu.py b/backend/gpu.py
@@ -127,7 +127,7 @@ def checkGPUNeedSet(self, freqMin:int, freqMax:int):
                 if os.path.exists(GPUFREQ_PATH):
                     freq_string = open(GPUFREQ_PATH,"r").read()
                     # 使用正则表达式提取频率信息
-                    od_sclk_matches = re.findall(r"OD_SCLK:\s*0:\s*(\d+)Mhz\s*1:\s*(\d+)Mhz", freq_string)
+                    od_sclk_matches = re.findall(r"OD_SCLK:?\s*0:\s*(\d+)Mhz\s*1:\s*(\d+)Mhz", freq_string)
                     if od_sclk_matches:
                         qfreqMin = int(od_sclk_matches[0][0])
                         qfreqMax = int(od_sclk_matches[0][1])
@@ -178,7 +178,7 @@ def get_gpuFreqRange(self):
             open(GPULEVEL_PATH,'w').write("manual")
             freq_string = open(GPUFREQ_PATH,"r").read()
             # 使用正则表达式提取频率信息
-            od_sclk_matches = re.findall(r"OD_RANGE:\s*SCLK:\s*(\d+)Mhz\s*(\d+)Mhz", freq_string)
+            od_sclk_matches = re.findall(r"OD_RANGE:?\s*SCLK:\s*(\d+)Mhz\s*(\d+)Mhz", freq_string)
             logging.debug(f"get_gpuFreqRange {od_sclk_matches[0][0]} {od_sclk_matches[0][1]}")
             if od_sclk_matches:
                 self.gpu_freqRange = [int(od_sclk_matches[0][0]),int(od_sclk_matches[0][1])]