fix ci job

pytorch · Aug 26, 2022 · 05c3d0b · 05c3d0b
1 parent c628f9d
commit 05c3d0b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git 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 <<parameters.layers>> --config_file ./benchmarks/config.json --root ./benchmarks/results/raw/ --cont
-            IFS=$' ';layers=(<<parameters.layers>>); mkdir -p /tmp/report_layers; IFS=$'\n'; files=`( echo "${layers[*]}" ) | sed 's/.*/.\/benchmarks\/results\/raw\/&*/'`
+            IFS=$' ';layers=(<<parameters.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
@@ -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