blob: 15238b9a13b4f806d63c24843e538e8ce78b4af3 (
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
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
|
#!/bin/bash
# Copyright 2016 Google Inc. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
total_cpus=$(nproc)
config_nvme()
{
current_cpu=0
for dev in /sys/bus/pci/drivers/nvme/*
do
if [ ! -d "$dev" ]
then
continue
fi
for irq_info in $dev/msi_irqs/*
do
if [ ! -f "$irq_info" ]
then
continue
fi
current_cpu=$((current_cpu % total_cpus))
cpu_mask=$(printf "%x" $((1<<current_cpu)))
irq=$(basename "$irq_info")$a
echo "Setting IRQ $irq smp_affinity to $cpu_mask."
echo "$cpu_mask" > "/proc/irq/$irq/smp_affinity"
current_cpu=$((current_cpu+1))
done
done
}
config_scsi()
{
irqs=()
for device in /sys/bus/virtio/drivers/virtio_scsi/virtio*
do
ssd=0
for target_path in $device/host*/target*/*
do
if [ ! -f "$target_path/model" ]
then
continue
fi
model=$(cat "$target_path/model")
if [[ $model =~ .*EphemeralDisk.* ]]
then
ssd=1
for queue_path in $target_path/block/sd*/queue
do
echo noop > "$queue_path/scheduler"
echo 0 > "$queue_path/add_random"
echo 512 > "$queue_path/nr_requests"
echo 0 > "$queue_path/rotational"
echo 0 > "$queue_path/rq_affinity"
echo 1 > "$queue_path/nomerges"
done
fi
done
if [[ $ssd == 1 ]]
then
request_queue=$(basename "$device")-request
irq=$(cat /proc/interrupts | grep "$request_queue" | awk '{print $1}'| sed 's/://')
irqs+=($irq)
fi
done
irq_count=${#irqs[@]}
if [ "$irq_count" != 0 ]
then
stride=$((total_cpus / irq_count))
stride=$((stride < 1 ? 1 : stride))
current_cpu=0
for irq in "${irqs[@]}"
do
current_cpu=$(($current_cpu % $total_cpus))
cpu_mask=$(printf "%x" $((1<<$current_cpu)))
echo "Setting IRQ $irq smp_affinity to $cpu_mask."
echo "$cpu_mask" > "/proc/irq/$irq/smp_affinity"
current_cpu=$((current_cpu+stride))
done
fi
}
config_nvme
config_scsi
|
