data_ptr.hpp

#pragma once
 
#include <functional>
#include <iostream>
#include <memory>
 
namespace lmp::tensor::detail {
 
 
struct DataPtr {
  std::shared_ptr<void> ptr;
 
  DataPtr() = default;
  DataPtr(void* data_ptr, const std::function<void(void*)>& dealloc)
      : ptr(data_ptr, [dealloc](void* data) {
          try {
            dealloc(data);
          } catch (const std::exception& e) {
            std::cerr << "DataPtr deleter error: " << e.what() << "\n";
          } catch (...) {
            std::cerr << "DataPtr deleter threw non-std::exception\n";
          }
        }) {}
 
  void* data() const { return ptr.get(); }
};
 
}  // namespace lmp::tensor::detail