Class CLMemory

tensorflow/lite/delegates/gpu/cl/cl_memory.h:30–81

RAII wrapper for OpenCL memory object. Image is moveable but not copyable.

//
// Image is moveable but not copyable.
class CLMemory {
 public:
  // Creates invalid object.
  CLMemory() : CLMemory(nullptr, false) {}

  CLMemory(cl_mem memory, bool has_ownership)
      : memory_(memory), has_ownership_(has_ownership) {}

  // Move-only
  CLMemory(const CLMemory&) = delete;
  CLMemory& operator=(const CLMemory&) = delete;
  CLMemory(CLMemory&& image)
      : memory_(image.memory_), has_ownership_(image.has_ownership_) {
    image.memory_ = nullptr;
  }

  ~CLMemory() { Invalidate(); }

  CLMemory& operator=(CLMemory&& image) {
    if (this != &image) {
      Invalidate();
      std::swap(memory_, image.memory_);
      has_ownership_ = image.has_ownership_;
    }
    return *this;
  }

  cl_mem memory() const { return memory_; }

  bool is_valid() const { return memory_ != nullptr; }

  // @return true if this object actually owns corresponding CL memory
  //         and manages it's lifetime.
  bool has_ownership() const { return has_ownership_; }

  cl_mem Release() {
    cl_mem to_return = memory_;
    memory_ = nullptr;
    return to_return;
  }

 private:
  void Invalidate() {
    if (memory_ && has_ownership_) {
      clReleaseMemObject(memory_);
    }
    memory_ = nullptr;
  }

  cl_mem memory_ = nullptr;
  bool has_ownership_ = false;
};

cl_mem_flags ToClMemFlags(AccessType access_type);