From b0bde3a36aa4b7ee9d43a7de1782bf9e6f8075fd Mon Sep 17 00:00:00 2001 From: BianTanggui Date: Wed, 15 Mar 2023 10:02:30 +0800 Subject: [PATCH] Match-id-5a6ca6e3c42a6babbbe9904bca30e78ab84de696 --- hook/main.go | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/hook/main.go b/hook/main.go index 9a88bab..f7af535 100644 --- a/hook/main.go +++ b/hook/main.go @@ -40,13 +40,13 @@ const ( operateLogPath = "/var/log/ascend-docker-runtime/hook-operate.log" ascendRuntimeOptions = "ASCEND_RUNTIME_OPTIONS" ascendRuntimeMounts = "ASCEND_RUNTIME_MOUNTS" + ascendVisibleDevices = "ASCEND_VISIBLE_DEVICES" ascendDockerCli = "ascend-docker-cli" defaultAscendDockerCli = "/usr/local/bin/ascend-docker-cli" configDir = "/etc/ascend-docker-runtime.d" baseConfig = "base" configFileSuffix = "list" - borderNum = 2 kvPairSize = 2 maxCommandLength = 65535 ) @@ -327,6 +327,10 @@ func doPrestartHook() error { return fmt.Errorf("failed to get container config: %v", err) } + if visibleDevices := getValueByKey(containerConfig.Env, ascendVisibleDevices); visibleDevices == "" { + return nil + } + mountConfigs := parseMounts(getValueByKey(containerConfig.Env, ascendRuntimeMounts)) fileMountList, dirMountList, err := readConfigsOfDir(configDir, mountConfigs)