Skip to content

Commit f57e7cc

Browse files
[Feature] Support Tenstorrent's Wormhole accelerators #2573 (mount /dev/hugepages-1G) (#2589)
1 parent cda4583 commit f57e7cc

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

runner/internal/shim/docker.go

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1073,6 +1073,14 @@ func configureGpus(config *container.Config, hostConfig *container.HostConfig, v
10731073
},
10741074
)
10751075
}
1076+
// Check and mount hugepages-1G if it exists
1077+
if _, err := os.Stat("/dev/hugepages-1G"); err == nil {
1078+
hostConfig.Mounts = append(hostConfig.Mounts, mount.Mount{
1079+
Type: mount.TypeBind,
1080+
Source: "/dev/hugepages-1G",
1081+
Target: "/dev/hugepages-1G",
1082+
})
1083+
}
10761084
case host.GpuVendorIntel:
10771085
// All options are listed here:
10781086
// https://docs.habana.ai/en/latest/Installation_Guide/Additional_Installation/Docker_Installation.html

0 commit comments

Comments
 (0)