diff --git a/runtime/main.go b/runtime/main.go
index 8d5fedd2c47fdb28dc85c7fb885bdacf3867c36b..d04d0c9e0641a414c71be75fbed90121a9f6c6e3 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)
 		}