mirror of
https://github.com/nerfstudio-project/nerfstudio.git
synced 2026-11-11 18:44:21 +08:00
fixed profiler bug with variable options (#254)
This commit is contained in:
1 parent
a95f47cd08
commit
6281896e1d
1 file changed
+27
-18
@@ -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
|
||||
Reference in new issue
Block a user