Commit 9cd20184 authored by songlinfeng's avatar songlinfeng
Browse files

support dcu-tracker for --gpus

parent 48c080c4
......@@ -117,7 +117,8 @@ func NewCommonHCUDiscoverer(logger logger.Interface, dtkCDIHookPath string, driv
var trackHook Hook
value := containerImage.Getenv(image.EnvVarDTKVisibleDevices)
if len(value) > 0 {
value1 := containerImage.Getenv(image.EnvVarNvidiaVisibleDevices)
if len(value) > 0 || len(value1) > 0{
trackHook = CreateTrackHook(dtkCDIHookPath, containerImage.ContainerId)
}
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment