mirror of
https://github.com/clearlinux/dockerfiles.git
synced 2026-08-28 13:25:43 +00:00
Qi Zheng
58 lines
1.7 KiB
Bash
58 lines
1.7 KiB
Bash
#!/bin/sh
|
|
|
|
# get core numbers
|
|
lcores=$(grep "^processor" -c /proc/cpuinfo)
|
|
htn=$(lscpu | grep "^Thread(s) per core" | awk '{print $NF}')
|
|
pcores=$(awk 'BEGIN{ x="'$lcores'" / "'$htn'"; printf("%.0f\n", (x == int(x))?x:int(x)+1) }')
|
|
csets=$((0x$(cat /proc/self/status | grep "^Cpus_allowed:" | awk '{print $NF}')))
|
|
cfull=$(((1<<cores) - 1))
|
|
|
|
if (( csets == cfull )); then
|
|
# calculate physical core numbers
|
|
pnums=$pcores
|
|
lnums=$lcores
|
|
else
|
|
# if set cpu affinity
|
|
# get hyperthread sibling topology
|
|
st=0; i=0
|
|
until [[ $st > 0 || $i > $lcores ]]; do
|
|
if [ -r /sys/devices/system/cpu/cpu$i/topology/thread_siblings_list ]; then
|
|
x=$(awk -F, '{print $1}' /sys/devices/system/cpu/cpu$i/topology/thread_siblings_list)
|
|
y=$(awk -F, '{print $2}' /sys/devices/system/cpu/cpu$i/topology/thread_siblings_list)
|
|
st=$((y-x))
|
|
fi
|
|
((i++))
|
|
done
|
|
|
|
# calculate logical & physical core number by the toplogy
|
|
lnums=0; pnums=0; mask=0; n=csets
|
|
while (( n != 0 ))
|
|
do
|
|
m=$((n & (~n + 1)))
|
|
if (( (m & mask) == 0 )); then
|
|
mask=$((mask|m|m<<st))
|
|
((pnums++))
|
|
fi
|
|
|
|
n=$((n - m))
|
|
((lnums++))
|
|
done
|
|
fi
|
|
|
|
# cpu quota
|
|
qus=$(cat /sys/fs/cgroup/cpu/cpu.cfs_quota_us)
|
|
pus=$(cat /sys/fs/cgroup/cpu/cpu.cfs_period_us)
|
|
|
|
# calculate nums by assigned logical core and total physical core numbers
|
|
nums=$(((lnums < pcores)?lnums:pcores))
|
|
# calculate qnums by CPU quota and round up
|
|
if [[ $qus != "-1" ]]; then
|
|
qnums=$(awk 'BEGIN{ x="'$qus'" / "'$pus'"; printf("%.0f\n", (x == int(x))?x:int(x)+1) }')
|
|
nums=$(((nums < qnums)?nums:qnums))
|
|
fi
|
|
|
|
# for special single physical core case
|
|
nums=$(((nums == 2 && pnums == 1)?pnums:nums))
|
|
|
|
export OMP_NUM_THREADS=$nums
|