#!/usr/bin/env python
|
# SPDX-License-Identifier: GPL-2.0-only
|
# -*- coding: utf-8 -*-
|
#
|
""" This utility can be used to debug and tune the performance of the
|
intel_pstate driver. This utility can be used in two ways:
|
- If there is Linux trace file with pstate_sample events enabled, then
|
this utility can parse the trace file and generate performance plots.
|
- If user has not specified a trace file as input via command line parameters,
|
then this utility enables and collects trace data for a user specified interval
|
and generates performance plots.
|
|
Prerequisites:
|
Python version 2.7.x or higher
|
gnuplot 5.0 or higher
|
gnuplot-py 1.8 or higher
|
(Most of the distributions have these required packages. They may be called
|
gnuplot-py, phython-gnuplot or phython3-gnuplot, gnuplot-nox, ... )
|
|
HWP (Hardware P-States are disabled)
|
Kernel config for Linux trace is enabled
|
|
see print_help(): for Usage and Output details
|
|
"""
|
from __future__ import print_function
|
from datetime import datetime
|
import subprocess
|
import os
|
import time
|
import re
|
import signal
|
import sys
|
import getopt
|
import Gnuplot
|
from numpy import *
|
from decimal import *
|
|
__author__ = "Srinivas Pandruvada"
|
__copyright__ = " Copyright (c) 2017, Intel Corporation. "
|
__license__ = "GPL version 2"
|
|
|
MAX_CPUS = 256
|
|
# Define the csv file columns
|
C_COMM = 18
|
C_GHZ = 17
|
C_ELAPSED = 16
|
C_SAMPLE = 15
|
C_DURATION = 14
|
C_LOAD = 13
|
C_BOOST = 12
|
C_FREQ = 11
|
C_TSC = 10
|
C_APERF = 9
|
C_MPERF = 8
|
C_TO = 7
|
C_FROM = 6
|
C_SCALED = 5
|
C_CORE = 4
|
C_USEC = 3
|
C_SEC = 2
|
C_CPU = 1
|
|
global sample_num, last_sec_cpu, last_usec_cpu, start_time, testname
|
|
# 11 digits covers uptime to 115 days
|
getcontext().prec = 11
|
|
sample_num =0
|
last_sec_cpu = [0] * MAX_CPUS
|
last_usec_cpu = [0] * MAX_CPUS
|
|
def print_help():
|
print('intel_pstate_tracer.py:')
|
print(' Usage:')
|
print(' If the trace file is available, then to simply parse and plot, use (sudo not required):')
|
print(' ./intel_pstate_tracer.py [-c cpus] -t <trace_file> -n <test_name>')
|
print(' Or')
|
print(' ./intel_pstate_tracer.py [--cpu cpus] ---trace_file <trace_file> --name <test_name>')
|
print(' To generate trace file, parse and plot, use (sudo required):')
|
print(' sudo ./intel_pstate_tracer.py [-c cpus] -i <interval> -n <test_name> -m <kbytes>')
|
print(' Or')
|
print(' sudo ./intel_pstate_tracer.py [--cpu cpus] --interval <interval> --name <test_name> --memory <kbytes>')
|
print(' Optional argument:')
|
print(' cpus: comma separated list of CPUs')
|
print(' kbytes: Kilo bytes of memory per CPU to allocate to the trace buffer. Default: 10240')
|
print(' Output:')
|
print(' If not already present, creates a "results/test_name" folder in the current working directory with:')
|
print(' cpu.csv - comma seperated values file with trace contents and some additional calculations.')
|
print(' cpu???.csv - comma seperated values file for CPU number ???.')
|
print(' *.png - a variety of PNG format plot files created from the trace contents and the additional calculations.')
|
print(' Notes:')
|
print(' Avoid the use of _ (underscore) in test names, because in gnuplot it is a subscript directive.')
|
print(' Maximum number of CPUs is {0:d}. If there are more the script will abort with an error.'.format(MAX_CPUS))
|
print(' Off-line CPUs cause the script to list some warnings, and create some empty files. Use the CPU mask feature for a clean run.')
|
print(' Empty y range warnings for autoscaled plots can occur and can be ignored.')
|
|
def plot_perf_busy_with_sample(cpu_index):
|
""" Plot method to per cpu information """
|
|
file_name = 'cpu{:0>3}.csv'.format(cpu_index)
|
if os.path.exists(file_name):
|
output_png = "cpu%03d_perf_busy_vs_samples.png" % cpu_index
|
g_plot = common_all_gnuplot_settings(output_png)
|
# autoscale this one, no set y1 range
|
g_plot('set y2range [0:200]')
|
g_plot('set y2tics 0, 10')
|
g_plot('set title "{} : cpu perf busy vs. sample : CPU {:0>3} : {:%F %H:%M}"'.format(testname, cpu_index, datetime.now()))
|
# Override common
|
g_plot('set xlabel "Samples"')
|
g_plot('set ylabel "P-State"')
|
g_plot('set y2label "Scaled Busy/performance/io-busy(%)"')
|
set_4_plot_linestyles(g_plot)
|
g_plot('plot "' + file_name + '" using {:d}:{:d} with linespoints linestyle 1 axis x1y2 title "performance",\\'.format(C_SAMPLE, C_CORE))
|
g_plot('"' + file_name + '" using {:d}:{:d} with linespoints linestyle 2 axis x1y2 title "scaled-busy",\\'.format(C_SAMPLE, C_SCALED))
|
g_plot('"' + file_name + '" using {:d}:{:d} with linespoints linestyle 3 axis x1y2 title "io-boost",\\'.format(C_SAMPLE, C_BOOST))
|
g_plot('"' + file_name + '" using {:d}:{:d} with linespoints linestyle 4 axis x1y1 title "P-State"'.format(C_SAMPLE, C_TO))
|
|
def plot_perf_busy(cpu_index):
|
""" Plot some per cpu information """
|
|
file_name = 'cpu{:0>3}.csv'.format(cpu_index)
|
if os.path.exists(file_name):
|
output_png = "cpu%03d_perf_busy.png" % cpu_index
|
g_plot = common_all_gnuplot_settings(output_png)
|
# autoscale this one, no set y1 range
|
g_plot('set y2range [0:200]')
|
g_plot('set y2tics 0, 10')
|
g_plot('set title "{} : perf busy : CPU {:0>3} : {:%F %H:%M}"'.format(testname, cpu_index, datetime.now()))
|
g_plot('set ylabel "P-State"')
|
g_plot('set y2label "Scaled Busy/performance/io-busy(%)"')
|
set_4_plot_linestyles(g_plot)
|
g_plot('plot "' + file_name + '" using {:d}:{:d} with linespoints linestyle 1 axis x1y2 title "performance",\\'.format(C_ELAPSED, C_CORE))
|
g_plot('"' + file_name + '" using {:d}:{:d} with linespoints linestyle 2 axis x1y2 title "scaled-busy",\\'.format(C_ELAPSED, C_SCALED))
|
g_plot('"' + file_name + '" using {:d}:{:d} with linespoints linestyle 3 axis x1y2 title "io-boost",\\'.format(C_ELAPSED, C_BOOST))
|
g_plot('"' + file_name + '" using {:d}:{:d} with linespoints linestyle 4 axis x1y1 title "P-State"'.format(C_ELAPSED, C_TO))
|
|
def plot_durations(cpu_index):
|
""" Plot per cpu durations """
|
|
file_name = 'cpu{:0>3}.csv'.format(cpu_index)
|
if os.path.exists(file_name):
|
output_png = "cpu%03d_durations.png" % cpu_index
|
g_plot = common_all_gnuplot_settings(output_png)
|
# autoscale this one, no set y range
|
g_plot('set title "{} : durations : CPU {:0>3} : {:%F %H:%M}"'.format(testname, cpu_index, datetime.now()))
|
g_plot('set ylabel "Timer Duration (MilliSeconds)"')
|
# override common
|
g_plot('set key off')
|
set_4_plot_linestyles(g_plot)
|
g_plot('plot "' + file_name + '" using {:d}:{:d} with linespoints linestyle 1 axis x1y1'.format(C_ELAPSED, C_DURATION))
|
|
def plot_loads(cpu_index):
|
""" Plot per cpu loads """
|
|
file_name = 'cpu{:0>3}.csv'.format(cpu_index)
|
if os.path.exists(file_name):
|
output_png = "cpu%03d_loads.png" % cpu_index
|
g_plot = common_all_gnuplot_settings(output_png)
|
g_plot('set yrange [0:100]')
|
g_plot('set ytics 0, 10')
|
g_plot('set title "{} : loads : CPU {:0>3} : {:%F %H:%M}"'.format(testname, cpu_index, datetime.now()))
|
g_plot('set ylabel "CPU load (percent)"')
|
# override common
|
g_plot('set key off')
|
set_4_plot_linestyles(g_plot)
|
g_plot('plot "' + file_name + '" using {:d}:{:d} with linespoints linestyle 1 axis x1y1'.format(C_ELAPSED, C_LOAD))
|
|
def plot_pstate_cpu_with_sample():
|
""" Plot all cpu information """
|
|
if os.path.exists('cpu.csv'):
|
output_png = 'all_cpu_pstates_vs_samples.png'
|
g_plot = common_all_gnuplot_settings(output_png)
|
# autoscale this one, no set y range
|
# override common
|
g_plot('set xlabel "Samples"')
|
g_plot('set ylabel "P-State"')
|
g_plot('set title "{} : cpu pstate vs. sample : {:%F %H:%M}"'.format(testname, datetime.now()))
|
title_list = subprocess.check_output('ls cpu???.csv | sed -e \'s/.csv//\'',shell=True).decode('utf-8').replace('\n', ' ')
|
plot_str = "plot for [i in title_list] i.'.csv' using {:d}:{:d} pt 7 ps 1 title i".format(C_SAMPLE, C_TO)
|
g_plot('title_list = "{}"'.format(title_list))
|
g_plot(plot_str)
|
|
def plot_pstate_cpu():
|
""" Plot all cpu information from csv files """
|
|
output_png = 'all_cpu_pstates.png'
|
g_plot = common_all_gnuplot_settings(output_png)
|
# autoscale this one, no set y range
|
g_plot('set ylabel "P-State"')
|
g_plot('set title "{} : cpu pstates : {:%F %H:%M}"'.format(testname, datetime.now()))
|
|
# the following command is really cool, but doesn't work with the CPU masking option because it aborts on the first missing file.
|
# plot_str = 'plot for [i=0:*] file=sprintf("cpu%03d.csv",i) title_s=sprintf("cpu%03d",i) file using 16:7 pt 7 ps 1 title title_s'
|
#
|
title_list = subprocess.check_output('ls cpu???.csv | sed -e \'s/.csv//\'',shell=True).decode('utf-8').replace('\n', ' ')
|
plot_str = "plot for [i in title_list] i.'.csv' using {:d}:{:d} pt 7 ps 1 title i".format(C_ELAPSED, C_TO)
|
g_plot('title_list = "{}"'.format(title_list))
|
g_plot(plot_str)
|
|
def plot_load_cpu():
|
""" Plot all cpu loads """
|
|
output_png = 'all_cpu_loads.png'
|
g_plot = common_all_gnuplot_settings(output_png)
|
g_plot('set yrange [0:100]')
|
g_plot('set ylabel "CPU load (percent)"')
|
g_plot('set title "{} : cpu loads : {:%F %H:%M}"'.format(testname, datetime.now()))
|
|
title_list = subprocess.check_output('ls cpu???.csv | sed -e \'s/.csv//\'',shell=True).decode('utf-8').replace('\n', ' ')
|
plot_str = "plot for [i in title_list] i.'.csv' using {:d}:{:d} pt 7 ps 1 title i".format(C_ELAPSED, C_LOAD)
|
g_plot('title_list = "{}"'.format(title_list))
|
g_plot(plot_str)
|
|
def plot_frequency_cpu():
|
""" Plot all cpu frequencies """
|
|
output_png = 'all_cpu_frequencies.png'
|
g_plot = common_all_gnuplot_settings(output_png)
|
# autoscale this one, no set y range
|
g_plot('set ylabel "CPU Frequency (GHz)"')
|
g_plot('set title "{} : cpu frequencies : {:%F %H:%M}"'.format(testname, datetime.now()))
|
|
title_list = subprocess.check_output('ls cpu???.csv | sed -e \'s/.csv//\'',shell=True).decode('utf-8').replace('\n', ' ')
|
plot_str = "plot for [i in title_list] i.'.csv' using {:d}:{:d} pt 7 ps 1 title i".format(C_ELAPSED, C_FREQ)
|
g_plot('title_list = "{}"'.format(title_list))
|
g_plot(plot_str)
|
|
def plot_duration_cpu():
|
""" Plot all cpu durations """
|
|
output_png = 'all_cpu_durations.png'
|
g_plot = common_all_gnuplot_settings(output_png)
|
# autoscale this one, no set y range
|
g_plot('set ylabel "Timer Duration (MilliSeconds)"')
|
g_plot('set title "{} : cpu durations : {:%F %H:%M}"'.format(testname, datetime.now()))
|
|
title_list = subprocess.check_output('ls cpu???.csv | sed -e \'s/.csv//\'',shell=True).decode('utf-8').replace('\n', ' ')
|
plot_str = "plot for [i in title_list] i.'.csv' using {:d}:{:d} pt 7 ps 1 title i".format(C_ELAPSED, C_DURATION)
|
g_plot('title_list = "{}"'.format(title_list))
|
g_plot(plot_str)
|
|
def plot_scaled_cpu():
|
""" Plot all cpu scaled busy """
|
|
output_png = 'all_cpu_scaled.png'
|
g_plot = common_all_gnuplot_settings(output_png)
|
# autoscale this one, no set y range
|
g_plot('set ylabel "Scaled Busy (Unitless)"')
|
g_plot('set title "{} : cpu scaled busy : {:%F %H:%M}"'.format(testname, datetime.now()))
|
|
title_list = subprocess.check_output('ls cpu???.csv | sed -e \'s/.csv//\'',shell=True).decode('utf-8').replace('\n', ' ')
|
plot_str = "plot for [i in title_list] i.'.csv' using {:d}:{:d} pt 7 ps 1 title i".format(C_ELAPSED, C_SCALED)
|
g_plot('title_list = "{}"'.format(title_list))
|
g_plot(plot_str)
|
|
def plot_boost_cpu():
|
""" Plot all cpu IO Boosts """
|
|
output_png = 'all_cpu_boost.png'
|
g_plot = common_all_gnuplot_settings(output_png)
|
g_plot('set yrange [0:100]')
|
g_plot('set ylabel "CPU IO Boost (percent)"')
|
g_plot('set title "{} : cpu io boost : {:%F %H:%M}"'.format(testname, datetime.now()))
|
|
title_list = subprocess.check_output('ls cpu???.csv | sed -e \'s/.csv//\'',shell=True).decode('utf-8').replace('\n', ' ')
|
plot_str = "plot for [i in title_list] i.'.csv' using {:d}:{:d} pt 7 ps 1 title i".format(C_ELAPSED, C_BOOST)
|
g_plot('title_list = "{}"'.format(title_list))
|
g_plot(plot_str)
|
|
def plot_ghz_cpu():
|
""" Plot all cpu tsc ghz """
|
|
output_png = 'all_cpu_ghz.png'
|
g_plot = common_all_gnuplot_settings(output_png)
|
# autoscale this one, no set y range
|
g_plot('set ylabel "TSC Frequency (GHz)"')
|
g_plot('set title "{} : cpu TSC Frequencies (Sanity check calculation) : {:%F %H:%M}"'.format(testname, datetime.now()))
|
|
title_list = subprocess.check_output('ls cpu???.csv | sed -e \'s/.csv//\'',shell=True).decode('utf-8').replace('\n', ' ')
|
plot_str = "plot for [i in title_list] i.'.csv' using {:d}:{:d} pt 7 ps 1 title i".format(C_ELAPSED, C_GHZ)
|
g_plot('title_list = "{}"'.format(title_list))
|
g_plot(plot_str)
|
|
def common_all_gnuplot_settings(output_png):
|
""" common gnuplot settings for multiple CPUs one one graph. """
|
|
g_plot = common_gnuplot_settings()
|
g_plot('set output "' + output_png + '"')
|
return(g_plot)
|
|
def common_gnuplot_settings():
|
""" common gnuplot settings. """
|
|
g_plot = Gnuplot.Gnuplot(persist=1)
|
# The following line is for rigor only. It seems to be assumed for .csv files
|
g_plot('set datafile separator \",\"')
|
g_plot('set ytics nomirror')
|
g_plot('set xtics nomirror')
|
g_plot('set xtics font ", 10"')
|
g_plot('set ytics font ", 10"')
|
g_plot('set tics out scale 1.0')
|
g_plot('set grid')
|
g_plot('set key out horiz')
|
g_plot('set key bot center')
|
g_plot('set key samplen 2 spacing .8 font ", 9"')
|
g_plot('set term png size 1200, 600')
|
g_plot('set title font ", 11"')
|
g_plot('set ylabel font ", 10"')
|
g_plot('set xlabel font ", 10"')
|
g_plot('set xlabel offset 0, 0.5')
|
g_plot('set xlabel "Elapsed Time (Seconds)"')
|
return(g_plot)
|
|
def set_4_plot_linestyles(g_plot):
|
""" set the linestyles used for 4 plots in 1 graphs. """
|
|
g_plot('set style line 1 linetype 1 linecolor rgb "green" pointtype -1')
|
g_plot('set style line 2 linetype 1 linecolor rgb "red" pointtype -1')
|
g_plot('set style line 3 linetype 1 linecolor rgb "purple" pointtype -1')
|
g_plot('set style line 4 linetype 1 linecolor rgb "blue" pointtype -1')
|
|
def store_csv(cpu_int, time_pre_dec, time_post_dec, core_busy, scaled, _from, _to, mperf, aperf, tsc, freq_ghz, io_boost, common_comm, load, duration_ms, sample_num, elapsed_time, tsc_ghz):
|
""" Store master csv file information """
|
|
global graph_data_present
|
|
if cpu_mask[cpu_int] == 0:
|
return
|
|
try:
|
f_handle = open('cpu.csv', 'a')
|
string_buffer = "CPU_%03u, %05u, %06u, %u, %u, %u, %u, %u, %u, %u, %.4f, %u, %.2f, %.3f, %u, %.3f, %.3f, %s\n" % (cpu_int, int(time_pre_dec), int(time_post_dec), int(core_busy), int(scaled), int(_from), int(_to), int(mperf), int(aperf), int(tsc), freq_ghz, int(io_boost), load, duration_ms, sample_num, elapsed_time, tsc_ghz, common_comm)
|
f_handle.write(string_buffer);
|
f_handle.close()
|
except:
|
print('IO error cpu.csv')
|
return
|
|
graph_data_present = True;
|
|
def split_csv():
|
""" seperate the all csv file into per CPU csv files. """
|
|
global current_max_cpu
|
|
if os.path.exists('cpu.csv'):
|
for index in range(0, current_max_cpu + 1):
|
if cpu_mask[int(index)] != 0:
|
os.system('grep -m 1 common_cpu cpu.csv > cpu{:0>3}.csv'.format(index))
|
os.system('grep CPU_{:0>3} cpu.csv >> cpu{:0>3}.csv'.format(index, index))
|
|
def fix_ownership(path):
|
"""Change the owner of the file to SUDO_UID, if required"""
|
|
uid = os.environ.get('SUDO_UID')
|
gid = os.environ.get('SUDO_GID')
|
if uid is not None:
|
os.chown(path, int(uid), int(gid))
|
|
def cleanup_data_files():
|
""" clean up existing data files """
|
|
if os.path.exists('cpu.csv'):
|
os.remove('cpu.csv')
|
f_handle = open('cpu.csv', 'a')
|
f_handle.write('common_cpu, common_secs, common_usecs, core_busy, scaled_busy, from, to, mperf, aperf, tsc, freq, boost, load, duration_ms, sample_num, elapsed_time, tsc_ghz, common_comm')
|
f_handle.write('\n')
|
f_handle.close()
|
|
def clear_trace_file():
|
""" Clear trace file """
|
|
try:
|
f_handle = open('/sys/kernel/debug/tracing/trace', 'w')
|
f_handle.close()
|
except:
|
print('IO error clearing trace file ')
|
sys.exit(2)
|
|
def enable_trace():
|
""" Enable trace """
|
|
try:
|
open('/sys/kernel/debug/tracing/events/power/pstate_sample/enable'
|
, 'w').write("1")
|
except:
|
print('IO error enabling trace ')
|
sys.exit(2)
|
|
def disable_trace():
|
""" Disable trace """
|
|
try:
|
open('/sys/kernel/debug/tracing/events/power/pstate_sample/enable'
|
, 'w').write("0")
|
except:
|
print('IO error disabling trace ')
|
sys.exit(2)
|
|
def set_trace_buffer_size():
|
""" Set trace buffer size """
|
|
try:
|
with open('/sys/kernel/debug/tracing/buffer_size_kb', 'w') as fp:
|
fp.write(memory)
|
except:
|
print('IO error setting trace buffer size ')
|
sys.exit(2)
|
|
def free_trace_buffer():
|
""" Free the trace buffer memory """
|
|
try:
|
open('/sys/kernel/debug/tracing/buffer_size_kb'
|
, 'w').write("1")
|
except:
|
print('IO error freeing trace buffer ')
|
sys.exit(2)
|
|
def read_trace_data(filename):
|
""" Read and parse trace data """
|
|
global current_max_cpu
|
global sample_num, last_sec_cpu, last_usec_cpu, start_time
|
|
try:
|
data = open(filename, 'r').read()
|
except:
|
print('Error opening ', filename)
|
sys.exit(2)
|
|
for line in data.splitlines():
|
search_obj = \
|
re.search(r'(^(.*?)\[)((\d+)[^\]])(.*?)(\d+)([.])(\d+)(.*?core_busy=)(\d+)(.*?scaled=)(\d+)(.*?from=)(\d+)(.*?to=)(\d+)(.*?mperf=)(\d+)(.*?aperf=)(\d+)(.*?tsc=)(\d+)(.*?freq=)(\d+)'
|
, line)
|
|
if search_obj:
|
cpu = search_obj.group(3)
|
cpu_int = int(cpu)
|
cpu = str(cpu_int)
|
|
time_pre_dec = search_obj.group(6)
|
time_post_dec = search_obj.group(8)
|
core_busy = search_obj.group(10)
|
scaled = search_obj.group(12)
|
_from = search_obj.group(14)
|
_to = search_obj.group(16)
|
mperf = search_obj.group(18)
|
aperf = search_obj.group(20)
|
tsc = search_obj.group(22)
|
freq = search_obj.group(24)
|
common_comm = search_obj.group(2).replace(' ', '')
|
|
# Not all kernel versions have io_boost field
|
io_boost = '0'
|
search_obj = re.search(r'.*?io_boost=(\d+)', line)
|
if search_obj:
|
io_boost = search_obj.group(1)
|
|
if sample_num == 0 :
|
start_time = Decimal(time_pre_dec) + Decimal(time_post_dec) / Decimal(1000000)
|
sample_num += 1
|
|
if last_sec_cpu[cpu_int] == 0 :
|
last_sec_cpu[cpu_int] = time_pre_dec
|
last_usec_cpu[cpu_int] = time_post_dec
|
else :
|
duration_us = (int(time_pre_dec) - int(last_sec_cpu[cpu_int])) * 1000000 + (int(time_post_dec) - int(last_usec_cpu[cpu_int]))
|
duration_ms = Decimal(duration_us) / Decimal(1000)
|
last_sec_cpu[cpu_int] = time_pre_dec
|
last_usec_cpu[cpu_int] = time_post_dec
|
elapsed_time = Decimal(time_pre_dec) + Decimal(time_post_dec) / Decimal(1000000) - start_time
|
load = Decimal(int(mperf)*100)/ Decimal(tsc)
|
freq_ghz = Decimal(freq)/Decimal(1000000)
|
# Sanity check calculation, typically anomalies indicate missed samples
|
# However, check for 0 (should never occur)
|
tsc_ghz = Decimal(0)
|
if duration_ms != Decimal(0) :
|
tsc_ghz = Decimal(tsc)/duration_ms/Decimal(1000000)
|
store_csv(cpu_int, time_pre_dec, time_post_dec, core_busy, scaled, _from, _to, mperf, aperf, tsc, freq_ghz, io_boost, common_comm, load, duration_ms, sample_num, elapsed_time, tsc_ghz)
|
|
if cpu_int > current_max_cpu:
|
current_max_cpu = cpu_int
|
# End of for each trace line loop
|
# Now seperate the main overall csv file into per CPU csv files.
|
split_csv()
|
|
def signal_handler(signal, frame):
|
print(' SIGINT: Forcing cleanup before exit.')
|
if interval:
|
disable_trace()
|
clear_trace_file()
|
# Free the memory
|
free_trace_buffer()
|
sys.exit(0)
|
|
signal.signal(signal.SIGINT, signal_handler)
|
|
interval = ""
|
filename = ""
|
cpu_list = ""
|
testname = ""
|
memory = "10240"
|
graph_data_present = False;
|
|
valid1 = False
|
valid2 = False
|
|
cpu_mask = zeros((MAX_CPUS,), dtype=int)
|
|
try:
|
opts, args = getopt.getopt(sys.argv[1:],"ht:i:c:n:m:",["help","trace_file=","interval=","cpu=","name=","memory="])
|
except getopt.GetoptError:
|
print_help()
|
sys.exit(2)
|
for opt, arg in opts:
|
if opt == '-h':
|
print()
|
sys.exit()
|
elif opt in ("-t", "--trace_file"):
|
valid1 = True
|
location = os.path.realpath(os.path.join(os.getcwd(), os.path.dirname(__file__)))
|
filename = os.path.join(location, arg)
|
elif opt in ("-i", "--interval"):
|
valid1 = True
|
interval = arg
|
elif opt in ("-c", "--cpu"):
|
cpu_list = arg
|
elif opt in ("-n", "--name"):
|
valid2 = True
|
testname = arg
|
elif opt in ("-m", "--memory"):
|
memory = arg
|
|
if not (valid1 and valid2):
|
print_help()
|
sys.exit()
|
|
if cpu_list:
|
for p in re.split("[,]", cpu_list):
|
if int(p) < MAX_CPUS :
|
cpu_mask[int(p)] = 1
|
else:
|
for i in range (0, MAX_CPUS):
|
cpu_mask[i] = 1
|
|
if not os.path.exists('results'):
|
os.mkdir('results')
|
# The regular user needs to own the directory, not root.
|
fix_ownership('results')
|
|
os.chdir('results')
|
if os.path.exists(testname):
|
print('The test name directory already exists. Please provide a unique test name. Test re-run not supported, yet.')
|
sys.exit()
|
os.mkdir(testname)
|
# The regular user needs to own the directory, not root.
|
fix_ownership(testname)
|
os.chdir(testname)
|
|
# Temporary (or perhaps not)
|
cur_version = sys.version_info
|
print('python version (should be >= 2.7):')
|
print(cur_version)
|
|
# Left as "cleanup" for potential future re-run ability.
|
cleanup_data_files()
|
|
if interval:
|
filename = "/sys/kernel/debug/tracing/trace"
|
clear_trace_file()
|
set_trace_buffer_size()
|
enable_trace()
|
print('Sleeping for ', interval, 'seconds')
|
time.sleep(int(interval))
|
disable_trace()
|
|
current_max_cpu = 0
|
|
read_trace_data(filename)
|
|
if interval:
|
clear_trace_file()
|
# Free the memory
|
free_trace_buffer()
|
|
if graph_data_present == False:
|
print('No valid data to plot')
|
sys.exit(2)
|
|
for cpu_no in range(0, current_max_cpu + 1):
|
plot_perf_busy_with_sample(cpu_no)
|
plot_perf_busy(cpu_no)
|
plot_durations(cpu_no)
|
plot_loads(cpu_no)
|
|
plot_pstate_cpu_with_sample()
|
plot_pstate_cpu()
|
plot_load_cpu()
|
plot_frequency_cpu()
|
plot_duration_cpu()
|
plot_scaled_cpu()
|
plot_boost_cpu()
|
plot_ghz_cpu()
|
|
# It is preferrable, but not necessary, that the regular user owns the files, not root.
|
for root, dirs, files in os.walk('.'):
|
for f in files:
|
fix_ownership(f)
|
|
os.chdir('../../')
|
