container::kernels::resize_memory< T, Device > Struct Template Reference

A functor to resize memory allocation. More...

#include <memory.h>

Collaboration diagram for container::kernels::resize_memory< T, Device >:

Public Member Functions
void	operator() (T *&arr, const size_t &size, const char *record_in=nullptr)
	Resize memory allocation.

Detailed Description

template<typename T, typename Device>
struct container::kernels::resize_memory< T, Device >

A functor to resize memory allocation.

Template Parameters

T	Floating-point type of the allocated memory.
Device	Device type where the memory will be allocated.

Member Function Documentation

operator()()

template<typename T , typename Device >

void container::kernels::resize_memory< T, Device >::operator()	(	T *&	arr,
		const size_t &	size,
		const char *	record_in = nullptr
	)

Resize memory allocation.

Parameters

dev	Device where the memory will be allocated.
arr	Pointer to the allocated memory.
size	New size of the allocated memory.
record_in	Optional message to record the resize operation.

---

The documentation for this struct was generated from the following file:

• /home/runner/work/abacus-develop/abacus-develop/source/source_base/module_container/ATen/kernels/memory.h