remove reg match

zhewang1-intc · Jul 17, 2024 · e9034b8 · e9034b8
1 parent 0382709
commit e9034b8
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/gpu/gpu.go b/gpu/gpu.go
@@ -16,7 +16,6 @@ import (
 	"os"
 	"os/exec"
 	"path/filepath"
-	"regexp"
 	"runtime"
 	"strconv"
 	"strings"
@@ -193,10 +192,8 @@ func DetectInteliGpuMemStatus(gpuInfo *OneapiGPUInfo) {
 		return
 	}
 	result := strings.TrimSpace(string(output))
-	re := regexp.MustCompile("[0-9]+")
-	match := re.FindString(result)
-	if match != "" {
-		usedVRAM, _ := strconv.ParseUint(match, 10, 64)
+	if result != "" {
+		usedVRAM, _ := strconv.ParseUint(result, 10, 64)
 		gpuInfo.FreeMemory = gpuInfo.TotalMemory - usedVRAM
 	} else {
 		gpuInfo.GpuInfo.UnreliableFreeMemory = true