memory.hpp

#pragma once
 
#include <cstddef>
#include "lamppp/tensor/data_ptr.hpp"
#include "lamppp/tensor/data_type.hpp"
#include "lamppp/tensor/device_type.hpp"
 
namespace lmp::tensor::detail::cpu {
 
void copy_cpu(DeviceType to_device, const void* src, void* dest, size_t size,
              DataType src_dtype, DataType dest_dtype);
DataPtr empty_cpu(size_t byte_size);
void fill_cpu(void* ptr, size_t size, Scalar t, DataType type);
void resize_cpu(DataPtr dptr, size_t old_byte_size, size_t new_byte_size);
 
template <typename U, typename V>
void vecCopy(size_t size, const U* in, V* out);
 
}  // namespace lmp::tensor::detail::cpu