Transfer.Copy.h

Go to the documentation of this file.

#pragma once
 
#include <cuda_runtime.h>
#include <cstddef>
 
namespace Mila::Dnn::Compute::Cuda
{
    template <typename SrcT, typename DstT>
    void launch_convert_copy_kernel( const SrcT* d_src, DstT* d_or_h_dst, size_t n, cudaStream_t stream );

    template <typename T>
    void launch_fast_copy_kernel( const T* d_src, T* d_dst, size_t n, cudaStream_t stream );
}