offloadable.hpp

// SPDX-License-Identifier: MIT
// Copyright (c) 2021  Kenji Koide (k.koide@aist.go.jp)
 
#pragma once
 
#include <memory>
#include <vector>
#include <atomic>
#include <cstdint>
 
// forward declaration
struct CUstream_st;
 
namespace gtsam_points {
 
class OffloadableGPU {
public:
  using Ptr = std::shared_ptr<OffloadableGPU>;
  using ConstPtr = std::shared_ptr<const OffloadableGPU>;
 
  OffloadableGPU();
  virtual ~OffloadableGPU();
 
  static std::uint64_t current_access_time();
 
  std::uint64_t last_accessed_time() const;
 
  virtual size_t memory_usage_gpu() const { return 0; }
 
  virtual bool loaded_on_gpu() const = 0;
 
  virtual bool touch(CUstream_st* stream = 0);
 
  virtual bool offload_gpu(CUstream_st* stream = 0) = 0;
 
  virtual bool reload_gpu(CUstream_st* stream = 0) = 0;
 
protected:
  static std::atomic_uint64_t access_counter;  
  std::uint64_t last_access;                   
};
 
}  // namespace gtsam_points