2525#include " GRU_d10_L20_h8_B1.hxx"
2626#include " LSTM_d10_L20_h8_B1.hxx"
2727#include " higgs_model_dense.hxx"
28- #include " DDB_B1.hxx" // CMS onnx model
29- #include " Conv2DTranspose_Relu_Sigmoid.hxx"
28+ // #include "DDB_B1.hxx" // CMS onnx model
29+ // #include "Conv2DTranspose_Relu_Sigmoid.hxx"
3030#include " ConvTrans2dModel_B1.hxx"
3131// #include "ConvTransposeM.hxx"
3232#include " ConvTModel_G4.hxx"
@@ -60,7 +60,7 @@ void BM_SOFIE_Inference(benchmark::State &state)
6060 std::generate (input.begin (), input.end (), []() { return distribution (generator); });
6161 }
6262 float *input_ptr = input.data ();
63- S s ( " " ) ;
63+ S s;
6464
6565 double totDuration = 0 ;
6666 int ntimes = 0 ;
@@ -143,7 +143,7 @@ void BM_SOFIE_Inference_3(benchmark::State &state)
143143 input3 = vector<float >(input3.size (),3 .);
144144 }
145145
146- S s ( " " ) ;
146+ S s;
147147
148148 // std::cout << "init done - do benchmark \n";
149149
@@ -169,7 +169,7 @@ void BM_SOFIE_Inference_3(benchmark::State &state)
169169// CMS benchmark (3 inputs)
170170// 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);
171171// Conv Transpose
172- BENCHMARK_TEMPLATE (BM_SOFIE_Inference, TMVA_SOFIE_Conv2DTranspose_Relu_Sigmoid::Session)->Name(" Conv2DTranspose_Relu_Sigmoid" )->Args({15 ,1 })->Unit(benchmark::kMillisecond );
172+ // BENCHMARK_TEMPLATE(BM_SOFIE_Inference, TMVA_SOFIE_Conv2DTranspose_Relu_Sigmoid::Session)->Name("Conv2DTranspose_Relu_Sigmoid")->Args({15,1})->Unit(benchmark::kMillisecond);
173173BENCHMARK_TEMPLATE (BM_SOFIE_Inference, TMVA_SOFIE_ConvTModel_G4::Session)->Name(" ConvTModel_G4" )->Args({15 ,1 })->Unit(benchmark::kMillisecond );
174174// BENCHMARK_TEMPLATE(BM_SOFIE_Inference, TMVA_SOFIE_ConvTransposeM::Session)->Name("ConvTransposeM")->Args({4*30*30,4})->Unit(benchmark::kMillisecond);
175175BENCHMARK_TEMPLATE (BM_SOFIE_Inference, TMVA_SOFIE_ConvTrans2dModel_B1::Session)->Name(" ConvTrans2dModel_B1" )->Args({4 *4 *4 ,1 })->Unit(benchmark::kMillisecond );
0 commit comments