starpu-max/examples/mult/xgemm_kernels.c

/* StarPU --- Runtime system for heterogeneous multicore architectures.
 *
 * Copyright (C) 2009, 2010  Université de Bordeaux 1
 * Copyright (C) 2010  Centre National de la Recherche Scientifique
 *
 * StarPU is free software; you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation; either version 2.1 of the License, or (at
 * your option) any later version.
 *
 * StarPU is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
 *
 * See the GNU Lesser General Public License in COPYING.LGPL for more details.
 */

#include <starpu.h>
#include <starpu_cuda.h>
#include <common/blas.h>

#define COMMON_CODE			\
	uint32_t nxC, nyC, nyA;		\
	uint32_t ldA, ldB, ldC;		\
					\
	TYPE *subA;			\
	TYPE *subB;			\
	TYPE *subC;			\
					\
	subA = (TYPE *)STARPU_MATRIX_GET_PTR(descr[0]);	\
	subB = (TYPE *)STARPU_MATRIX_GET_PTR(descr[1]);	\
	subC = (TYPE *)STARPU_MATRIX_GET_PTR(descr[2]);	\
					\
	nxC = STARPU_MATRIX_GET_NX(descr[2]);		\
	nyC = STARPU_MATRIX_GET_NY(descr[2]);		\
	nyA = STARPU_MATRIX_GET_NY(descr[0]);		\
					\
	ldA = STARPU_MATRIX_GET_LD(descr[0]);		\
	ldB = STARPU_MATRIX_GET_LD(descr[1]);		\
	ldC = STARPU_MATRIX_GET_LD(descr[2]);



#ifdef STARPU_USE_CUDA

#ifdef STARPU_HAVE_MAGMA
#define GPU_GEMM MAGMABLAS_GEMM
#else
#define GPU_GEMM CUBLAS_GEMM
#endif

void STARPU_GEMM(cublas_mult)(void *descr[], __attribute__((unused)) void *arg)
{
	COMMON_CODE

	starpu_trace_user_event(0x42);

	GPU_GEMM('n', 'n', nxC, nyC, nyA, (TYPE)1.0, subA, ldA, subB, ldB,
					     (TYPE)0.0, subC, ldC);
	cublasStatus st;
	st = cublasGetError();
	if (st != CUBLAS_STATUS_SUCCESS)
		STARPU_ABORT();

	cudaThreadSynchronize();

	starpu_trace_user_event(0x42);
}
#endif

void STARPU_GEMM(cpu_mult)(void *descr[], __attribute__((unused))  void *arg)
{
	COMMON_CODE

	starpu_trace_user_event(0x42);
	CPU_GEMM("N", "N", nxC, nyC, nyA, (TYPE)1.0, subA, ldA, subB, ldB, (TYPE)0.0, subC, ldC);
	starpu_trace_user_event(0x43);
}