container::kernels::synchronize_memory< T, Device_out, Device_in > Struct Template Reference

Synchronizes memory between devices. More...

#include <memory.h>

Collaboration diagram for container::kernels::synchronize_memory< T, Device_out, Device_in >:

Public Member Functions
void	operator() (T *arr_out, const T *arr_in, const size_t &size)
	Synchronizes memory between devices.

Detailed Description

template<typename T, typename Device_out, typename Device_in>
struct container::kernels::synchronize_memory< T, Device_out, Device_in >

Synchronizes memory between devices.

This class synchronizes memory between two different devices.

Template Parameters

T	The type of data in the arrays.
Device_out	The output device.
Device_in	The input device.

Member Function Documentation

operator()()

template<typename T , typename Device_out , typename Device_in >

void container::kernels::synchronize_memory< T, Device_out, Device_in >::operator()	(	T *	arr_out,
		const T *	arr_in,
		const size_t &	size
	)

Synchronizes memory between devices.

This method synchronizes 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