rocm-systems/rocclr/runtime/device/cpu/cpuvirtual.hpp

//
// Copyright (c) 2011 Advanced Micro Devices, Inc. All rights reserved.
//

#ifndef CPUVIRTUAL_HPP_
#define CPUVIRTUAL_HPP_

#include "top.hpp"
#include "device/device.hpp"
#include "thread/atomic.hpp"
#include "thread/thread.hpp"
#include "platform/ndrange.hpp"

//! \namespace cpu CPU Device Implementation
namespace cpu {

class WorkerThread;
class Device;

class VirtualCPU : public device::VirtualDevice {
 private:
  WorkerThread** cores_;                         //!< Pointer to array of Worker threads
  static amd::Atomic<size_t> numWorkerThreads_;  //!< Current Worker Threads number
  bool acceptingCommands_;

 public:
  VirtualCPU(cpu::Device& device);
  ~VirtualCPU();
  bool terminate();

  WorkerThread* getWorkerThread(size_t id) { return cores_[id]; }

  bool acceptingCommands() const { return acceptingCommands_; }

  virtual void submitReadMemory(amd::ReadMemoryCommand& command);
  virtual void submitWriteMemory(amd::WriteMemoryCommand& command);
  virtual void submitCopyMemory(amd::CopyMemoryCommand& command);
  virtual void submitCopyMemoryP2P(amd::CopyMemoryP2PCommand& command) {}
  virtual void submitMapMemory(amd::MapMemoryCommand& command);
  virtual void submitUnmapMemory(amd::UnmapMemoryCommand& command);
  virtual void submitKernel(amd::NDRangeKernelCommand& command);
  virtual void submitNativeFn(amd::NativeFnCommand& command);
  virtual void submitMarker(amd::Marker& command);
  virtual void submitFillMemory(amd::FillMemoryCommand& command);
  virtual void submitMigrateMemObjects(amd::MigrateMemObjectsCommand& cmd) {}
  virtual void submitAcquireExtObjects(amd::AcquireExtObjectsCommand& cmd);
  virtual void submitReleaseExtObjects(amd::ReleaseExtObjectsCommand& cmd);
  virtual void submitPerfCounter(amd::PerfCounterCommand& cmd);
  virtual void submitThreadTraceMemObjects(amd::ThreadTraceMemObjectsCommand& cmd);
  virtual void submitThreadTrace(amd::ThreadTraceCommand& cmd);
  virtual void flush(amd::Command* list = NULL, bool wait = false);
  virtual void submitSignal(amd::SignalCommand& cmd);
  virtual void submitMakeBuffersResident(amd::MakeBuffersResidentCommand& cmd);
  virtual void submitSvmFreeMemory(amd::SvmFreeMemoryCommand& cmd);
  virtual void submitSvmCopyMemory(amd::SvmCopyMemoryCommand& cmd);
  virtual void submitSvmFillMemory(amd::SvmFillMemoryCommand& cmd);
  virtual void submitSvmMapMemory(amd::SvmMapMemoryCommand& cmd);
  virtual void submitSvmUnmapMemory(amd::SvmUnmapMemoryCommand& cmd);

  virtual void computeLocalSizes(amd::NDRangeKernelCommand& command, amd::NDRange& local);

  static bool fillImage(amd::Image& image, address fillMem, const void* pattern,
                        const amd::Coord3D& origin, const amd::Coord3D& region, size_t rowPitch,
                        size_t slicePitch, size_t elementSize);
};

}  // namespace cpu

#endif  // CPUVIRTUAL_HPP_