forked from rrcarlosr/Jetpack
124 lines
4.2 KiB
Python
Executable File
124 lines
4.2 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
#
|
|
# Copyright (c) 2017, NVIDIA CORPORATION. All rights reserved.
|
|
#
|
|
# This program is free software; you can redistribute it and/or modify it
|
|
# under the terms and conditions of the GNU General Public License,
|
|
# version 2, as published by the Free Software Foundation.
|
|
#
|
|
# This program is distributed in the hope it will be useful, but WITHOUT
|
|
# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
|
|
# FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
|
|
# more details.
|
|
#
|
|
# usage: nvgpu_debug_hal.py [-h] [--csv] [--gk20a GK20A] [gops_filename]
|
|
#
|
|
# Analyze the HAL debugfs interface's output. With no arguments, prints out
|
|
# statistics on the gpu_ops struct based on analysis of gk20a.h
|
|
#
|
|
# positional arguments:
|
|
# gops_filename debugfs interface output file (from /d/gpu.0/hal/gops)
|
|
#
|
|
# optional arguments:
|
|
# -h, --help show this help message and exit
|
|
# --csv csv formatted output
|
|
# --gk20a GK20A path to gk20a.h
|
|
|
|
import argparse
|
|
import re
|
|
from os import environ
|
|
|
|
description_str = ('Analyze the HAL debugfs interface\'s output. '
|
|
'With no arguments, prints out statistics on the gpu_ops struct based on '
|
|
'analysis of gk20a.h')
|
|
|
|
parser = argparse.ArgumentParser(description=description_str);
|
|
parser.add_argument("--csv", help="csv formatted output", action="store_true");
|
|
parser.add_argument("--gk20a", help="path to gk20a.h");
|
|
parser.add_argument("gops_filename", help="debugfs interface output file (from /d/gpu.0/hal/gops)", nargs='?');
|
|
args = parser.parse_args();
|
|
|
|
if args.gk20a:
|
|
gk20a_h_path = args.gk20a
|
|
else:
|
|
top = environ.get('TOP');
|
|
if top is None:
|
|
print("$TOP is undefined, unable to find gk20a.h");
|
|
exit(-1);
|
|
gk20a_h_path = top + "/kernel/nvgpu/drivers/gpu/nvgpu/gk20a/gk20a.h"
|
|
|
|
def get_function_pointer_name(line):
|
|
matches = re.search('.*\(\*(?P<function_name>\w+)\)\(', line);
|
|
if matches is None:
|
|
return None
|
|
else:
|
|
return matches.group("function_name");
|
|
|
|
# Build the list of gpu_ops member function pointers from gk20a.h
|
|
non_function_pointer_members = [];
|
|
formatted_members = [];
|
|
gops_members = dict();
|
|
substruct_names = [];
|
|
lone_members = [];
|
|
with open(gk20a_h_path) as gk20a_h:
|
|
# Skip to start of gpu_ops struct
|
|
while gk20a_h.readline() != "struct gpu_ops {\n":
|
|
continue;
|
|
|
|
line = gk20a_h.readline();
|
|
while line != "};\n":
|
|
# If this is a substruct
|
|
if re.match('\t+struct.+\{', line):
|
|
# Read the contents of the substruct
|
|
line = gk20a_h.readline();
|
|
struct_contents = ""
|
|
while not re.match("\t*\} (\w+);", line):
|
|
struct_contents += line;
|
|
line = gk20a_h.readline();
|
|
# Split out the substruct name and the function pointer names
|
|
struct_name = re.match("\t*\} (?P<struct_name>\w+);", line).group("struct_name");
|
|
struct_members = re.findall(r".+?\(\s*\*\s*(\w+)\s*\).+?;", struct_contents, flags=re.DOTALL)
|
|
|
|
# Store the substruct as an entry
|
|
substruct_names.append(struct_name);
|
|
gops_members[struct_name] = struct_members;
|
|
# Format members
|
|
for member in struct_members:
|
|
formatted_members.append(struct_name + "." + member);
|
|
else:
|
|
# Lone members (function pointers or stuff not in a substruct)
|
|
match = re.match(".*\(\*(?P<function_name>\w+)\)\(", line);
|
|
if match is not None:
|
|
# It's a function pointer, keep track of it
|
|
lone_members.append(match.group("function_name"));
|
|
formatted_members.append(match.group("function_name"));
|
|
else:
|
|
# Not a function pointer, may also catch comments etc.
|
|
non_function_pointer_members.append(line.strip());
|
|
line = gk20a_h.readline();
|
|
if args.gops_filename:
|
|
# Interpret gops file
|
|
with open(args.gops_filename) as gops:
|
|
i = 0;
|
|
# Option for csv output
|
|
if args.csv:
|
|
format_string = '{0},{1}';
|
|
else:
|
|
format_string = '{0:<60} = {1}';
|
|
for line in gops:
|
|
print(format_string.format(formatted_members[i], line[:-1]));
|
|
i += 1;
|
|
else:
|
|
# Just print some stats on the gpu_ops struct
|
|
total = 0;
|
|
print("----- Lone Function Pointers -----");
|
|
print("Count =", len(lone_members));
|
|
total += len(lone_members);
|
|
for line in lone_members:
|
|
print(line);
|
|
print("----- Substruct Counts -----");
|
|
for name in substruct_names:
|
|
print(name, "=", len(gops_members[name]));
|
|
total += len(gops_members[name])
|
|
print("\n Total =", total);
|