mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
Pull Request resolved: https://github.com/pytorch/pytorch/pull/146282 Approved by: https://github.com/shunting314 ghstack dependencies: #146252, #146254, #146255, #146257
36 lines
836 B
Bash
Executable File
36 lines
836 B
Bash
Executable File
#!/bin/bash
|
|
# Check if the output file argument was provided
|
|
if [ $# -eq 0 ]
|
|
then
|
|
echo "Please provide the output file as an argument"
|
|
return
|
|
fi
|
|
|
|
# Check if the directory of Python programs argument was provided
|
|
if [ $# -eq 1 ]
|
|
then
|
|
echo "Please provide the directory of Python programs as an argument"
|
|
return
|
|
fi
|
|
|
|
# Set the output file
|
|
output_file=$1
|
|
# Set the directory of Python programs
|
|
python_programs_dir=$2
|
|
# Loop through all files in the directory of Python programs
|
|
|
|
start=`date +%s`
|
|
|
|
for file in $python_programs_dir/*.py
|
|
do
|
|
if [ "$file" != "$python_programs_dir/benchmark_base.py" ]; then
|
|
python $file $output_file
|
|
fi
|
|
done
|
|
end=`date +%s`
|
|
|
|
runtime=$((end-start))
|
|
echo "total time to run benchmarks is:"
|
|
echo $runtime
|
|
python benchmarks/dynamo/pr_time_benchmarks/log_benchmarking_time.py $runtime
|