Skip to content

Instantly share code, notes, and snippets.

@narwanimonish
Created January 11, 2020 14:55
Show Gist options
  • Save narwanimonish/f8a95f929b7242dd4daa74a0e2afa5e5 to your computer and use it in GitHub Desktop.
Save narwanimonish/f8a95f929b7242dd4daa74a0e2afa5e5 to your computer and use it in GitHub Desktop.
Simulate CPU load on all cores
from multiprocessing import Pool
from multiprocessing import cpu_count
import signal
stop_loop = 0
def exit_chld(x, y):
global stop_loop
stop_loop = 1
def f(x):
global stop_loop
while not stop_loop:
x*x
signal.signal(signal.SIGINT, exit_chld)
if __name__ == '__main__':
processes = cpu_count()
print('-' * 20)
print('Running load on CPU(s)')
print('Utilizing %d cores' % processes)
print('-' * 20)
pool = Pool(processes)
pool.map(f, range(processes))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment