From e3dd544e1c0a1cd4eb7e5d28412f845ef47ade0e Mon Sep 17 00:00:00 2001
From: limengxuan <391013634@qq.com>
Date: Mon, 24 Jun 2024 12:13:24 +0800
Subject: [PATCH] update

---
 internal/pkg/api/device/devices.go | 2 ++
 k8s-mock-plugin.yaml               | 4 +---
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/internal/pkg/api/device/devices.go b/internal/pkg/api/device/devices.go
index ab0504e..ef278a5 100644
--- a/internal/pkg/api/device/devices.go
+++ b/internal/pkg/api/device/devices.go
@@ -80,6 +80,8 @@ func Initialize() {
 func GlobalFlagSet() {
 	nvidia.ParseConfig()
 	cambricon.ParseConfig()
+	iluvatar.ParseConfig()
+	ascend.ParseConfig()
 	flag.BoolVar(&DebugMode, "debug", false, "debug mode")
 }
 
diff --git a/k8s-mock-plugin.yaml b/k8s-mock-plugin.yaml
index 14aaa4d..8d05164 100644
--- a/k8s-mock-plugin.yaml
+++ b/k8s-mock-plugin.yaml
@@ -19,9 +19,7 @@ spec:
       - key: CriticalAddonsOnly
         operator: Exists
       containers:
-      - image: ubuntu:20.04
-        #command: ["/bin/bash","-c","source /opt/hygondriver/env.sh && sleep infinity"]
-        command: ["sleep","infinity"]
+      - image: projecthami/mock-device-plugin:latest 
         name: hami-mock-dp-cntr
         env:
           - name: NODE_NAME