Classes | |
struct | caloRecHitCollections |
struct | InputDescTau |
Public Types | |
using | TracksHost = pixelTrack::TracksHostPhase1 |
Public Types inherited from edm::stream::EDProducer< edm::GlobalCache< L2TauNNProducerAlpakaCacheData > > | |
using | CacheTypes = CacheContexts< T... > |
using | GlobalCache = typename CacheTypes::GlobalCache |
using | HasAbility = AbilityChecker< T... > |
using | InputProcessBlockCache = typename CacheTypes::InputProcessBlockCache |
using | LuminosityBlockCache = typename CacheTypes::LuminosityBlockCache |
using | LuminosityBlockContext = LuminosityBlockContextT< LuminosityBlockCache, RunCache, GlobalCache > |
using | LuminosityBlockSummaryCache = typename CacheTypes::LuminosityBlockSummaryCache |
using | RunCache = typename CacheTypes::RunCache |
using | RunContext = RunContextT< RunCache, GlobalCache > |
using | RunSummaryCache = typename CacheTypes::RunSummaryCache |
Public Member Functions | |
L2TauNNProducerAlpaka (const edm::ParameterSet &, const L2TauNNProducerAlpakaCacheData *) | |
Public Member Functions inherited from edm::stream::EDProducer< edm::GlobalCache< L2TauNNProducerAlpakaCacheData > > | |
EDProducer ()=default | |
EDProducer (const EDProducer &)=delete | |
bool | hasAbilityToProduceInBeginLumis () const final |
bool | hasAbilityToProduceInBeginProcessBlocks () const final |
bool | hasAbilityToProduceInBeginRuns () const final |
bool | hasAbilityToProduceInEndLumis () const final |
bool | hasAbilityToProduceInEndProcessBlocks () const final |
bool | hasAbilityToProduceInEndRuns () const final |
const EDProducer & | operator= (const EDProducer &)=delete |
Static Public Member Functions | |
static void | fillDescriptions (edm::ConfigurationDescriptions &) |
static void | globalEndJob (L2TauNNProducerAlpakaCacheData *) |
static std::unique_ptr< L2TauNNProducerAlpakaCacheData > | initializeGlobalCache (const edm::ParameterSet &) |
Static Public Attributes | |
static constexpr float | dEta_width = 2 * L2TauTagNNv1::dR_max / static_cast<float>(L2TauTagNNv1::nCellEta) |
static constexpr float | dPhi_width = 2 * L2TauTagNNv1::dR_max / static_cast<float>(L2TauTagNNv1::nCellPhi) |
static constexpr float | dR2_max = L2TauTagNNv1::dR_max * L2TauTagNNv1::dR_max |
Private Member Functions | |
void | checknan (tensorflow::Tensor &tensor, int debugLevel) |
void | fillCaloRecHits (tensorflow::Tensor &cellGridMatrix, const std::vector< l1t::TauRef > &allTaus, const caloRecHitCollections &caloRecHits) |
void | fillL1TauVars (tensorflow::Tensor &cellGridMatrix, const std::vector< l1t::TauRef > &allTaus) |
void | fillPatatracks (tensorflow::Tensor &cellGridMatrix, const std::vector< l1t::TauRef > &allTaus, const TracksHost &patatracks_tsoa, const ZVertexHost &patavtx_soa, const reco::BeamSpot &beamspot, const MagneticField *magfi) |
template<typename VPos , typename LVec > | |
std::tuple< float, float, int, int > | getEtaPhiIndices (const VPos &position, const LVec &tau_p4) |
template<typename LVec > | |
std::tuple< float, float, int, int > | getEtaPhiIndices (float eta, float phi, const LVec &tau_p4) |
std::vector< float > | getTauScore (const tensorflow::Tensor &cellGridMatrix) |
std::pair< float, float > | impactParameter (int it, const TracksHost &patatracks_tsoa, float patatrackPhi, const reco::BeamSpot &beamspot, const MagneticField *magfi) |
void | produce (edm::Event &event, const edm::EventSetup &eventsetup) override |
void | selectGoodTracksAndVertices (const ZVertexHost &patavtx_soa, const TracksHost &patatracks_tsoa, std::vector< int > &trkGood, std::vector< int > &vtxGood) |
void | standardizeTensor (tensorflow::Tensor &tensor) |
Private Attributes | |
const edm::EDGetTokenT< reco::BeamSpot > | beamSpotToken_ |
const edm::ESGetToken< MagneticField, IdealMagneticFieldRecord > | bFieldToken_ |
const int | debugLevel_ |
const edm::EDGetTokenT< EcalRecHitCollection > | ebToken_ |
const edm::EDGetTokenT< EcalRecHitCollection > | eeToken_ |
const float | fractionSumPt2_ |
const edm::ESGetToken< CaloGeometry, CaloGeometryRecord > | geometryToken_ |
const edm::EDGetTokenT< HBHERecHitCollection > | hbheToken_ |
const edm::EDGetTokenT< HORecHitCollection > | hoToken_ |
std::string | inputTensorName_ |
std::vector< InputDescTau > | L1TauDesc_ |
const L2TauNNProducerAlpakaCacheData * | L2cacheData_ |
const unsigned int | maxVtx_ |
const float | minSumPt2_ |
std::string | outputTensorName_ |
const edm::EDGetTokenT< TracksHost > | pataTracksToken_ |
const edm::EDGetTokenT< ZVertexHost > | pataVerticesToken_ |
const edm::EDGetTokenT< trigger::TriggerFilterObjectWithRefs > | tauTriggerToken_ |
const float | trackChi2Max_ |
const float | trackPtMax_ |
const float | trackPtMin_ |
Definition at line 147 of file L2TauTagNNProducerAlpaka.cc.
Definition at line 149 of file L2TauTagNNProducerAlpaka.cc.
|
explicit |
Definition at line 290 of file L2TauTagNNProducerAlpaka.cc.
References looper::cfg, L2TauNNProducerAlpaka::InputDescTau::CollectionName, submitPVResolutionJobs::desc, Exception, L2TauNNProducerAlpakaCacheData::graphDef, inputTensorName_, L2TauNNProducerAlpaka::InputDescTau::inputToken_, L1TauDesc_, L2cacheData_, Skims_PA_cff::name, outputTensorName_, and AlCaHLTBitMon_QueryRunRegistry::string.
|
private |
Definition at line 327 of file L2TauTagNNProducerAlpaka.cc.
References ztail::d, HLT_2024v12_cff::debugLevel, Exception, input, edm::isNotFinite(), and L2TauTagNNv1::varNameMap.
Referenced by produce().
|
private |
Definition at line 437 of file L2TauTagNNProducerAlpaka.cc.
References reco::deltaR2(), dR2_max, L2TauNNProducerAlpaka::caloRecHitCollections::eb, L2TauNNProducerAlpaka::caloRecHitCollections::ee, L2TauNNProducerAlpaka::caloRecHitCollections::geometry, getEtaPhiIndices(), CaloGeometry::getGeometry(), L2TauNNProducerAlpaka::caloRecHitCollections::hbhe, L2TauNNProducerAlpaka::caloRecHitCollections::ho, input, L2TauTagNNv1::nCellEta, L2TauTagNNv1::nCellPhi, and position.
Referenced by produce().
|
static |
Definition at line 258 of file L2TauTagNNProducerAlpaka.cc.
References edm::ParameterSetDescription::add(), edm::ParameterSet::addParameter(), edm::ConfigurationDescriptions::addWithDefaultLabel(), submitPVResolutionJobs::desc, ProducerED_cfi::InputTag, AlCaHLTBitMon_QueryRunRegistry::string, and parallelization::uint.
|
private |
Definition at line 405 of file L2TauTagNNProducerAlpaka.cc.
References input, L2TauTagNNv1::nCellEta, and L2TauTagNNv1::nCellPhi.
Referenced by produce().
|
private |
Definition at line 655 of file L2TauTagNNProducerAlpaka.cc.
References reco::charge(), PortableHostCollection< T >::const_view(), reco::deltaR2(), dR2_max, spr::find(), getEtaPhiIndices(), impactParameter(), input, ALPAKA_ACCELERATOR_NAMESPACE::vertexFinder::it, SiStripPI::min, L2TauTagNNv1::nCellEta, L2TauTagNNv1::nCellPhi, nHits, TrackCollections2monitor_cff::nVertices, reco::phi(), selectGoodTracksAndVertices(), and PortableHostCollection< T >::view().
Referenced by produce().
|
private |
Definition at line 433 of file L2TauTagNNProducerAlpaka.cc.
References position.
Referenced by fillCaloRecHits(), and fillPatatracks().
|
private |
Definition at line 424 of file L2TauTagNNProducerAlpaka.cc.
References reco::deltaPhi(), dEta_width, dPhi_width, L2TauTagNNv1::dR_max, PVValHelper::eta, and phi.
|
private |
Definition at line 733 of file L2TauTagNNProducerAlpaka.cc.
References L2cacheData_, outputTensorName_, tensorflow::run(), and L2TauNNProducerAlpakaCacheData::session.
Referenced by produce().
|
static |
Definition at line 252 of file L2TauTagNNProducerAlpaka.cc.
References tensorflow::closeSession(), L2TauNNProducerAlpakaCacheData::graphDef, and L2TauNNProducerAlpakaCacheData::session.
|
private |
Definition at line 624 of file L2TauTagNNProducerAlpaka.cc.
References LocalTrajectoryParameters::charge(), TracksUtilities< TrackerTraits >::copyToDense(), funct::cos(), f, runTauDisplay::gp, ALPAKA_ACCELERATOR_NAMESPACE::vertexFinder::it, M_PI_2, LocalTrajectoryParameters::momentum(), phi, LocalTrajectoryParameters::position(), createTree::pp, funct::sin(), riemannFit::transformToPerigeePlane(), and PortableHostCollection< T >::view().
Referenced by fillPatatracks().
|
static |
Definition at line 226 of file L2TauTagNNProducerAlpaka.cc.
References looper::cfg, tensorflow::createSession(), edm::FileInPath::fullPath(), HLT_2024v12_cff::graphPath, submitPVResolutionJobs::key, tensorflow::loadGraphDef(), normDictElement::max, normDictElement::mean, normDictElement::min, HLT_2024v12_cff::normalizationDict, L2TauTagNNv1::nVars, tensorflow::setLogging(), normDictElement::std, AlCaHLTBitMon_QueryRunRegistry::string, heppy_batch::val, trigObjTnPSource_cfi::var, and L2TauTagNNv1::varNameMap.
|
overrideprivate |
Definition at line 749 of file L2TauTagNNProducerAlpaka.cc.
References beamSpotToken_, bFieldToken_, checknan(), debugLevel_, HLT_2024v12_cff::distance, L2TauNNProducerAlpaka::caloRecHitCollections::eb, ebToken_, L2TauNNProducerAlpaka::caloRecHitCollections::ee, eeToken_, fillCaloRecHits(), fillL1TauVars(), fillPatatracks(), spr::find(), relativeConstraints::geometry, L2TauNNProducerAlpaka::caloRecHitCollections::geometry, geometryToken_, edm::EventSetup::getHandle(), getTauScore(), photonIsolationHIProducer_cfi::hbhe, L2TauNNProducerAlpaka::caloRecHitCollections::hbhe, hbheToken_, photonIsolationHIProducer_cfi::ho, L2TauNNProducerAlpaka::caloRecHitCollections::ho, hoToken_, L1TauDesc_, eostools::move(), L2TauTagNNv1::nCellEta, L2TauTagNNv1::nCellPhi, L2TauTagNNv1::nVars, pataTracksToken_, pataVerticesToken_, DiDispStaMuonMonitor_cfi::pt, standardizeTensor(), and trigger::TriggerL1Tau.
|
private |
Definition at line 576 of file L2TauTagNNProducerAlpaka.cc.
References cms::cuda::assert(), PortableHostCollection< T >::const_view(), fractionSumPt2_, dqmiolumiharvest::j, pixelTrack::loose, DMR_cfg::maxTracks, maxVtx_, SiStripPI::min, minSumPt2_, nHits, quality, trackChi2Max_, trackPtMax_, trackPtMin_, and PortableHostCollection< T >::view().
Referenced by fillPatatracks().
|
private |
Definition at line 371 of file L2TauTagNNProducerAlpaka.cc.
References ztail::d, Exception, input, L2cacheData_, SiStripPI::max, SiStripPI::mean, SiStripPI::min, and L2TauNNProducerAlpakaCacheData::normVec.
Referenced by produce().
|
private |
Definition at line 214 of file L2TauTagNNProducerAlpaka.cc.
Referenced by produce().
|
private |
Definition at line 211 of file L2TauTagNNProducerAlpaka.cc.
Referenced by produce().
|
private |
Definition at line 203 of file L2TauTagNNProducerAlpaka.cc.
Referenced by produce().
|
static |
Definition at line 165 of file L2TauTagNNProducerAlpaka.cc.
Referenced by getEtaPhiIndices().
|
static |
Definition at line 166 of file L2TauTagNNProducerAlpaka.cc.
Referenced by getEtaPhiIndices().
|
static |
Definition at line 164 of file L2TauTagNNProducerAlpaka.cc.
Referenced by fillCaloRecHits(), and fillPatatracks().
|
private |
Definition at line 208 of file L2TauTagNNProducerAlpaka.cc.
Referenced by produce().
|
private |
Definition at line 209 of file L2TauTagNNProducerAlpaka.cc.
Referenced by produce().
|
private |
Definition at line 216 of file L2TauTagNNProducerAlpaka.cc.
Referenced by selectGoodTracksAndVertices().
|
private |
Definition at line 210 of file L2TauTagNNProducerAlpaka.cc.
Referenced by produce().
|
private |
Definition at line 206 of file L2TauTagNNProducerAlpaka.cc.
Referenced by produce().
|
private |
Definition at line 207 of file L2TauTagNNProducerAlpaka.cc.
Referenced by produce().
|
private |
Definition at line 221 of file L2TauTagNNProducerAlpaka.cc.
Referenced by L2TauNNProducerAlpaka().
|
private |
Definition at line 205 of file L2TauTagNNProducerAlpaka.cc.
Referenced by L2TauNNProducerAlpaka(), and produce().
|
private |
Definition at line 223 of file L2TauTagNNProducerAlpaka.cc.
Referenced by getTauScore(), L2TauNNProducerAlpaka(), and standardizeTensor().
|
private |
Definition at line 215 of file L2TauTagNNProducerAlpaka.cc.
Referenced by selectGoodTracksAndVertices().
|
private |
Definition at line 217 of file L2TauTagNNProducerAlpaka.cc.
Referenced by selectGoodTracksAndVertices().
|
private |
Definition at line 222 of file L2TauTagNNProducerAlpaka.cc.
Referenced by getTauScore(), and L2TauNNProducerAlpaka().
|
private |
Definition at line 213 of file L2TauTagNNProducerAlpaka.cc.
Referenced by produce().
|
private |
Definition at line 212 of file L2TauTagNNProducerAlpaka.cc.
Referenced by produce().
|
private |
Definition at line 204 of file L2TauTagNNProducerAlpaka.cc.
|
private |
Definition at line 220 of file L2TauTagNNProducerAlpaka.cc.
Referenced by selectGoodTracksAndVertices().
|
private |
Definition at line 219 of file L2TauTagNNProducerAlpaka.cc.
Referenced by selectGoodTracksAndVertices().
|
private |
Definition at line 218 of file L2TauTagNNProducerAlpaka.cc.
Referenced by selectGoodTracksAndVertices().