module cli.profiling
#
Short summary#
module onnxcustom.cli.profiling
Command lines for profiling.
Functions#
function |
truncated documentation |
---|---|
Converts traces produced by nvprof and saved with format sqlite3 (extension .sql). |
Documentation#
Command lines for profiling.
- onnxcustom.cli.profiling.nvprof2json(filename, output='', temporary_file='', verbose=1, fLOG=<built-in function print>)#
Converts traces produced by nvprof and saved with format sqlite3 (extension .sql).
- Parameters:
filename – filename
output – output file, if left empty, the result is printed on the standard output
temporary_file – if the file needs to be unzipped, this file will be created to be the unzipped file, it is not cleaned after the unzipping.
verbose – verbosity
fLOG – logging function
- Returns:
json (if output is None, the list of events otherwise)
Converts a profile stored by nvprof into json
The sqlite dump is generated with a command line similar to:
nvprof -o gpu_profile.sql python plot_gpu_training.py
The command produces a json file following the Trace Event Format.
<<<
python -m onnxcustom nvprof2json --help
>>>
usage: nvprof2json [-h] [-f FILENAME] [-o OUTPUT] [-t TEMPORARY_FILE] [-v VERBOSE] Converts traces produced by `nvprof` and saved with format `sqlite3` (extension `.sql`). optional arguments: -h, --help show this help message and exit -f FILENAME, --filename FILENAME filename (default: None) -o OUTPUT, --output OUTPUT output file, if left empty, the result is printed on the standard output (default: ) -t TEMPORARY_FILE, --temporary_file TEMPORARY_FILE if the file needs to be unzipped, this file will be created to be the unzipped file, it is not cleaned after the unzipping. (default: ) -v VERBOSE, --verbose VERBOSE verbosity (default: 1)