Skip to content

Commit 3da94c4

Browse files
committed
change for modifications in Sofie session constructor generated code
FOr the default weights one needs to do Session s instead of s("")
1 parent 029d456 commit 3da94c4

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

root/tmva/sofie/SOFIEInference.cxx

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -25,8 +25,8 @@
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);
173173
BENCHMARK_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);
175175
BENCHMARK_TEMPLATE(BM_SOFIE_Inference, TMVA_SOFIE_ConvTrans2dModel_B1::Session)->Name("ConvTrans2dModel_B1")->Args({4*4*4,1})->Unit(benchmark::kMillisecond);

0 commit comments

Comments
 (0)