@@ -540,7 +540,7 @@ class pidTPCModule
540540 constexpr int ExpectedInputDimensionsNNV3 = 8 ;
541541 constexpr auto NetworkVersionV2 = " 2" ;
542542 constexpr auto NetworkVersionV3 = " 3" ;
543- for (int i = 0 ; i < NParticleTypes; i ++) { // Loop over particle number for which network correction is used
543+ for (int j = 0 ; j < NParticleTypes; j ++) { // Loop over particle number for which network correction is used
544544 for (auto const & trk : tracks) {
545545 if (!trk.hasTPC ()) {
546546 continue ;
@@ -553,7 +553,7 @@ class pidTPCModule
553553 track_properties[counter_track_props] = trk.tpcInnerParam ();
554554 track_properties[counter_track_props + 1 ] = trk.tgl ();
555555 track_properties[counter_track_props + 2 ] = trk.signed1Pt ();
556- track_properties[counter_track_props + 3 ] = o2::track::pid_constants::sMasses [i ];
556+ track_properties[counter_track_props + 3 ] = o2::track::pid_constants::sMasses [j ];
557557 track_properties[counter_track_props + 4 ] = trk.has_collision () ? mults[trk.collisionId ()] / 11000 . : 1 .;
558558 track_properties[counter_track_props + 5 ] = std::sqrt (nNclNormalization / trk.tpcNClsFound ());
559559 if (input_dimensions == ExpectedInputDimensionsNNV2 && networkVersion == NetworkVersionV2) {
@@ -583,9 +583,9 @@ class pidTPCModule
583583 float * output_network = network.evalModel (track_properties);
584584 auto stop_network_eval = std::chrono::high_resolution_clock::now ();
585585 duration_network += std::chrono::duration<float , std::ratio<1 , 1000000000 >>(stop_network_eval - start_network_eval).count ();
586- for (uint64_t i = 0 ; i < prediction_size; i += output_dimensions) {
587- for (int j = 0 ; j < output_dimensions; j ++) {
588- network_prediction[i + j + prediction_size * loop_counter] = output_network[i + j ];
586+ for (uint64_t k = 0 ; k < prediction_size; k += output_dimensions) {
587+ for (int l = 0 ; l < output_dimensions; l ++) {
588+ network_prediction[k + l + prediction_size * loop_counter] = output_network[k + l ];
589589 }
590590 }
591591
0 commit comments