container::kernels::cast_memory< T_out, T_in, Device_out, Device_in > Struct Template Reference

Casts memory between devices. More...

#include <memory.h>

Collaboration diagram for container::kernels::cast_memory< T_out, T_in, Device_out, Device_in >:

Public Member Functions
void	operator() (T_out *arr_out, const T_in *arr_in, const size_t &size)
	Casts memory between devices.

Detailed Description

template<typename T_out, typename T_in, typename Device_out, typename Device_in>
struct container::kernels::cast_memory< T_out, T_in, Device_out, Device_in >

Casts memory between devices.

This class casts memory between two different devices.

Template Parameters

T_out	The output data type.
T_in	The input data type.
Device_out	The output device.
Device_in	The input device.

Member Function Documentation

operator()()

template<typename T_out , typename T_in , typename Device_out , typename Device_in >

void container::kernels::cast_memory< T_out, T_in, Device_out, Device_in >::operator()	(	T_out *	arr_out,
		const T_in *	arr_in,
		const size_t &	size
	)

Casts memory between devices.

This method casts memory between two different devices.

Parameters

dev_out	The output device.
dev_in	The input device.
arr_out	The output array.
arr_in	The input array.
size	The size of the array.

---

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