add resnet to benchmark

patflick · Jul 20, 2017 · 70abeae · 70abeae
1 parent ee24338
commit 70abeae
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/Makefile b/Makefile
@@ -12,7 +12,7 @@ all: alexnet resnet benchmark_wino layerwise gputop
 HEADERS=function.hpp layers.hpp miopen.hpp multi_layers.hpp tensor.hpp utils.hpp
 
 benchmark: all
- ./benchmark_wino | tee wino.log && ./layerwise | tee layerwise.log && ./alexnet | tee alexnet.log
+ ./benchmark_wino | tee wino.log && ./layerwise | tee layerwise.log && ./alexnet | tee alexnet.log && ./resnet | tee resnet50.log
 
 alexnet: alexnet.cpp $(HEADERS)
  $(HIPCC) $(HIPCC_FLAGS) alexnet.cpp $(LD_FLAGS) -o $@

diff --git a/benchmark_wino.cpp b/benchmark_wino.cpp
@@ -18,7 +18,7 @@ int main(int argc, char *argv[])
 
  // benchmark fwd
  BenchmarkLogger::new_session("wino_conv");
- BenchmarkLogger::fwd_layer_benchmark(m, 1000);
+ BenchmarkLogger::fwd_layer_benchmark(m, 5000);
 
  return 0;
 }