vtune_op_wrapper.h

Go to the documentation of this file.

// -----------------------------------------------------------------------------
//
// Copyright (C) 2021 CERN & University of Surrey for the benefit of the
// BioDynaMo collaboration. All Rights Reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
//
// See the LICENSE file distributed with this work for details.
// See the NOTICE file distributed with this work for additional information
// regarding copyright ownership.
//
// -----------------------------------------------------------------------------
 
#ifndef CORE_OPERATION_VTUNE_OP_WRAPPER_H_
#define CORE_OPERATION_VTUNE_OP_WRAPPER_H_
 
#include "core/util/vtune.h"
 
namespace bdm {
 
template <typename Op>
class VTuneOpWrapper : public Op {
 public:
  template <typename... Args>
  explicit VTuneOpWrapper(Args&&... args) : Op{std::forward<Args>(args)...} {
    domain_ = __itt_domain_create("MyTraces.MyDomain");
    task_ = __itt_string_handle_create(typeid(Op).name());
  }
 
  template <typename daosoa>
  void Compute(daosoa* cells) const {
    __itt_task_begin(domain_, __itt_null, __itt_null, task_);
    Op::Compute(cells);
    __itt_task_end(domain_);
  }
 
 private:
  __itt_domain* domain_;
  __itt_string_handle* task_;
};
 
}  // namespace bdm
 
#endif  // CORE_OPERATION_VTUNE_OP_WRAPPER_H_