def _del_hp_fstab_mount(pagesize): log("Remove {} mountpoint from fstab".format(pagesize)) mnt_point = '/dev/hugepages{}'.format(pagesize) lfstab = fstab.Fstab() fstab_entry = lfstab.get_entry_by_attr('mountpoint', mnt_point) if fstab_entry: umount(mnt_point) lfstab.remove_entry(fstab_entry)
def _add_hp_fstab_mount(pagesize, mount=True): mnt_point = '/dev/hugepages{}'.format(pagesize) mkdir(mnt_point, owner='root', group='root', perms=0o755) lfstab = fstab.Fstab() fstab_entry = lfstab.get_entry_by_attr('mountpoint', mnt_point) if fstab_entry: lfstab.remove_entry(fstab_entry) entry = lfstab.Entry('hugetlbfs', mnt_point, 'hugetlbfs', 'pagesize={}'.format(pagesize), 0, 0) lfstab.add_entry(entry) if mount: fstab_mount(mnt_point)
def _add_hp_fstab_mount(pagesize, mount=True): log("Add {} mountpoint from fstab".format(pagesize)) mnt_point = '/dev/hugepages{}'.format(pagesize) mkdir(mnt_point, owner='root', group='root', perms=0o755) lfstab = fstab.Fstab() fstab_entry = lfstab.get_entry_by_attr('mountpoint', mnt_point) if fstab_entry: lfstab.remove_entry(fstab_entry) # use different device name for 1G and 2M. # this name actually is not used by the system # but add_antry filter by device name. device = 'hugetlbfs{}'.format(pagesize) entry = lfstab.Entry(device, mnt_point, 'hugetlbfs', 'pagesize={}'.format(pagesize), 0, 0) lfstab.add_entry(entry) if mount: fstab_mount(mnt_point)
def hugepage_support(user, group='hugetlb', nr_hugepages=256, max_map_count=65536, mnt_point='/run/hugepages/kvm', pagesize='2MB', mount=True, set_shmmax=False): """Enable hugepages on system. Args: user (str) -- Username to allow access to hugepages to group (str) -- Group name to own hugepages nr_hugepages (int) -- Number of pages to reserve max_map_count (int) -- Number of Virtual Memory Areas a process can own mnt_point (str) -- Directory to mount hugepages on pagesize (str) -- Size of hugepages mount (bool) -- Whether to Mount hugepages """ group_info = add_group(group) gid = group_info.gr_gid add_user_to_group(user, group) if max_map_count < 2 * nr_hugepages: max_map_count = 2 * nr_hugepages sysctl_settings = { 'vm.nr_hugepages': nr_hugepages, 'vm.max_map_count': max_map_count, 'vm.hugetlb_shm_group': gid, } if set_shmmax: shmmax_current = int(check_output(['sysctl', '-n', 'kernel.shmmax'])) shmmax_minsize = bytes_from_string(pagesize) * nr_hugepages if shmmax_minsize > shmmax_current: sysctl_settings['kernel.shmmax'] = shmmax_minsize sysctl.create(yaml.dump(sysctl_settings), '/etc/sysctl.d/10-hugepage.conf') mkdir(mnt_point, owner='root', group='root', perms=0o755, force=False) lfstab = fstab.Fstab() fstab_entry = lfstab.get_entry_by_attr('mountpoint', mnt_point) if fstab_entry: lfstab.remove_entry(fstab_entry) entry = lfstab.Entry('nodev', mnt_point, 'hugetlbfs', 'mode=1770,gid={},pagesize={}'.format(gid, pagesize), 0, 0) lfstab.add_entry(entry) if mount: fstab_mount(mnt_point)