GPUMemorySizeScalers.cxx

Go to the documentation of this file.

// Copyright 2019-2020 CERN and copyright holders of ALICE O2.
// See https://alice-o2.web.cern.ch/copyright for details of the copyright holders.
// All rights not expressly granted are reserved.
//
// This software is distributed under the terms of the GNU General Public
// License v3 (GPL Version 3), copied verbatim in the file "COPYING".
//
// In applying this license CERN does not waive the privileges and immunities
// granted to it by virtue of its status as an Intergovernmental Organization
// or submit itself to any jurisdiction.
 
 
#include "GPUMemorySizeScalers.h"
#include "GPULogging.h"
 
#include <random>
 
using namespace o2::gpu;
 
void GPUMemorySizeScalers::rescaleMaxMem(size_t newAvailableMemory)
{
  GPUMemorySizeScalers tmp;
  double scaleFactor = (double)newAvailableMemory / tmp.availableMemory;
  if (scaleFactor != 1.) {
    GPUInfo("Rescaling buffer size limits from %lu to %lu bytes of memory (factor %f)", tmp.availableMemory, newAvailableMemory, scaleFactor);
  }
  tpcMaxPeaks = (double)tmp.tpcMaxPeaks * scaleFactor;
  tpcMaxClusters = (double)tmp.tpcMaxClusters * scaleFactor;
  tpcMaxStartHits = (double)tmp.tpcMaxStartHits * scaleFactor;
  tpcMaxRowStartHits = (double)tmp.tpcMaxRowStartHits * scaleFactor;
  tpcMaxTracklets = (double)tmp.tpcMaxTracklets * scaleFactor;
  tpcMaxTrackletHits = (double)tmp.tpcMaxTrackletHits * scaleFactor;
  tpcMaxSectorTracks = (double)tmp.tpcMaxSectorTracks * scaleFactor;
  tpcMaxSectorTrackHits = (double)tmp.tpcMaxSectorTrackHits * scaleFactor;
  tpcMaxMergedTracks = (double)tmp.tpcMaxMergedTracks * scaleFactor;
  tpcMaxMergedTrackHits = (double)tmp.tpcMaxMergedTrackHits * scaleFactor;
  availableMemory = newAvailableMemory;
}
 
double GPUMemorySizeScalers::getScalingFactor()
{
  if (!doFuzzing) {
    return scalingFactor;
  }
  static std::uniform_int_distribution<uint32_t> dist(0, 1000000);
  static std::mt19937 rng;
  if (fuzzSeed) {
    rng = std::mt19937(fuzzSeed);
    fuzzLimit = dist(rng) / 10;
    fuzzSeed = 0;
  }
  if (dist(rng) > fuzzLimit) {
    return scalingFactor;
  }
  return scalingFactor * 0.000001 * dist(rng);
}
 
void GPUMemorySizeScalers::fuzzScalingFactor(uint64_t seed)
{
  fuzzSeed = seed;
  doFuzzing = true;
}