30 lines
557 B
Common Lisp
30 lines
557 B
Common Lisp
|
|
||
|
|
|
||
|
|
|
||
|
|
__kernel void MatrMltMatr(__global double* A, __global double* B, __global double* C, __global int* gW)
|
||
|
|
{
|
||
|
|
int cX = get_global_id(0);
|
||
|
|
int cY = get_global_id(1);
|
||
|
|
int W = *gW;
|
||
|
|
|
||
|
|
double sum = 0.0;
|
||
|
|
for (int i=0; i<W; i++)
|
||
|
|
sum += A[cX + i*W] * B[i + cY*W];
|
||
|
|
|
||
|
|
C[cX + cY*W] = sum;
|
||
|
|
}
|
||
|
|
|
||
|
|
__kernel void MatrMltVec(__global double* C, __global double* V, __global double* V2, __global int* gW)
|
||
|
|
{
|
||
|
|
int i = get_global_id(0);
|
||
|
|
int W = *gW;
|
||
|
|
|
||
|
|
double sum = 0.0;
|
||
|
|
for (int j=0; j<W; j++)
|
||
|
|
sum += C[j + i*W] * V[j];
|
||
|
|
|
||
|
|
V2[i] = sum;
|
||
|
|
}
|
||
|
|
|
||
|
|
|