Apache SINGA
A distributed deep learning platform .
 All Classes Namespaces Files Functions Variables Typedefs Macros
Namespaces | Functions
tensor_gpu-inl.hpp File Reference

implementation of GPU host code More...

#include "tensor.h"

Go to the source code of this file.

Namespaces

 mshadow
 namespace for mshadow
 

Functions

void mshadow::InitTensorEngine (int device_id=0)
 initialize tensor engine, used to call intialization functions of dependent libs this function should be called before all GPU tensor operations, for using tensors in CPU, this call is actually not needed More...
 
void mshadow::ShutdownTensorEngine (void)
 Shutdown tensor engine, this function should be called after all GPU tensor operations, for using tensors in CPU, this call is actually not needed.
 
template<int dim>
void mshadow::AllocSpace (Tensor< gpu, dim > &obj, bool pad=MSHADOW_ALLOC_PAD)
 refer to comment of cpu ver More...
 
template<int dim>
void mshadow::FreeSpace (Tensor< gpu, dim > &obj)
 refer to comment of cpu ver More...
 
template<typename A , typename B , int dim>
void mshadow::Copy (Tensor< A, dim > _dst, Tensor< B, dim > _src, cudaMemcpyKind kind)
 
template<int dim>
void mshadow::Copy (Tensor< cpu, dim > dst, const Tensor< gpu, dim > &src)
 refer to comment of cpu ver More...
 
template<int dim>
void mshadow::Copy (Tensor< gpu, dim > dst, const Tensor< gpu, dim > &src)
 refer to comment of cpu ver More...
 
template<int dim>
void mshadow::Copy (Tensor< gpu, dim > dst, const Tensor< cpu, dim > &src)
 refer to comment of cpu ver More...
 

Detailed Description

implementation of GPU host code

Author
Bing Xu, Tianqi Chen