qemu-taskset.py

import sys
import psutil
import random

CPUSET = [int(x) for x in sys.argv[1].split(',')]
PID = int(sys.argv[2])

threads = []
for p in psutil.Process(PID).children(recursive=True):
	if 'qemu-system' in p.name():
		threads += p.threads()
# random.shuffle(threads)
threads.sort(key=lambda thread: thread.user_time + thread.system_time, reverse=True)

random.shuffle(CPUSET)
for i, thread in enumerate(threads):
	cpu = CPUSET[i % len(CPUSET)]
	psutil.Process(thread.id).cpu_affinity([cpu])
	# print("Thread %d bound to CPU #%d" % (thread.id, cpu))