From ee56cd5c72d7d2d42313ddc9d48ddd9e805552e3 Mon Sep 17 00:00:00 2001 From: kingeasternsun Date: Thu, 11 Apr 2024 03:16:42 +0000 Subject: [PATCH] =?UTF-8?q?=E6=AD=A3=E5=88=99=E6=A8=A1=E5=9D=97=E5=8F=AA?= =?UTF-8?q?=E9=9C=80=E8=A6=81=E5=88=9D=E5=A7=8B=E5=8C=96=E4=B8=80=E6=AC=A1?= =?UTF-8?q?=20regexp.MustCompile(`^Ascend(910|310|310B|310P)-(\d+)$`)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: kingeasternsun --- runtime/main.go | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/runtime/main.go b/runtime/main.go index 8d5fedd..d04d0c9 100644 --- a/runtime/main.go +++ b/runtime/main.go @@ -323,13 +323,14 @@ func parseDevices(visibleDevices string) ([]int, error) { return removeDuplication(devices), nil } +var ascendRex = regexp.MustCompile(`^Ascend(910|310|310B|310P)-(\d+)$`) func parseAscendDevices(visibleDevices string) ([]int, error) { devicesList := strings.Split(visibleDevices, ",") devices := make([]int, 0, len(devicesList)) chipType := "" for _, d := range devicesList { - matchGroups := regexp.MustCompile(`^Ascend(910|310|310B|310P)-(\d+)$`).FindStringSubmatch(strings.TrimSpace(d)) + matchGroups := ascendRex.FindStringSubmatch(strings.TrimSpace(d)) if matchGroups == nil { return nil, fmt.Errorf("invalid device format: %s", d) } -- Gitee