tink/nerfstudio
4
0
Fork 0
mirror of https://github.com/nerfstudio-project/nerfstudio.git synced 2026-11-11 10:34:29 +08:00
Code

fixed profiler bug with variable options (#254)

Browse Source
This commit is contained in:
evonneng 2022-08-08 09:53:12 -07:00 committed by GitHub
parent a95f47cd08
commit 6281896e1d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 27 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

45
scripts/debugging/profile.sh
View File

@ -16,28 +16,37 @@ helpFunction_top()
}
while getopts "t:o:p:a:" opt; do
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)"
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
# 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
py-spy record -o "$output_file" python "$program"
elif [[ "$type" = "top" ]]; then
if [ -z "$program" ]; then
echo "Some or all of the parameters are empty";
helpFunction_top
fi
py-spy top python "$program"
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