diff --git a/runtime/dcmi/dcmi.go b/runtime/dcmi/dcmi.go
index dcf9b49..0a5fbe7 100644
--- a/runtime/dcmi/dcmi.go
+++ b/runtime/dcmi/dcmi.go
@@ -5,118 +5,12 @@
 package dcmi
 
 // #cgo LDFLAGS: -ldl
-/* #include <stddef.h>
-  #include <dlfcn.h>
-  #include <stdlib.h>
-  #include <stdio.h>
-
-  #include "dcmi_interface_api.h"
-
-  void *dcmiHandle;
-  #define SO_NOT_FOUND  -99999
-  #define FUNCTION_NOT_FOUND  -99998
-  #define SUCCESS  0
-  #define ERROR_UNKNOWN  -99997
-  #define CALL_FUNC(name,...) if(name##_func==NULL){return FUNCTION_NOT_FOUND;}return name##_func(__VA_ARGS__);
-
-  // dcmi
-  int (*dcmi_init_func)();
-  int dcmi_init(){
-	  CALL_FUNC(dcmi_init)
-  }
-
-  int (*dcmi_get_card_num_list_func)(int *card_num, int *card_list, int list_length);
-  int dcmi_get_card_num_list(int *card_num, int *card_list, int list_length){
-	  CALL_FUNC(dcmi_get_card_num_list,card_num,card_list,list_length)
-  }
-
-  int (*dcmi_get_device_num_in_card_func)(int card_id, int *device_num);
-  int dcmi_get_device_num_in_card(int card_id, int *device_num){
-	  CALL_FUNC(dcmi_get_device_num_in_card,card_id,device_num)
-  }
-
-  int (*dcmi_get_device_logic_id_func)(int *device_logic_id, int card_id, int device_id);
-  int dcmi_get_device_logic_id(int *device_logic_id, int card_id, int device_id){
-	  CALL_FUNC(dcmi_get_device_logic_id,device_logic_id,card_id,device_id)
-  }
-
-  int (*dcmi_create_vdevice_func)(int card_id, int device_id, int vdev_id, const char *template_name,
-    struct dcmi_create_vdev_out *out);
-  int dcmi_create_vdevice(int card_id, int device_id, int vdev_id, const char *template_name,
-    struct dcmi_create_vdev_out *out){
-	  CALL_FUNC(dcmi_create_vdevice,card_id,device_id,vdev_id,template_name,out)
-  }
-
-  int (*dcmi_set_destroy_vdevice_func)(int card_id, int device_id, unsigned int VDevid);
-  int dcmi_set_destroy_vdevice(int card_id, int device_id, unsigned int VDevid){
-	  CALL_FUNC(dcmi_set_destroy_vdevice,card_id,device_id,VDevid)
-  }
-
-  // load .so files and functions
-  int dcmiInit_dl(void){
-	  dcmiHandle = dlopen("libdcmi.so",RTLD_LAZY | RTLD_GLOBAL);
-	  if (dcmiHandle == NULL){
-		  fprintf (stderr,"%s\n",dlerror());
-		  return SO_NOT_FOUND;
-	  }
-
-	  dcmi_init_func = dlsym(dcmiHandle,"dcmi_init");
-
-	  dcmi_get_card_num_list_func = dlsym(dcmiHandle,"dcmi_get_card_num_list");
-
-	  dcmi_get_device_num_in_card_func = dlsym(dcmiHandle,"dcmi_get_device_num_in_card");
-
-	  dcmi_get_device_logic_id_func = dlsym(dcmiHandle,"dcmi_get_device_logic_id");
-
-      dcmi_create_vdevice_func = dlsym(dcmiHandle,"dcmi_create_vdevice");
-
-	  dcmi_set_destroy_vdevice_func = dlsym(dcmiHandle,"dcmi_set_destroy_vdevice");
-
-	  return SUCCESS;
-  }
-
-  int dcmiShutDown(void){
-	  if (dcmiHandle == NULL) {
-		  return SUCCESS;
-	  }
-	  return (dlclose(dcmiHandle) ? ERROR_UNKNOWN : SUCCESS);
-  }
-
-  int (*dsmi_get_logicid_from_phyid_func)(unsigned int phyid, unsigned int *logicid);
-  int dsmi_get_logicid_from_phyid(unsigned int phyid, unsigned int *logicid){
-    CALL_FUNC(dsmi_get_logicid_from_phyid,phyid,logicid)
-  }
-  void *dsmiHandle;
-  int dsmiInit_dl(void){
-	dsmiHandle = dlopen("libdrvdsmi_host.so",RTLD_LAZY);
-	if (dsmiHandle == NULL) {
-		dsmiHandle = dlopen("libdrvdsmi.so",RTLD_LAZY);
-	}
-	if (dsmiHandle == NULL){
-		return SO_NOT_FOUND;
-	}
-
-	dsmi_get_logicid_from_phyid_func = dlsym(dsmiHandle,"dsmi_get_logicid_from_phyid");
-
-	return SUCCESS;
-}
-
-int dsmiShutDown(void){
-	if (dsmiHandle == NULL) {
-   	 	return SUCCESS;
-  	}
-	return (dlclose(dsmiHandle) ? ERROR_UNKNOWN : SUCCESS);
-}
-*/
+// #include "dcmi_interface_api.h"
 import "C"
 import (
 	"fmt"
 	"math"
-	"strconv"
-	"strings"
 	"unsafe"
-
-	"github.com/opencontainers/runtime-spec/specs-go"
 )
 
 const (
@@ -130,15 +24,12 @@ const (
 	hiAIMaxCardNum = 16
 )
 
-// VDeviceInfo vdevice created info
-type VDeviceInfo struct {
-	CardID    int32
-	DeviceID  int32
-	VdeviceID int32
+// NpuWorker Dcmi worker
+type NpuWorker struct {
 }
 
-// InitDcmi dcmi/dsmi lib
-func InitDcmi() error {
+// Initialize dcmi lib init
+func (w *NpuWorker) Initialize() error {
 	if err := C.dcmiInit_dl(); err != C.SUCCESS {
 		errInfo := fmt.Errorf("dcmi lib load failed, , error code: %d", int32(err))
 		return errInfo
@@ -147,24 +38,18 @@ func InitDcmi() error {
 		errInfo := fmt.Errorf("dcmi init failed, , error code: %d", int32(err))
 		return errInfo
 	}
-	if err := C.dsmiInit_dl(); err != C.SUCCESS {
-		errInfo := fmt.Errorf("dsmi lib load failed, , error code: %d", int32(err))
-		return errInfo
-	}
 	return nil
 }
 
-// ShutDownDcmi shutdown dcmi/dsmi lib
-func ShutDownDcmi() {
+// ShutDown shutdown dcmi lib
+func (w *NpuWorker) ShutDown() {
 	if err := C.dcmiShutDown(); err != C.SUCCESS {
 		println(fmt.Errorf("dcmi shut down failed, error code: %d", int32(err)))
 	}
-	if err := C.dsmiShutDown(); err != C.SUCCESS {
-		println(fmt.Errorf("dsmi shut down failed, error code: %d", int32(err)))
-	}
 }
 
-func getCardList() (int32, []int32, error) {
+// GetCardList  list all cards on system
+func GetCardList() (int32, []int32, error) {
 	var ids [hiAIMaxCardNum]C.int
 	var cNum C.int
 	if err := C.dcmi_get_card_num_list(&cNum, &ids[0], hiAIMaxCardNum); err != 0 {
@@ -172,7 +57,7 @@ func getCardList() (int32, []int32, error) {
 		return retError, nil, errInfo
 	}
 	// checking card's quantity
-	if cNum <= 0 {
+	if cNum <= 0 || cNum > hiAIMaxCardNum {
 		errInfo := fmt.Errorf("get error card quantity: %d", int32(cNum))
 		return retError, nil, errInfo
 	}
@@ -218,8 +103,8 @@ func GetDeviceLogicID(cardID, deviceID int32) (int32, error) {
 	return int32(logicID), nil
 }
 
-// SetCreateVDevice create virtual device
-func SetCreateVDevice(cardID, deviceID int32, coreNum string) (uint32, error) {
+// CreateVDevice create virtual device
+func (w *NpuWorker) CreateVDevice(cardID, deviceID int32, coreNum string) (int32, error) {
 	var createInfo C.struct_dcmi_create_vdev_out
 	createInfo.vdev_id = C.uint(math.MaxUint32)
 	coreTemplate := C.CString(coreNum)
@@ -227,13 +112,19 @@ func SetCreateVDevice(cardID, deviceID int32, coreNum string) (uint32, error) {
 	err := C.dcmi_create_vdevice(C.int(cardID), C.int(deviceID), C.int(0), coreTemplate, &createInfo)
 	if err != 0 {
 		errInfo := fmt.Errorf("create virtual device failed, error code: %d", int32(err))
-		return uint32(math.MaxUint32), errInfo
+		return math.MaxInt32, errInfo
 	}
-	return uint32(createInfo.vdev_id), nil
+	if createInfo.vdev_id > math.MaxInt32 {
+		return math.MaxInt32, fmt.Errorf("create virtual device failed, vdeviceId too large")
+	}
+	return int32(createInfo.vdev_id), nil
 }
 
-// SetDestroyVDevice destroy virtual device
-func SetDestroyVDevice(cardID, deviceID int32, vDevID uint32) error {
+// DestroyVDevice destroy virtual device
+func (w *NpuWorker) DestroyVDevice(cardID, deviceID int32, vDevID int32) error {
+	if vDevID < 0 {
+		return fmt.Errorf("param error on vDevID")
+	}
 	if err := C.dcmi_set_destroy_vdevice(C.int(cardID), C.int(deviceID), C.uint(vDevID)); err != 0 {
 		errInfo := fmt.Errorf("destroy virtual device failed, error code: %d", int32(err))
 		return errInfo
@@ -241,78 +132,31 @@ func SetDestroyVDevice(cardID, deviceID int32, vDevID uint32) error {
 	return nil
 }
 
-// CreateVDevice will create virtual device
-func CreateVDevice(spec *specs.Spec) (VDeviceInfo, error) {
-	visibleDevice, splitDevice, err := extractVpuParam(spec)
-	invalidVDevice := VDeviceInfo{CardID: -1, DeviceID: -1, VdeviceID: -1}
-	if err != nil || visibleDevice == -1 {
-		return invalidVDevice, err
+// FindDevice find device by phyical id
+func (w *NpuWorker) FindDevice(visibleDevice int32) (int32, int32, error) {
+	var logicID C.uint
+	if err := C.dcmi_get_device_logicid_from_phyid(C.uint(visibleDevice), &logicID); err != 0 {
+		return 0, 0, fmt.Errorf("phy id can not be converted to logic id : %v", err)
 	}
-	if err := InitDcmi(); err != nil {
-		return invalidVDevice, fmt.Errorf("cannot init dcmi : %v", err)
+	_, cardList, err := GetCardList()
+	if err != nil {
+		return 0, 0, fmt.Errorf("get card list err : %v", err)
 	}
-	defer ShutDownDcmi()
-	var dsmiLogicID C.uint
-	if err := C.dsmi_get_logicid_from_phyid(C.uint(visibleDevice), &dsmiLogicID); err != 0 {
-		return invalidVDevice, fmt.Errorf("phy id can not be converted to logic id : %v", err)
-	}
-	_, cardList, err := getCardList()
 	targetDeviceID, targetCardID := int32(math.MaxInt32), int32(math.MaxInt32)
 	for _, cardID := range cardList {
 		deviceCount, err := GetDeviceNumInCard(cardID)
 		if err != nil {
-			return invalidVDevice, fmt.Errorf("cannot get device num in card : %v", err)
+			return 0, 0, fmt.Errorf("cannot get device num in card : %v", err)
 		}
 		for deviceID := int32(0); deviceID < deviceCount; deviceID++ {
 			logicID, err := GetDeviceLogicID(cardID, deviceID)
 			if err != nil {
-				return invalidVDevice, fmt.Errorf("cannot get logic id : %v", err)
+				return 0, 0, fmt.Errorf("cannot get logic id : %v", err)
 			}
-			if logicID == int32(dsmiLogicID) {
+			if logicID == int32(logicID) {
 				targetCardID, targetDeviceID = cardID, deviceID
 			}
 		}
 	}
-
-	vdeviceID, err := SetCreateVDevice(targetCardID, targetDeviceID, splitDevice)
-	if err != nil || int(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 := allowSplit[words[1]]; split != "" {
-				splitDevice = split
-				needSplit = true
-			} else {
-				return -1, "", fmt.Errorf("cannot parse param : %v", words[1])
-			}
-		}
-	}
-	if needSplit {
-		if cardID, err := strconv.Atoi(visibleDeviceLine); err == nil {
-			visibleDevice = int32(cardID)
-		} else {
-			return -1, "", fmt.Errorf("cannot parse param : %v %v", err, visibleDeviceLine)
-		}
-	} else {
-		return -1, "", nil
-	}
-	return visibleDevice, splitDevice, nil
+	return targetDeviceID, targetCardID, nil
 }
diff --git a/runtime/dcmi/dcmi_api.go b/runtime/dcmi/dcmi_api.go
new file mode 100644
index 0000000..10e6cf3
--- /dev/null
+++ b/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
+}
diff --git a/runtime/dcmi/dcmi_api_test.go b/runtime/dcmi/dcmi_api_test.go
new file mode 100644
index 0000000..b7e0cb3
--- /dev/null
+++ b/runtime/dcmi/dcmi_api_test.go
@@ -0,0 +1,70 @@
+// Copyright (c) Huawei Technologies Co., Ltd. 2020-2022. All rights reserved.
+
+// Description: dcmi DT测试
+package dcmi
+
+import (
+	"testing"
+
+	"github.com/opencontainers/runtime-spec/specs-go"
+)
+
+const mockDeviceId = 100
+
+type mockWorker struct{}
+
+func (w *mockWorker) Initialize() error {
+	return nil
+}
+
+// ShutDown shutdown mock lib
+func (w *mockWorker) ShutDown() {
+	return
+}
+
+// CreateVDevice create v device
+func (w *mockWorker) CreateVDevice(_, _ int32, _ string) (int32, error) {
+
+	return int32(mockDeviceId), nil
+}
+
+// DestroyVDevice destroy virtual device
+func (w *mockWorker) DestroyVDevice(_, _, _ int32) error {
+	return nil
+}
+
+// FindDevice find device by phyical id
+func (w *mockWorker) FindDevice(_ int32) (int32, int32, error) {
+	return 0, 0, nil
+}
+
+func TestCreateVDevice(t *testing.T) {
+	t.Log("TestCreateVDevice start")
+	process := specs.Process{}
+	spec := specs.Spec{Process: &process}
+	spec.Process.Env = []string{}
+
+	// no split, all ok
+	vdevice, err := CreateVDevice(&mockWorker{}, &spec)
+	if err != nil {
+		t.Fatalf("%v %v", vdevice, err)
+	}
+
+	// no npu assigin for split
+	spec.Process.Env = []string{"ASCEND_VNPU_SPECS=vir04"}
+	vdevice, err = CreateVDevice(&mockWorker{}, &spec)
+	if err == nil {
+		t.Fatalf("%v %v", vdevice, err)
+	}
+
+	// split ok
+	spec.Process.Env = []string{"ASCEND_VNPU_SPECS=vir04", "ASCEND_VISIBLE_DEVICES=0"}
+	vdevice, err = CreateVDevice(&mockWorker{}, &spec)
+	if err != nil {
+		t.Fatalf("%v %v", vdevice, err)
+	}
+	if vdevice.VdeviceID != mockDeviceId {
+		t.Fatalf("%v %v", vdevice, err)
+	}
+
+}
diff --git a/runtime/dcmi/dcmi_interface_api.h b/runtime/dcmi/dcmi_interface_api.h
index e5b47a8..e7e3f7a 100644
--- a/runtime/dcmi/dcmi_interface_api.h
+++ b/runtime/dcmi/dcmi_interface_api.h
@@ -5,22 +5,23 @@
 
 #ifndef __DCMI_INTERFACE_API_H__
 #define __DCMI_INTERFACE_API_H__
-
+#include <stddef.h>
+#include <dlfcn.h>
+#include <stdlib.h>
+#include <stdio.h>
 #ifdef __cplusplus
 #if __cplusplus
 extern "C" {
 #endif
 #endif /* __cplusplus */
 
-#ifdef __linux
-#define DCMIDLLEXPORT
-#else
-#define DCMIDLLEXPORT _declspec(dllexport)
-#endif
-
-#define DCMI_VDEV_RES_NAME_LEN 16
-#define DCMI_VDEV_FOR_RESERVE 32
-#define DCMI_VDEV_MAX_NUM 32
+void *dcmiHandle;
+#define SO_NOT_FOUND  (-99999)
+#define FUNCTION_NOT_FOUND  (-99998)
+#define SUCCESS  (0)
+#define ERROR_UNKNOWN  (-99997)
+#define CALL_FUNC(name, ...) if (name##_func == NULL) {return FUNCTION_NOT_FOUND;}return name##_func(__VA_ARGS__)
+#define DCMI_VDEV_FOR_RESERVE (32)
 struct dcmi_create_vdev_out {
     unsigned int vdev_id;
     unsigned int pcie_bus;
@@ -30,142 +31,84 @@ struct dcmi_create_vdev_out {
     unsigned char reserved[DCMI_VDEV_FOR_RESERVE];
 };
 
-enum dcmi_main_cmd {
-    DCMI_MAIN_CMD_DVPP = 0,
-    DCMI_MAIN_CMD_ISP,
-    DCMI_MAIN_CMD_TS_GROUP_NUM,
-    DCMI_MAIN_CMD_CAN,
-    DCMI_MAIN_CMD_UART,
-    DCMI_MAIN_CMD_UPGRADE,
-    DCMI_MAIN_CMD_TEMP = 50,
-    DCMI_MAIN_CMD_SVM = 51,
-    DCMI_MAIN_CMD_VDEV_MNG,
-    DCMI_MAIN_CMD_DEVICE_SHARE = 0x8001,
-    DCMI_MAIN_CMD_EX_CERT = 0x8003,
-    DCMI_MAIN_CMD_MAX
-};
+// dcmi
+int (*dcmi_init_func)();
+int dcmi_init()
+{
+    CALL_FUNC(dcmi_init);
+}
 
-/* DSMI sub vdev mng CMD def */
-typedef enum {
-    DCMI_VMNG_SUB_CMD_GET_VDEV_RESOURCE,
-    DCMI_VMNG_SUB_CMD_GET_TOTAL_RESOURCE,
-    DCMI_VMNG_SUB_CMD_GET_FREE_RESOURCE,
-    DCMI_VMNG_SUB_CMD_MAX,
-} DCMI_VDEV_MNG_SUB_CMD;
+int (*dcmi_get_card_num_list_func)(int *card_num, int *card_list, int list_length);
+int dcmi_get_card_num_list(int *card_num, int *card_list, int list_length)
+{
+    CALL_FUNC(dcmi_get_card_num_list, card_num, card_list, list_length);
+}
 
-struct dcmi_base_resource {
-    unsigned long long token;
-    unsigned long long token_max;
-    unsigned long long task_timeout;
-    unsigned int vfg_id;
-    unsigned char vip_mode;
-    unsigned char reserved[DCMI_VDEV_FOR_RESERVE - 1];  /* bytes aligned */
-};
+int (*dcmi_get_device_num_in_card_func)(int card_id, int *device_num);
+int dcmi_get_device_num_in_card(int card_id, int *device_num)
+{
+    CALL_FUNC(dcmi_get_device_num_in_card, card_id, device_num);
+}
 
-/* total types of computing resource */
-struct dcmi_computing_resource {
-    /* accelator resource */
-    float aic;
-    float aiv;
-    unsigned short dsa;
-    unsigned short rtsq;
-    unsigned short acsq;
-    unsigned short cdqm;
-    unsigned short c_core;
-    unsigned short ffts;
-    unsigned short sdma;
-    unsigned short pcie_dma;
+int (*dcmi_get_device_logic_id_func)(int *device_logic_id, int card_id, int device_id);
+int dcmi_get_device_logic_id(int *device_logic_id, int card_id, int device_id)
+{
+    CALL_FUNC(dcmi_get_device_logic_id, device_logic_id, card_id, device_id);
+}
 
-    /* memory resource, MB as unit */
-    unsigned long long memory_size;
+int (*dcmi_create_vdevice_func)(int card_id, int device_id, int vdev_id, const char *template_name,
+                                struct dcmi_create_vdev_out *out);
+int dcmi_create_vdevice(int card_id, int device_id, int vdev_id, const char *template_name,
+                        struct dcmi_create_vdev_out *out)
+{
+    CALL_FUNC(dcmi_create_vdevice, card_id, device_id, vdev_id, template_name, out);
+}
 
-    /* id resource */
-    unsigned int event_id;
-    unsigned int notify_id;
-    unsigned int stream_id;
-    unsigned int model_id;
+int (*dcmi_set_destroy_vdevice_func)(int card_id, int device_id, unsigned int VDevid);
+int dcmi_set_destroy_vdevice(int card_id, int device_id, unsigned int VDevid)
+{
+    CALL_FUNC(dcmi_set_destroy_vdevice, card_id, device_id, VDevid);
+}
 
-    /* cpu resource */
-    unsigned short topic_schedule_aicpu;
-    unsigned short host_ctrl_cpu;
-    unsigned short host_aicpu;
-    unsigned short device_aicpu;
-    unsigned short topic_ctrl_cpu_slot;
+int (*dcmi_get_device_logicid_from_phyid_func)(unsigned int phyid, unsigned int *logicid);
+int dcmi_get_device_logicid_from_phyid(unsigned int phyid, unsigned int *logicid)
+{
+    CALL_FUNC(dcmi_get_device_logicid_from_phyid, phyid, logicid);
+}
 
-    unsigned char reserved[DCMI_VDEV_FOR_RESERVE];
-};
+// load .so files and functions
+int dcmiInit_dl(void)
+{
+    dcmiHandle = dlopen("libdcmi.so", RTLD_LAZY | RTLD_GLOBAL);
+    if (dcmiHandle == NULL) {
+        fprintf (stderr, "%s\n", dlerror());
+        return SO_NOT_FOUND;
+    }
 
-struct dcmi_media_resource {
-    /* dvpp resource */
-    float jpegd;
-    float jpege;
-    float vpc;
-    float vdec;
-    float pngd;
-    float venc;
-    unsigned char reserved[DCMI_VDEV_FOR_RESERVE];
-};
+    dcmi_init_func = dlsym(dcmiHandle, "dcmi_init");
 
-struct dcmi_vdev_query_info {
-    char name[DCMI_VDEV_RES_NAME_LEN];
-    unsigned int status;
-    unsigned int is_container_used;
-    unsigned int vfid;
-    unsigned int vfg_id;
-    unsigned long long container_id;
-    struct dcmi_base_resource base;
-    struct dcmi_computing_resource computing;
-    struct dcmi_media_resource media;
-};
+    dcmi_get_card_num_list_func = dlsym(dcmiHandle, "dcmi_get_card_num_list");
 
-/* for single search */
-// vdev
-struct dcmi_vdev_query_stru {
-    unsigned int vdev_id;
-    struct dcmi_vdev_query_info query_info;
-};
+    dcmi_get_device_num_in_card_func = dlsym(dcmiHandle, "dcmi_get_device_num_in_card");
 
-// free
-struct dcmi_soc_free_resource {
-    unsigned int vfg_num;
-    unsigned int vfg_bitmap;
-    struct dcmi_base_resource base;
-    struct dcmi_computing_resource computing;
-    struct dcmi_media_resource media;
-};
+    dcmi_get_device_logic_id_func = dlsym(dcmiHandle, "dcmi_get_device_logic_id");
 
-// total
-struct dcmi_soc_total_resource {
-    unsigned int vdev_num;
-    unsigned int vdev_id[DCMI_VDEV_MAX_NUM];
-    unsigned int vfg_num;
-    unsigned int vfg_bitmap;
-    struct dcmi_base_resource base;
-    struct dcmi_computing_resource computing;
-    struct dcmi_media_resource media;
-};
+    dcmi_create_vdevice_func = dlsym(dcmiHandle, "dcmi_create_vdevice");
 
-enum dcmi_unit_type {
-    NPU_TYPE = 0,
-    MCU_TYPE = 1,
-    CPU_TYPE = 2,
-    INVALID_TYPE = 0xFF
-};
+    dcmi_set_destroy_vdevice_func = dlsym(dcmiHandle, "dcmi_set_destroy_vdevice");
 
-DCMIDLLEXPORT int dcmi_init(void);
+    dcmi_get_device_logicid_from_phyid_func = dlsym(dcmiHandle, "dcmi_get_device_logicid_from_phyid");
 
-DCMIDLLEXPORT int dcmi_get_card_num_list(int *card_num, int *card_list, int list_len);
+    return SUCCESS;
+}
 
-DCMIDLLEXPORT int dcmi_get_device_num_in_card(int card_id, int *device_num);
-
-DCMIDLLEXPORT int dcmi_create_vdevice(int card_id, int device_id, int vdev_id, const char *template_name,
-    struct dcmi_create_vdev_out *out);
-
-DCMIDLLEXPORT int dcmi_set_destroy_vdevice(int card_id, int device_id, unsigned int vdevid);
-
-DCMIDLLEXPORT int dcmi_get_device_logic_id(int *device_logic_id, int card_id, int device_id);
-
-DCMIDLLEXPORT int dcmi_get_device_type(int card_id, int device_id, enum dcmi_unit_type *device_type);
+int dcmiShutDown(void)
+{
+    if (dcmiHandle == NULL) {
+        return SUCCESS;
+    }
+    return (dlclose(dcmiHandle) ? ERROR_UNKNOWN : SUCCESS);
+}
 
 #ifdef __cplusplus
 #if __cplusplus
diff --git a/runtime/main.go b/runtime/main.go
index ab6d095..135ba52 100644
--- a/runtime/main.go
+++ b/runtime/main.go
@@ -100,7 +100,7 @@ func addHook(spec *specs.Spec) error {
 		})
 	}
 
-	vdevice, err := dcmi.CreateVDevice(spec)
+	vdevice, err := dcmi.CreateVDevice(&dcmi.NpuWorker{}, spec)
 
 	if err != nil {
 		return err