memory_internal_device.h

Go to the documentation of this file.

#pragma once

#if defined(_HAS_CUDA_)

namespace magmadnn {
namespace internal {

template <typename T>
T get_device_array_element(T *arr, unsigned int idx);


template <typename T>
void set_device_array_element(T *arr, unsigned int idx, T val);

} // namespace internal
} // namespace magmadnn

#endif