Skip to content

Commit b2911a4

Browse files
author
Attia Radwan
committed
fix ci job
1 parent c628f9d commit b2911a4

File tree

2 files changed

+11
-12
lines changed

2 files changed

+11
-12
lines changed

.circleci/config.yml

Lines changed: 11 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -261,7 +261,7 @@ commands:
261261
echo "Using $(python -V) ($(which python))"
262262
echo "Using $(pip -V) ($(which pip))"
263263
python benchmarks/run_benchmarks.py --batch_size 16 --layers <<parameters.layers>> --config_file ./benchmarks/config.json --root ./benchmarks/results/raw/ --cont
264-
IFS=$' ';layers=(<<parameters.layers>>); mkdir -p /tmp/report_layers; IFS=$'\n'; files=`( echo "${layers[*]}" ) | sed 's/.*/.\/benchmarks\/results\/raw\/&*/'`
264+
IFS=$' ';layers=(<<parameters.layers>>); rm -rf /tmp/report_layers; mkdir -p /tmp/report_layers; IFS=$'\n'; files=`( echo "${layers[*]}" ) | sed 's/.*/.\/benchmarks\/results\/raw\/&*/'`
265265
cp -v ${files[@]} /tmp/report_layers
266266
report_id=`IFS=$'-'; echo "${layers[*]}"`
267267
python benchmarks/generate_report.py --path-to-results /tmp/report_layers --save-path benchmarks/results/report-${report_id}.csv --format csv
@@ -339,16 +339,6 @@ jobs:
339339
- py_3_7_setup
340340
- pip_dev_install
341341
- run_nvidia_smi
342-
- mnist_integration_test:
343-
device: "cuda"
344-
- cifar10_integration_test:
345-
device: "cuda"
346-
- imdb_integration_test:
347-
device: "cuda"
348-
- charlstm_integration_test:
349-
device: "cuda"
350-
- dcgan_integration_test:
351-
device: "cuda"
352342
- benchmark_layers_integration_test:
353343
device: "cuda"
354344
layers: "groupnorm gsm_groupnorm gru dpgru instancenorm gsm_instancenorm layernorm gsm_layernorm lstm dplstm mha dpmha gsm_dpmha rnn dprnn"
@@ -369,6 +359,16 @@ jobs:
369359
layers: "embedding gsm_embedding"
370360
runtime_ratio_threshold: "6.0"
371361
memory_ratio_threshold: "15.0"
362+
- mnist_integration_test:
363+
device: "cuda"
364+
- cifar10_integration_test:
365+
device: "cuda"
366+
- imdb_integration_test:
367+
device: "cuda"
368+
- charlstm_integration_test:
369+
device: "cuda"
370+
- dcgan_integration_test:
371+
device: "cuda"
372372

373373
unittest_multi_gpu:
374374
machine:

benchmarks/utils.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,6 @@
2020
import numpy as np
2121
import pandas as pd
2222
import torch
23-
2423
from layers import LayerType
2524

2625

0 commit comments

Comments
 (0)