add ops
This commit is contained in:
15
torch_mlu_ops-v1.3.2/benchmarks/benchmark_test.sh
Normal file
15
torch_mlu_ops-v1.3.2/benchmarks/benchmark_test.sh
Normal file
@@ -0,0 +1,15 @@
|
||||
#!/bin/bash
|
||||
LOG_PATH=${LOG_PATH:-.}
|
||||
|
||||
files=($(ls benchmark_*.py))
|
||||
for file in "${files[@]}"; do
|
||||
echo "test ${file}..."
|
||||
op_name=$(basename "$file" .py)
|
||||
python "$file" > ${LOG_PATH}/${op_name}.log 2>&1
|
||||
ret_tmp=$?
|
||||
cat ${LOG_PATH}/${op_name}.log
|
||||
if [ $ret_tmp != 0 ]; then
|
||||
echo "${sc} test failed..."
|
||||
exit $ret_tmp
|
||||
fi
|
||||
done
|
||||
Reference in New Issue
Block a user