@@ -393,20 +393,20 @@ GPUdii() void GPUTPCNNClusterizerKernels::Thread<GPUTPCNNClusterizerKernels::pub
393393
394394 if (dtype == 0 ) {
395395 pc.setFull (central_charge * clustererNN.mOutputDataReg1_16 [model_output_index + 4 ].ToFloat (),
396- static_cast <float >(peak.pad ()) + clustererNN.mOutputDataReg1_16 [model_output_index].ToFloat (),
397- notSinglePad ? clustererNN.mOutputDataReg1_16 [model_output_index + 2 ].ToFloat () : 0 .f ,
398- (clusterer.mPmemory ->fragment ).start + static_cast <float >(peak.time ()) + clustererNN.mOutputDataReg1_16 [model_output_index + 1 ].ToFloat (),
399- notSingleTime ? clustererNN.mOutputDataReg1_16 [model_output_index + 3 ].ToFloat () : 0 .f ,
400- clustererNN.mClusterFlags [2 * glo_idx],
401- clustererNN.mClusterFlags [2 * glo_idx + 1 ]);
396+ static_cast <float >(peak.pad ()) + clustererNN.mOutputDataReg1_16 [model_output_index].ToFloat (),
397+ notSinglePad ? clustererNN.mOutputDataReg1_16 [model_output_index + 2 ].ToFloat () : 0 .f ,
398+ (clusterer.mPmemory ->fragment ).start + static_cast <float >(peak.time ()) + clustererNN.mOutputDataReg1_16 [model_output_index + 1 ].ToFloat (),
399+ notSingleTime ? clustererNN.mOutputDataReg1_16 [model_output_index + 3 ].ToFloat () : 0 .f ,
400+ clustererNN.mClusterFlags [2 * glo_idx],
401+ clustererNN.mClusterFlags [2 * glo_idx + 1 ]);
402402 } else if (dtype == 1 ) {
403403 pc.setFull (central_charge * clustererNN.mOutputDataReg1_32 [model_output_index + 4 ],
404- static_cast <float >(peak.pad ()) + clustererNN.mOutputDataReg1_32 [model_output_index],
405- notSinglePad ? clustererNN.mOutputDataReg1_32 [model_output_index + 2 ] : 0 .f ,
406- (clusterer.mPmemory ->fragment ).start + static_cast <float >(peak.time ()) + clustererNN.mOutputDataReg1_32 [model_output_index + 1 ],
407- notSingleTime ? clustererNN.mOutputDataReg1_32 [model_output_index + 3 ] : 0 .f ,
408- clustererNN.mClusterFlags [2 * glo_idx],
409- clustererNN.mClusterFlags [2 * glo_idx + 1 ]);
404+ static_cast <float >(peak.pad ()) + clustererNN.mOutputDataReg1_32 [model_output_index],
405+ notSinglePad ? clustererNN.mOutputDataReg1_32 [model_output_index + 2 ] : 0 .f ,
406+ (clusterer.mPmemory ->fragment ).start + static_cast <float >(peak.time ()) + clustererNN.mOutputDataReg1_32 [model_output_index + 1 ],
407+ notSingleTime ? clustererNN.mOutputDataReg1_32 [model_output_index + 3 ] : 0 .f ,
408+ clustererNN.mClusterFlags [2 * glo_idx],
409+ clustererNN.mClusterFlags [2 * glo_idx + 1 ]);
410410 }
411411
412412 tpc::ClusterNative myCluster;
@@ -501,20 +501,20 @@ GPUdii() void GPUTPCNNClusterizerKernels::Thread<GPUTPCNNClusterizerKernels::pub
501501 // Cluster 1
502502 if (dtype == 0 ) {
503503 pc.setFull (central_charge * clustererNN.mOutputDataReg2_16 [model_output_index + 8 ].ToFloat (),
504- static_cast <float >(peak.pad ()) + clustererNN.mOutputDataReg2_16 [model_output_index].ToFloat (),
505- clustererNN.mOutputDataReg2_16 [model_output_index + 4 ].ToFloat (),
506- (clusterer.mPmemory ->fragment ).start + static_cast <float >(peak.time ()) + clustererNN.mOutputDataReg2_16 [model_output_index + 2 ].ToFloat (),
507- clustererNN.mOutputDataReg2_16 [model_output_index + 6 ].ToFloat (),
508- clustererNN.mClusterFlags [2 * glo_idx],
509- clustererNN.mClusterFlags [2 * glo_idx + 1 ]);
504+ static_cast <float >(peak.pad ()) + clustererNN.mOutputDataReg2_16 [model_output_index].ToFloat (),
505+ clustererNN.mOutputDataReg2_16 [model_output_index + 4 ].ToFloat (),
506+ (clusterer.mPmemory ->fragment ).start + static_cast <float >(peak.time ()) + clustererNN.mOutputDataReg2_16 [model_output_index + 2 ].ToFloat (),
507+ clustererNN.mOutputDataReg2_16 [model_output_index + 6 ].ToFloat (),
508+ clustererNN.mClusterFlags [2 * glo_idx],
509+ clustererNN.mClusterFlags [2 * glo_idx + 1 ]);
510510 } else if (dtype == 1 ) {
511511 pc.setFull (central_charge * clustererNN.mOutputDataReg2_32 [model_output_index + 8 ],
512- static_cast <float >(peak.pad ()) + clustererNN.mOutputDataReg2_32 [model_output_index],
513- clustererNN.mOutputDataReg2_32 [model_output_index + 4 ],
514- (clusterer.mPmemory ->fragment ).start + static_cast <float >(peak.time ()) + clustererNN.mOutputDataReg2_32 [model_output_index + 2 ],
515- clustererNN.mOutputDataReg2_32 [model_output_index + 6 ],
516- clustererNN.mClusterFlags [2 * glo_idx],
517- clustererNN.mClusterFlags [2 * glo_idx + 1 ]);
512+ static_cast <float >(peak.pad ()) + clustererNN.mOutputDataReg2_32 [model_output_index],
513+ clustererNN.mOutputDataReg2_32 [model_output_index + 4 ],
514+ (clusterer.mPmemory ->fragment ).start + static_cast <float >(peak.time ()) + clustererNN.mOutputDataReg2_32 [model_output_index + 2 ],
515+ clustererNN.mOutputDataReg2_32 [model_output_index + 6 ],
516+ clustererNN.mClusterFlags [2 * glo_idx],
517+ clustererNN.mClusterFlags [2 * glo_idx + 1 ]);
518518 }
519519
520520 tpc::ClusterNative myCluster;
@@ -547,20 +547,20 @@ GPUdii() void GPUTPCNNClusterizerKernels::Thread<GPUTPCNNClusterizerKernels::pub
547547 // Cluster 2
548548 if (dtype == 0 ) {
549549 pc.setFull (central_charge * clustererNN.mOutputDataReg2_16 [model_output_index + 9 ].ToFloat (),
550- static_cast <float >(peak.pad ()) + clustererNN.mOutputDataReg2_16 [model_output_index + 1 ].ToFloat (),
551- clustererNN.mOutputDataReg2_16 [model_output_index + 5 ].ToFloat (),
552- (clusterer.mPmemory ->fragment ).start + static_cast <float >(peak.time ()) + clustererNN.mOutputDataReg2_16 [model_output_index + 3 ].ToFloat (),
553- clustererNN.mOutputDataReg2_16 [model_output_index + 7 ].ToFloat (),
554- clustererNN.mClusterFlags [2 * glo_idx],
555- clustererNN.mClusterFlags [2 * glo_idx + 1 ]);
550+ static_cast <float >(peak.pad ()) + clustererNN.mOutputDataReg2_16 [model_output_index + 1 ].ToFloat (),
551+ clustererNN.mOutputDataReg2_16 [model_output_index + 5 ].ToFloat (),
552+ (clusterer.mPmemory ->fragment ).start + static_cast <float >(peak.time ()) + clustererNN.mOutputDataReg2_16 [model_output_index + 3 ].ToFloat (),
553+ clustererNN.mOutputDataReg2_16 [model_output_index + 7 ].ToFloat (),
554+ clustererNN.mClusterFlags [2 * glo_idx],
555+ clustererNN.mClusterFlags [2 * glo_idx + 1 ]);
556556 } else if (dtype == 1 ) {
557557 pc.setFull (central_charge * clustererNN.mOutputDataReg2_32 [model_output_index + 9 ],
558- static_cast <float >(peak.pad ()) + clustererNN.mOutputDataReg2_32 [model_output_index + 1 ],
559- clustererNN.mOutputDataReg2_32 [model_output_index + 5 ],
560- (clusterer.mPmemory ->fragment ).start + static_cast <float >(peak.time ()) + clustererNN.mOutputDataReg2_32 [model_output_index + 3 ],
561- clustererNN.mOutputDataReg2_32 [model_output_index + 7 ],
562- clustererNN.mClusterFlags [2 * glo_idx],
563- clustererNN.mClusterFlags [2 * glo_idx + 1 ]);
558+ static_cast <float >(peak.pad ()) + clustererNN.mOutputDataReg2_32 [model_output_index + 1 ],
559+ clustererNN.mOutputDataReg2_32 [model_output_index + 5 ],
560+ (clusterer.mPmemory ->fragment ).start + static_cast <float >(peak.time ()) + clustererNN.mOutputDataReg2_32 [model_output_index + 3 ],
561+ clustererNN.mOutputDataReg2_32 [model_output_index + 7 ],
562+ clustererNN.mClusterFlags [2 * glo_idx],
563+ clustererNN.mClusterFlags [2 * glo_idx + 1 ]);
564564 }
565565
566566 rejectCluster = !pc.toNative (peak, central_charge, myCluster, clusterer.Param (), chargeMap);
0 commit comments