Skip to content

Commit

Permalink
edits added
Browse files Browse the repository at this point in the history
  • Loading branch information
quinnanm committed Jan 26, 2024
1 parent dd6f8b5 commit d5058b1
Showing 1 changed file with 17 additions and 13 deletions.
30 changes: 17 additions & 13 deletions L1Trigger/L1TGlobal/src/AXOL1TLCondition.cc
Original file line number Diff line number Diff line change
Expand Up @@ -91,11 +91,9 @@ const bool l1t::AXOL1TLCondition::evaluateCondition(const int bxEval) const {

//HLS4ML stuff
std::string AXOL1TLmodelversion = m_AXOL1TLmodelversion; //config loading method
// std::string AXOL1TLmodelversion = "L1Trigger/L1TGlobal/test/GTADModel_v3"; //for .so file in test dir loading method
hls4mlEmulator::ModelLoader loader(AXOL1TLmodelversion);
std::shared_ptr<hls4mlEmulator::Model> model;
model = loader.load_model();
// cout << "loading model... " << AXOL1TLmodelversion << std::endl;

// //pointers to objects
const BXVector<const l1t::Muon*>* candMuVec = m_gtGTB->getCandL1Mu();
Expand All @@ -117,24 +115,30 @@ const bool l1t::AXOL1TLCondition::evaluateCondition(const int bxEval) const {
//total # inputs in vector is (4+10+4+1)*3 = 57
const int NInputs = 57;

//types of inputs and outputs
typedef ap_fixed<18, 13> inputtype;
typedef std::array<ap_fixed<10, 7, AP_RND_CONV, AP_SAT>, 8> resulttype; //v3
typedef ap_ufixed<18, 14> losstype;
typedef std::pair<resulttype, losstype> pairtype;
// typedef std::array<ap_fixed<10, 7>, 13> resulttype; //deprecated v1 type:

//define zero
ap_fixed<18, 13> fillzero = 0.0;
inputtype fillzero = 0.0;

//AD vector declaration, will fill later
ap_fixed<18, 13> ADModelInput[NInputs] = {};
inputtype ADModelInput[NInputs] = {};

//initializing vector by type for my sanity
ap_fixed<18, 13> MuInput[MuVecSize];
ap_fixed<18, 13> JetInput[JVecSize];
ap_fixed<18, 13> EgammaInput[EGVecSize];
ap_fixed<18, 13> EtSumInput[EtSumVecSize];
inputtype MuInput[MuVecSize];
inputtype JetInput[JVecSize];
inputtype EgammaInput[EGVecSize];
inputtype EtSumInput[EtSumVecSize];

//declare result vectors +score
std::array<ap_fixed<10, 7, AP_RND_CONV, AP_SAT>, 8> result; //v3
ap_ufixed<18, 14> loss;
std::pair<std::array<ap_fixed<10, 7, AP_RND_CONV, AP_SAT>, 8>, ap_ufixed<18, 14>>
ADModelResult; //model outputs a pair of the (result vector, loss)
float score = -1.0; //not sure what the best default is hm??
resulttype result;
losstype loss;
pairtype ADModelResult; //model outputs a pair of the (result vector, loss)
float score = -1.0; //not sure what the best default is hm??

//check number of input objects we actually have (muons, jets etc)
int NCandMu = candMuVec->size(useBx);
Expand Down

0 comments on commit d5058b1

Please sign in to comment.