動態切換 Linux 使用的 CPU 數量

由於要測試一些代碼，其運行結果會受到多核並行的影響，所以希望能夠調整使用的 CPU 數量。網絡上之前看到的方法是在內核的啟動參數上添加一個 maxcpus，但是如果這樣的話每切換一次都要重啟一次，是在太麻煩了。想想 Linux 應該是很強大的，所以可以動態修改 CPU 數量才對。

無意中看到 Linux 代碼的 Documentation 文件夾下有個文件叫做 cpu-hotplug.txt，於是就看了一下，發現可以在 /sys/devices/system/cpu 看到代表各 CPU 的文件夾按照 cpuX 的命名方式，如 cpu0、cpu1、cpu2 等。這些文件夾裡面有一個 online 文件，如果其值為0則禁用該 CPU，如果為1則啟用該 CPU。注意，這裡需要 root 權限哦。

因為我只要在單核和多核之間切換，所以我寫了兩個腳本放在 /usr/local/sbin 裡面：

singlecore

#!/bin/bash
 
cpus_dir="/sys/devices/system/cpu"
 
for cpu in $(ls "$cpus_dir" | grep 'cpu[0-9]\+')
do
	cpu_online="$cpus_dir/$cpu/online"
	if [[ -e "$cpu_online" && $(cat $cpu_online) = 1 ]]
	then
		echo 0 > "$cpu_online"
	fi
done

-----------------------
multicore
-----------------------

#!/bin/bash
 
cpus_dir="/sys/devices/system/cpu"
 
for cpu in $(ls "$cpus_dir" | grep 'cpu[0-9]\+')
do
	cpu_online="$cpus_dir/$cpu/online"
	if [[ -e "$cpu_online" && $(cat $cpu_online) = 0 ]]
	then
		echo 1 > "$cpu_online"
	fi
done

之後需要切換的時候，只要運行 sudo singlecore 或者 sudo multicore 就可以了~

順便說一句，我當時在想，如果我禁用了所有的 CPU 會怎麼樣呢？結果發現 cpu0 是沒有 online 文件的，也就是 Linux 至少保證一個 CPU 處於可用狀態。