Skip to content

Commit 9cb47ca

Browse files
committed
add conv2DTranspose model
1 parent 5457d8f commit 9cb47ca

File tree

2 files changed

+4
-1
lines changed

2 files changed

+4
-1
lines changed

root/tmva/sofie/SOFIEInference.cxx

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,7 @@
2626
#include "LSTM_d10_L20_h8_B1.hxx"
2727
#include "higgs_model_dense.hxx"
2828
#include "DDB_B1.hxx" // CMS onnx model
29+
#include "Conv2DTranspose_Relu_Sigmoid.hxx"
2930

3031
#include "resnet18v1.hxx"
3132
#include "TMath.h"
@@ -102,7 +103,7 @@ void BM_SOFIE_Inference_3(benchmark::State &state)
102103

103104
S s("");
104105

105-
std::cout << "initi done - do benchmark \n";
106+
//std::cout << "init done - do benchmark \n";
106107

107108
double totDuration = 0;
108109
int ntimes = 0;
@@ -125,6 +126,8 @@ void BM_SOFIE_Inference_3(benchmark::State &state)
125126

126127
// CMS benchmark (3 inputs)
127128
BENCHMARK_TEMPLATE(BM_SOFIE_Inference_3, TMVA_SOFIE_DDB_B1::Session)->Name("DDB_B1")->Args({1, 1*27, 60*8, 5*2})->Unit(benchmark::kMillisecond);
129+
// Conv Transpose
130+
BENCHMARK_TEMPLATE(BM_SOFIE_Inference, TMVA_SOFIE_Conv2DTranspose_Relu_Sigmoid::Session)->Name("Cov2DTranspose_B1")->Args({1, 1*15})->Unit(benchmark::kMillisecond);
128131

129132
//Gemm benchmarks
130133
BENCHMARK_TEMPLATE(BM_SOFIE_Inference, TMVA_SOFIE_Linear_16::Session)->Name("Linear_16")->Args({100, 16})->Unit(benchmark::kMillisecond);
2.28 MB
Binary file not shown.

0 commit comments

Comments
 (0)