Python | os.sched_setaffinity() method

OS module in Python provides functions for interacting with the operating system. OS comes under Python’s standard utility modules. This module provides a portable way of using operating system dependent functionality.

os.sched_setaffinity() method in Python is used to set the CPU affinity mask of a process indicated by the specified process id. A process’s CPU affinity mask determines the set of CPUs on which it is eligible to run.

Note: This method is only available on some UNIX platforms.



Syntax: os.sched_setaffinity(pid, mask)

Parameter:
pid: The process id of the process whose CPU affinity mask is to be set required. Process’s CPU affinity mask determines the set of CPUs on which it is eligible to run.
A pid of 0 represents the calling process.
mask: An iterable of integers representing the set of CPUs to which the process should be restricted.

Return Type: This method does not return any value.

Code: Use of os.sched_setaffinity() method

filter_none

edit
close

play_arrow

link
brightness_4
code

# Python program to explain os.sched_setaffinity() method  
  
# importing os module 
import os
  
# Get the number of CPUs
# in the system
# using os.cpu_count() method
print("Number of CPUs:", os.cpu_count())
  
# Get the set of CPUs
# on which the calling process
# is eligible to run. using
# os.sched_getaffinity() method
# 0 as PID represnts the
# calling process
pid = 0
affinity = os.sched_getaffinity(pid)
  
# Print the result
print("Process is eligibl to run on:", affinity)
  
  
# Change the CPU affinity mask
# of the calling process
# using os.sched_setaffinity() method
  
# Below CPU affinity mask will
# restrict a process to only
# these 2 CPUs (0, 1) i.e process can
# run on these CPUs only
affinity_mask = {0, 1}
pid = 0
os.sched_setaffinity(0, affinity_mask)
print("CPU affinity mask is modified for process id % s" % pid) 
  
  
# Now again, Get the set of CPUs
# on which the calling process
# is eligible to run.
pid = 0
affinity = os.sched_getaffinity(pid)
  
# Print the result
print("Now, process is eligibl to run on:", affinity)

chevron_right


Output:

Number of CPUs: 4
Process is eligibl to run on: {0, 1, 2, 3}
CPU affinity mask is modified for process id 0
Now, process is eligibl to run on: {0, 1}

References:



My Personal Notes arrow_drop_up

self motivated and passionate programmer

If you like GeeksforGeeks and would like to contribute, you can also write an article using contribute.geeksforgeeks.org or mail your article to contribute@geeksforgeeks.org. See your article appearing on the GeeksforGeeks main page and help other Geeks.

Please Improve this article if you find anything incorrect by clicking on the "Improve Article" button below.




Article Tags :

Be the First to upvote.


Please write to us at contribute@geeksforgeeks.org to report any issue with the above content.