mirror of
https://github.com/nerfstudio-project/nerfstudio.git
synced 2026-11-10 18:17:31 +08:00
52 lines
1.4 KiB
Bash
Executable File
52 lines
1.4 KiB
Bash
Executable File
#!/bin/bash
|
|
|
|
helpFunction_flame()
|
|
{
|
|
echo "Usage: $0 -o output_file -p program"
|
|
echo -e "\t-o output .svg filename, where to save flame graph"
|
|
echo -e "\t-p .py program we want to profile with associated arguments"
|
|
exit 1 # Exit program after printing help
|
|
}
|
|
|
|
helpFunction_top()
|
|
{
|
|
echo "Usage: $0 -p program"
|
|
echo -e "\t-p .py program we want to profile with associated arguments"
|
|
exit 1 # Exit program after printing help
|
|
}
|
|
|
|
|
|
while getopts "t:o:p" opt; do
|
|
case "$opt" in
|
|
t ) type="$OPTARG" ;;
|
|
o ) output_file="$OPTARG" ;;
|
|
p ) program="$OPTARG" ;;
|
|
? ) helpFunction ;;
|
|
esac
|
|
done
|
|
|
|
if [ -z "$type" ]; then
|
|
echo "Missing profiling type (flame / top)"
|
|
fi
|
|
shift $((OPTIND-1))
|
|
program="$@"
|
|
program=${program/"--"/"-- --"}
|
|
|
|
# Print helpFunction in case parameters are empty
|
|
if [[ "$type" = "flame" ]]; then
|
|
if [ -z "$output_file" ] || [ -z "$program" ]; then
|
|
echo "Some or all of the parameters are empty";
|
|
helpFunction_flame
|
|
fi
|
|
command=`echo py-spy record -o "$output_file" python "$program"`
|
|
echo "Running..." $command
|
|
eval "$command"
|
|
elif [[ "$type" = "top" ]]; then
|
|
if [ -z "$program" ]; then
|
|
echo "Some or all of the parameters are empty";
|
|
helpFunction_top
|
|
fi
|
|
command=`echo py-spy top python "$program"`
|
|
echo "Running..." $command
|
|
eval "$command"
|
|
fi |