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)