memory_ops.hpp

#pragma once
 
#include "lamppp/tensor/data_ptr.hpp"
#include "lamppp/tensor/device_type.hpp"
#include "lamppp/tensor/dispatch_stub.hpp"
#include "lamppp/tensor/scalar.hpp"
 
namespace lmp::tensor {
class Tensor;
 
namespace ops {
 
using copy_fn = void (*)(DeviceType, const void*, void*, size_t, DataType,
                         DataType);
using empty_fn = detail::DataPtr (*)(size_t);
using fill_fn = void (*)(void*, size_t, Scalar, DataType type);
using resize_fn = void (*)(detail::DataPtr, size_t, size_t);
 
LMP_DECLARE_DISPATCH(copy_fn, copy_stub);
LMP_DECLARE_DISPATCH(empty_fn, empty_stub);
LMP_DECLARE_DISPATCH(fill_fn, fill_stub);
LMP_DECLARE_DISPATCH(resize_fn, resize_stub);
 
Tensor to(const Tensor& a, DeviceType to_device);
 
}  // namespace lmp::tensor::ops
}