From 05c3d0b361661555d02b21f20b2ee4b0a89349dd Mon Sep 17 00:00:00 2001 From: Attia Radwan Date: Fri, 26 Aug 2022 11:17:41 +0200 Subject: [PATCH] fix ci job --- .circleci/config.yml | 2 +- benchmarks/benchmark_layer.py | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/.circleci/config.yml b/.circleci/config.yml index 25de9ca15..e56776053 100644 --- a/.circleci/config.yml +++ b/.circleci/config.yml @@ -261,7 +261,7 @@ commands: echo "Using $(python -V) ($(which python))" echo "Using $(pip -V) ($(which pip))" python benchmarks/run_benchmarks.py --batch_size 16 --layers <> --config_file ./benchmarks/config.json --root ./benchmarks/results/raw/ --cont - IFS=$' ';layers=(<>); mkdir -p /tmp/report_layers; IFS=$'\n'; files=`( echo "${layers[*]}" ) | sed 's/.*/.\/benchmarks\/results\/raw\/&*/'` + IFS=$' ';layers=(<>); rm -rf /tmp/report_layers; mkdir -p /tmp/report_layers; IFS=$'\n'; files=`( echo "${layers[*]}" ) | sed 's/.*/.\/benchmarks\/results\/raw\/&*/'` cp -v ${files[@]} /tmp/report_layers report_id=`IFS=$'-'; echo "${layers[*]}"` python benchmarks/generate_report.py --path-to-results /tmp/report_layers --save-path benchmarks/results/report-${report_id}.csv --format csv diff --git a/benchmarks/benchmark_layer.py b/benchmarks/benchmark_layer.py index 3b3ecd791..05c530f62 100644 --- a/benchmarks/benchmark_layer.py +++ b/benchmarks/benchmark_layer.py @@ -18,6 +18,7 @@ import torch import torch.utils.benchmark as benchmark + from layers import LayerFactory, LayerType from utils import get_layer_set, reset_peak_memory_stats