Skip to content

A Python library to capture the energy consumption of code snippets

License

Notifications You must be signed in to change notification settings

powerapi-ng/pyJoules

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

PyJoules

License: MIT Build Status Doc Status

About

pyJoules is a software toolkit to measure the energy footprint of a host machine along the execution of a piece of Python code. It monitors the energy consumed by specific device of the host machine such as :

  • intel CPU socket package
  • RAM (for intel server architectures)
  • intel integrated GPU (for client architectures)
  • nvidia GPU

Limitation

CPU, RAM and integrated GPU

pyJoules uses the Intel "Running Average Power Limit" (RAPL) technology that estimates power consumption of the CPU, ram and integrated GPU. This technology is available on Intel CPU since the Sandy Bridge generation(2010).

Nvidia GPU

pyJoules uses the nvidia "Nvidia Management Library" technology to measure energy consumption of nvidia devices. The energy measurement API is only available on nvidia GPU with Volta architecture(2018)

Windows and MacOS

Only GNU/Linux support is available for the moment. We are working on Mac support

Known issues

RAPL energy counters overflow after several minutes or hours, potentially causing false-negative energy readings.

pyJoules takes this into account and adds the counter's maximum possible value, max_energy_range_uj, to negative energy measurements. However, if a counter overflows twice during a single energy measurement, the reported energy will be max_energy_range_uj less than the expected value.

Installation

Measurement frequency

PyJoule use hardware measurement tools (intel RAPL, nvidia GPU tools, ...) to measure device energy consumption. Theses tools have a mesasurement frequency that depend of the device. Thus, you can't use Pyjoule to measure energy consumption during a period shorter than the device energy measurement frequency. Pyjoule will return null values if the measurement period is to short.

Requirements

  • python >= 3.7
  • nvml (if you want nvidia GPU support)

Installation

You can install pyJoules with pip: pip install pyJoules

if you want to use pyJoule to also measure nvidia GPU energy consumption, you have to install it with nvidia driver support using this command : pip install pyJoules[nvidia].

Basic usage

This Readme describe basic usage of pyJoules. For more in depth description, read the documentation here

Here are some basic usages of pyJoules. Please note that the reported energy consumption is not only the energy consumption of the code you are running. This includes the global energy consumption of all the process running on the machine during this period, thus including the operating system and other applications. That is why we recommend to eliminate any extra programs that may alter the energy consumption of the machine hosting experiments and to keep only the code under measurement (i.e., no extra applications, such as graphical interface, background running task...). This will give the closest measure to the real energy consumption of the measured code.

Decorate a function to measure its energy consumption

To measure the energy consumed by the machine during the execution of the function foo() run the following code:

from pyJoules.energy_meter import measure_energy

@measure_energy
def foo():
	# Instructions to be evaluated.

foo()

This will print on the console the recorded energy consumption of all the monitorable devices during the execution of function foo.

Output description

decorator basic usage will print iformation with this format :

begin timestamp : XXX; tag : YYY; duration : ZZZ;device_name: AAAA

with :

  • begin timestamp : monitored function launching time
  • tag: tag of the measure, if nothing is specified, this will be the function name
  • duration: function execution duration
  • device_name: power consumption of the device device_name in uJ

for cpu and ram devices, device_name match the RAPL domain described on the image below plus the CPU socket id. Rapl domain are described here

Configure the decorator specifying the device to monitor

You can easily configure which device to monitor using the parameters of the measureit decorator. For example, the following example only monitors the CPU power consumption on the CPU socket 1 and the Nvidia GPU 0. By default, pyJoules monitors all the available devices of the CPU sockets.

from pyJoules.energy_meter import measure_energy
from pyJoules.device.rapl_device import RaplPackageDomain
from pyJoules.device.nvidia_device import NvidiaGPUDomain
	
@measure_energy(domains=[RaplPackageDomain(1), NvidiaGPUDomain(0)])
def foo():
	# Instructions to be evaluated.
	
foo()	

You can append the following domain list to monitor them :

  • pyJoules.device.rapl_device.RaplPackageDomain : CPU (specify the socket id in parameter)
  • pyJoules.device.rapl_device.RaplDramDomain : RAM (specify the socket id in parameter)
  • pyJoules.device.rapl_device.RaplUncoreDomain : integrated GPU (specify the socket id in parameter)
  • pyJoules.device.rapl_device.RaplCoreDomain : RAPL Core domain (specify the socket id in parameter)
  • pyJoules.device.nvidia_device.NvidiaGPUDomain : Nvidia GPU (specify the socket id in parameter)

to understand which par of the cpu each RAPL domain monitor, see this section

Configure the output of the decorator

If you want to handle data with different output than the standard one, you can configure the decorator with an EnergyHandler instance from the pyJoules.handler module.

As an example, if you want to write the recorded energy consumption in a .csv file:

from pyJoules.energy_meter import measure_energy
from pyJoules.handler.csv_handler import CSVHandler
	
csv_handler = CSVHandler('result.csv')
	
@measure_energy(handler=csv_handler)
def foo():
	# Instructions to be evaluated.

for _ in range(100):
	foo()
		
csv_handler.save_data()

This will produce a csv file of 100 lines. Each line containing the energy consumption recorded during one execution of the function foo. Other predefined Handler classes exist to export data to MongoDB and Panda dataframe.

Use a context manager to add tagged "breakpoint" in your measurment

If you want to know where is the "hot spots" where your python code consume the most energy you can add "breakpoints" during the measurement process and tag them to know amount of energy consumed between this breakpoints.

For this, you have to use a context manager to measure the energy consumption. It is configurable as the decorator. For example, here we use an EnergyContext to measure the power consumption of CPU 1 and nvidia gpu 0 and report it in a csv file :

from pyJoules.energy_meter import EnergyContext
from pyJoules.device.rapl_device import RaplPackageDomain
from pyJoules.device.nvidia_device import NvidiaGPUDomain
from pyJoules.handler.csv_handler import CSVHandler
	
csv_handler = CSVHandler('result.csv')

with EnergyContext(handler=csv_handler, domains=[RaplPackageDomain(1), NvidiaGPUDomain(0)], start_tag='foo') as ctx:
	foo()
	ctx.record(tag='bar')
	bar()

csv_handler.save_data()

This will record the energy consumed :

  • between the beginning of the EnergyContext and the call of the ctx.record method
  • between the call of the ctx.record method and the end of the EnergyContext

Each measured part will be written in the csv file. One line per part.

RAPL domain description

RAPL domains match part of the cpu socket as described in this image :

  • Package : correspond to the wall cpu energy consumption
  • core : correpond to the sum of all cpu core energy consumption
  • uncore : correspond to the integrated GPU

Miscellaneous

About

pyJoules is an open-source project developed by the Spirals research group (University of Lille and Inria) that is part of the PowerAPI initiative.

The documentation is available here.

Mailing list

You can follow the latest news and asks questions by subscribing to our mailing list.

Contributing

If you would like to contribute code, you can do so via GitHub by forking the repository and sending a pull request.

When submitting code, please make every effort to follow existing coding conventions and style in order to keep the code as readable as possible.