Match-id-f711436095c57c791fafc20b43bfe743878d4814

runtime/dcmi/dcmi_api.go

@@ -0,0 +1,87 @@
+// Copyright(C) Huawei Technologies Co.,Ltd. 2020-2021. All rights reserved.
+
+// Package dcmi is used to work with Ascend devices
+package dcmi
+
+import (
+	"fmt"
+	"math"
+	"strconv"
+	"strings"
+
+	"github.com/opencontainers/runtime-spec/specs-go"
+)
+
+// VDeviceInfo vdevice created info
+type VDeviceInfo struct {
+	CardID    int32
+	DeviceID  int32
+	VdeviceID int32
+}
+
+// WorkerInterface worker interface
+type WorkerInterface interface {
+	Initialize() error
+	ShutDown()
+	FindDevice(visibleDevice int32) (int32, int32, error)
+	CreateVDevice(cardID, deviceID int32, coreNum string) (int32, error)
+	DestroyVDevice(cardID, deviceID int32, vDevID int32) error
+}
+
+// CreateVDevice will create virtual device
+func CreateVDevice(w WorkerInterface, spec *specs.Spec) (VDeviceInfo, error) {
+	visibleDevice, splitDevice, err := extractVpuParam(spec)
+	invalidVDevice := VDeviceInfo{CardID: -1, DeviceID: -1, VdeviceID: -1}
+	if err != nil || visibleDevice < 0 {
+		return invalidVDevice, err
+	}
+	if err := w.Initialize(); err != nil {
+		return invalidVDevice, fmt.Errorf("cannot init dcmi : %v", err)
+	}
+	defer w.ShutDown()
+	targetDeviceID, targetCardID, err := w.FindDevice(visibleDevice)
+	if err != nil {
+		return invalidVDevice, err
+	}
+
+	vdeviceID, err := w.CreateVDevice(targetCardID, targetDeviceID, splitDevice)
+	if err != nil || vdeviceID < 0 {
+		return invalidVDevice, fmt.Errorf("cannot create vd or vdevice is wrong: %v %v", vdeviceID, err)
+	}
+	return VDeviceInfo{CardID: targetCardID, DeviceID: targetDeviceID, VdeviceID: int32(vdeviceID)}, nil
+}
+
+func extractVpuParam(spec *specs.Spec) (int32, string, error) {
+	visibleDevice, splitDevice, needSplit, visibleDeviceLine := int32(-1), "", false, ""
+	allowSplit := map[string]string{
+		"vir01": "vir01", "vir02": "vir02", "vir04": "vir04", "vir08": "vir08", "vir16": "vir16",
+	}
+
+	for _, line := range spec.Process.Env {
+		words := strings.Split(line, "=")
+		const LENGTH int = 2
+		if len(words) != LENGTH {
+			continue
+		}
+		if strings.TrimSpace(words[0]) == "ASCEND_VISIBLE_DEVICES" {
+			visibleDeviceLine = words[1]
+		}
+		if strings.TrimSpace(words[0]) == "ASCEND_VNPU_SPECS" {
+			if split, ok := allowSplit[words[1]]; split != "" && ok {
+				splitDevice = split
+				needSplit = true
+			} else {
+				return -1, "", fmt.Errorf("cannot parse param : %v", words[1])
+			}
+		}
+	}
+	if !needSplit {
+		return -1, "", nil
+	}
+	if cardID, err := strconv.Atoi(visibleDeviceLine); err == nil && cardID >= 0 && cardID <= math.MaxInt32 {
+		visibleDevice = int32(cardID)
+	} else {
+		return -1, "", fmt.Errorf("cannot parse param : %v %v", err, visibleDeviceLine)
+	}
+	return visibleDevice, splitDevice, nil
+}