Skip to content

Commit

Permalink
fix ci job
Browse files Browse the repository at this point in the history
  • Loading branch information
Attia Radwan committed Aug 26, 2022
1 parent c628f9d commit b2911a4
Show file tree
Hide file tree
Showing 2 changed files with 11 additions and 12 deletions.
22 changes: 11 additions & 11 deletions .circleci/config.yml
Expand Up @@ -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
Expand Down Expand Up @@ -339,16 +339,6 @@ jobs:
- py_3_7_setup
- pip_dev_install
- run_nvidia_smi
- mnist_integration_test:
device: "cuda"
- cifar10_integration_test:
device: "cuda"
- imdb_integration_test:
device: "cuda"
- charlstm_integration_test:
device: "cuda"
- dcgan_integration_test:
device: "cuda"
- benchmark_layers_integration_test:
device: "cuda"
layers: "groupnorm gsm_groupnorm gru dpgru instancenorm gsm_instancenorm layernorm gsm_layernorm lstm dplstm mha dpmha gsm_dpmha rnn dprnn"
Expand All @@ -369,6 +359,16 @@ jobs:
layers: "embedding gsm_embedding"
runtime_ratio_threshold: "6.0"
memory_ratio_threshold: "15.0"
- mnist_integration_test:
device: "cuda"
- cifar10_integration_test:
device: "cuda"
- imdb_integration_test:
device: "cuda"
- charlstm_integration_test:
device: "cuda"
- dcgan_integration_test:
device: "cuda"

unittest_multi_gpu:
machine:
Expand Down
1 change: 0 additions & 1 deletion benchmarks/utils.py
Expand Up @@ -20,7 +20,6 @@
import numpy as np
import pandas as pd
import torch

from layers import LayerType


Expand Down

0 comments on commit b2911a4

Please sign in to comment.