Ejemplos de cublasSasum en C++ (Cpp)

Ejemplo n.º 1

Archivo: Spoc_cublas.c Proyecto: archonSTB/SPOC

CAMLprim value spoc_cublasSasum (value n, value x, value incx, value dev){
	CAMLparam4(n,x,incx, dev);
	CAMLlocal4(dev_vec_array, dev_vec, res, gi);
	CUdeviceptr d_A;
	float result;
	int id;
	GET_VEC(x, d_A);
	CUBLAS_GET_CONTEXT;
	result = cublasSasum(Int_val(n), (float*)d_A, Int_val(incx));
	CUBLAS_CHECK_CALL(cublasGetError());
	res = caml_copy_double((double)result);
	CUDA_RESTORE_CONTEXT;
	CAMLreturn((res));
}

Ejemplo n.º 2

Archivo: gpuops.c Proyecto: frankong/bart

static double cuda_asum(long size, const float* src)
{
	return cublasSasum(size, src, 1);
}

Ejemplo n.º 3

Archivo: AssomOnline.cpp Proyecto: thusithaC/GASSOM_CPP

float AssomOnline::getErr()
{
	float error;
	cublasSasum(cuhandle,AGENTSNUM,winErr,1,&error);
	return error/AGENTSNUM;
}

Ejemplo n.º 4

Archivo: math_functions.cpp Proyecto: xiaomi646/caffe_2d_mlabel

void caffe_gpu_asum<float>(const int n, const float* x, float* y) {
  CUBLAS_CHECK(cublasSasum(Caffe::cublas_handle(), n, x, 1, y));
}

Ejemplo n.º 5

float magma_sasum(
    magma_int_t n,
    const float *dx, magma_int_t incx )
{
    return cublasSasum( n, dx, incx );
}