blob: 3bb52d9cf9197e06a1ab6828ff76a8fd69fcc62c (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
|
#!/bin/bash
export ODP_PLATFORM_PARAMS=${ODP_PLATFORM_PARAMS:--n 4}
# where to mount huge pages
export HUGEPAGEDIR=${HUGEPAGEDIR:-/mnt/huge}
# Make sure huge pages are released when a unit test crashes "make check"
trap ctrl_c INT
ctrl_c() {
echo "** Trapped CTRL-C"
if grep -qs "$HUGEPAGEDIR" /proc/mounts; then
echo "** Umounting hugetlbfs"
sleep 1 && sudo umount -a -t hugetlbfs
fi
}
function mount_and_reserve() {
export PATH_NR="/sys/devices/system/node/node0/hugepages/hugepages-${SIZE_KB}kB/nr_hugepages"
export PATH_FREE="/sys/devices/system/node/node0/hugepages/hugepages-${SIZE_KB}kB/free_hugepages"
if grep -qs "$HUGEPAGEDIR" /proc/mounts; then
echo "Umounting hugetlbfs from previous use!"
sudo umount -a -t hugetlbfs
fi
echo "Trying $SIZE pages"
sudo mount -t hugetlbfs -o pagesize=$SIZE nodev $HUGEPAGEDIR 2>/dev/null
res=$?
if [ $res -ne 0 ]; then
echo "ERROR: can't mount hugepages"
return $res
fi
sudo sh -c "echo $RESERVE > $PATH_NR"
if [ `cat $PATH_NR` -lt 1 ]; then
echo "Failed to reserve at least 1 huge page!"
return 1
else
echo "Total number: `cat $PATH_NR`"
echo "Free pages: `cat $PATH_FREE`"
fi
}
if [ ! -d $HUGEPAGEDIR ]; then
sudo mkdir -p $HUGEPAGEDIR
fi
echo "Mounting hugetlbfs"
export SIZE=1G
export SIZE_KB=1048576
export RESERVE=1
mount_and_reserve
res=$?
if [ $res -ne 0 ]; then
export SIZE=2MB
export SIZE_KB=2048
export RESERVE=1024
mount_and_reserve
res=$?
if [ $res -ne 0 ]; then
echo "ERROR: can't mount hugepages with any size"
exit $res
fi
fi
echo "running $1!"
if [ ${1: -3} == ".sh" ]
then
sudo TEST_DIR=${TEST_DIR} \
ODP_PLATFORM_PARAMS="$ODP_PLATFORM_PARAMS" \
ODP_GDB=$ODP_GDB $1
else
sudo TEST_DIR=${TEST_DIR} ODP_PLATFORM_PARAMS="$ODP_PLATFORM_PARAMS" \
$ODP_GDB $1
fi
res=$?
echo "Unmounting hugetlbfs"
sleep 0.3 && sudo umount -a -t hugetlbfs
exit $res
|
