_gpu_kernel_mtx_mul.c File Reference

Go to the source code of this file.

Functions
__kernel void	matrixMul (__global float *C, __global float *A, __global float *B, int wA, int wB)

Function Documentation

matrixMul()

__kernel void matrixMul	(	__global float *	C,
		__global float *	A,
		__global float *	B,
		int	wA,
		int	wB
	)

OpenCL C GPU kernel

Definition at line 4 of file _gpu_kernel_mtx_mul.c.

                                {
 
    int tx = get_global_id(0);
    int ty = get_global_id(1);
 
    // value stores the element that is
    // computed by the thread
    float value = 0;
    for (int k = 0; k < wA; ++k) {
        float elementA = A[ty * wA + k];
        float elementB = B[k * wB + tx];
        value += elementA * elementB;
    }
 
    // Write the matrix to device memory each
    // thread writes one element
    C[ty * wA + tx] = value;
}

References python.linalg::A, python.linalg::B, and python.linalg::C.