tfaot::BatchRule Class Reference

#include <Batching.h>

Public Member Functions
	BatchRule (size_t batchSize, const std::vector< size_t > &sizes, size_t lastPadding=0)
	BatchRule (const std::string &ruleString)
size_t	getBatchSize () const
size_t	getLastPadding () const
size_t	getSize (size_t i) const
const std::vector< size_t > &	getSizes () const
size_t	nSizes () const
	~BatchRule ()=default

Private Member Functions
void	validate () const

Private Attributes
size_t	batchSize_
size_t	lastPadding_
std::vector< size_t >	sizes_

Detailed Description

Definition at line 19 of file Batching.h.

Constructor & Destructor Documentation

BatchRule() [1/2]

tfaot::BatchRule::BatchRule ( size_t  batchSize, const std::vector< size_t > &  sizes, size_t  lastPadding = 0  )

explicit

Definition at line 16 of file Batching.cc.

References validate().

      : batchSize_(batchSize), sizes_(sizes), lastPadding_(lastPadding) {
    validate();
  }

BatchRule() [2/2]

tfaot::BatchRule::BatchRule

(

const std::string &

ruleString

)

Definition at line 21 of file Batching.cc.

References HLT_FULL_cff::batchSize, batchSize_, Exception, lastPadding_, sizes_, AlCaHLTBitMon_QueryRunRegistry::string, and validate().

                                                  {
    // extract the target batch size from the front
    std::string rule = ruleString;
    auto pos = rule.find(":");
    if (pos == std::string::npos) {
      throw cms::Exception("InvalidBatchRule") << "invalid batch rule format: " << ruleString;
    }
    size_t batchSize = std::stoi(rule.substr(0, pos));
    rule = rule.substr(pos + 1);

    // loop through remaining comma-separated sizes
    std::vector<size_t> sizes;
    size_t sumSizes = 0;
    while (!rule.empty()) {
      pos = rule.find(",");
      sizes.push_back(std::stoi(rule.substr(0, pos)));
      sumSizes += sizes.back();
      rule = pos == std::string::npos ? "" : rule.substr(pos + 1);
    }

    // the sum of composite batch sizes should never be smaller than the target batch size
    if (sumSizes < batchSize) {
      throw cms::Exception("InvalidBatchRule")
          << "sum of composite batch sizes is smaller than target batch size: " << ruleString;
    }

    // set members and validate
    batchSize_ = batchSize;
    sizes_ = sizes;
    lastPadding_ = sumSizes - batchSize;
    validate();
  }

~BatchRule()

tfaot::BatchRule::~BatchRule ( )

default

Member Function Documentation

getBatchSize()

size_t tfaot::BatchRule::getBatchSize ( ) const

inline

Definition at line 32 of file Batching.h.

References batchSize_.

Referenced by tfaot::operator<<(), and tfaot::BatchStrategy::setRule().

{ return batchSize_; }

getLastPadding()

size_t tfaot::BatchRule::getLastPadding ( ) const

inline

Definition at line 38 of file Batching.h.

References lastPadding_.

Referenced by tfaot::operator<<(), and tfaot::Model< W >::run().

{ return lastPadding_; }

getSize()

size_t tfaot::BatchRule::getSize ( size_t  i ) const

inline

Definition at line 44 of file Batching.h.

References mps_fire::i, and sizes_.

Referenced by tfaot::Model< W >::run().

{ return sizes_[i]; }

getSizes()

const std::vector<size_t>& tfaot::BatchRule::getSizes ( ) const

inline

Definition at line 35 of file Batching.h.

References sizes_.

Referenced by tfaot::operator<<().

{ return sizes_; }

nSizes()

size_t tfaot::BatchRule::nSizes ( ) const

inline

Definition at line 41 of file Batching.h.

References sizes_.

Referenced by tfaot::operator<<(), and tfaot::Model< W >::run().

{ return sizes_.size(); }

validate()

void tfaot::BatchRule::validate ( ) const

private

Definition at line 54 of file Batching.cc.

References batchSize_, Exception, lastPadding_, alignCSCRings::s, and sizes_.

Referenced by BatchRule().

                                 {
    // sizes must not be empty
    if (sizes_.size() == 0) {
      throw cms::Exception("EmptySizes") << "no batch sizes provided for stitching";
    }

    // the padding must be smaller than the last size
    size_t lastSize = sizes_[sizes_.size() - 1];
    if (lastPadding_ >= lastSize) {
      throw cms::Exception("WrongPadding")
          << "padding " << lastPadding_ << " must be smaller than last size " << lastSize;
    }

    // compute the covered batch size
    size_t sizeSum = 0;
    for (const size_t& s : sizes_) {
      sizeSum += s;
    }
    if (lastPadding_ > sizeSum) {
      throw cms::Exception("WrongPadding")
          << "padding " << lastPadding_ << " must not be larger than sum of sizes " << sizeSum;
    }
    sizeSum -= lastPadding_;

    // compare to given batch size
    if (batchSize_ != sizeSum) {
      throw cms::Exception("WrongBatchSize")
          << "batch size " << batchSize_ << " does not match sum of sizes - padding " << sizeSum;
    }
  }

Member Data Documentation

batchSize_

size_t tfaot::BatchRule::batchSize_

private

Definition at line 47 of file Batching.h.

Referenced by BatchRule(), getBatchSize(), and validate().

lastPadding_

size_t tfaot::BatchRule::lastPadding_

private

Definition at line 49 of file Batching.h.

Referenced by BatchRule(), getLastPadding(), and validate().

sizes_

std::vector<size_t> tfaot::BatchRule::sizes_

private

Definition at line 48 of file Batching.h.

Referenced by BatchRule(), getSize(), getSizes(), nSizes(), and validate().