zishuo/ascend-docker-runtime
1
0
Fork 0
mirror of https://github.com/Ascend/ascend-docker-runtime.git synced 2025-11-01 05:42:40 +08:00
Code Issues Packages Projects Releases Wiki Activity

Match-id-5a6ca6e3c42a6babbbe9904bca30e78ab84de696

Browse Source
This commit is contained in:
BianTanggui
2023-03-15 10:02:30 +08:00
parent ccd1208d67
commit b0bde3a36a
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
hook/main.go
View File

@@ -40,13 +40,13 @@ const (
operateLogPath = "/var/log/ascend-docker-runtime/hook-operate.log" operateLogPath = "/var/log/ascend-docker-runtime/hook-operate.log"
ascendRuntimeOptions = "ASCEND_RUNTIME_OPTIONS" ascendRuntimeOptions = "ASCEND_RUNTIME_OPTIONS"
ascendRuntimeMounts = "ASCEND_RUNTIME_MOUNTS" ascendRuntimeMounts = "ASCEND_RUNTIME_MOUNTS"
ascendVisibleDevices = "ASCEND_VISIBLE_DEVICES"
ascendDockerCli = "ascend-docker-cli" ascendDockerCli = "ascend-docker-cli"
defaultAscendDockerCli = "/usr/local/bin/ascend-docker-cli" defaultAscendDockerCli = "/usr/local/bin/ascend-docker-cli"
configDir = "/etc/ascend-docker-runtime.d" configDir = "/etc/ascend-docker-runtime.d"
baseConfig = "base" baseConfig = "base"
configFileSuffix = "list" configFileSuffix = "list"
borderNum = 2
kvPairSize = 2 kvPairSize = 2
maxCommandLength = 65535 maxCommandLength = 65535
) )
@@ -327,6 +327,10 @@ func doPrestartHook() error {
return fmt.Errorf("failed to get container config: %v", err) 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)) mountConfigs := parseMounts(getValueByKey(containerConfig.Env, ascendRuntimeMounts))
fileMountList, dirMountList, err := readConfigsOfDir(configDir, mountConfigs) fileMountList, dirMountList, err := readConfigsOfDir(configDir, mountConfigs)