instanceManager.hpp

Go to the documentation of this file.

/*
 *   Copyright 2025 Huawei Technologies Co., Ltd.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
 
#pragma once
 
#include <memory>
#include <pthread.h>
 
#include <hicr/core/instanceManager.hpp>
 
#include "core.hpp"
#include "instance.hpp"
 
namespace HiCR::backend::pthreads
{
 
class InstanceManager final : public HiCR::InstanceManager
{
  public:
 
  InstanceManager(Core &core)
    : HiCR::InstanceManager(),
      _core(core)
  {
    // Create and set current instance in the base class
    setCurrentInstance(_core.addInstance(pthread_self()));
 
    // Get root instance id
    _rootInstanceId = _core.getRootInstanceId();
 
    // Wait for all the threads to add their own instance
    _core.fence();
 
    // Add all the instances to the base class
    for (auto &i : _core.getInstances()) { addInstance(i); }
  }
 
  ~InstanceManager() override = default;
 
  std::shared_ptr<HiCR::Instance> createInstanceImpl(const HiCR::InstanceTemplate instanceTemplate) override { HICR_THROW_RUNTIME("This backend does not support this operation"); }
 
  std::shared_ptr<HiCR::Instance> addInstanceImpl(Instance::instanceId_t instanceId) override
  {
    // Get the instance from the core. Someone else already created it
    auto instance = _core.getInstance(instanceId);
    return instance;
  }
 
  void terminateInstanceImpl(const std::shared_ptr<HiCR::Instance> instance) override
  {
    // Nothing to do here
  }
 
  void finalize() override {}
 
  void abort(int errorCode) override { exit(errorCode); }
 
  HiCR::Instance::instanceId_t getRootInstanceId() const override { return _rootInstanceId; }
 
  private:
 
  Core &_core;
 
  HiCR::Instance::instanceId_t _rootInstanceId;
};
} // namespace HiCR::backend::pthreads