5 changed files with 78 additions and 12 deletions
@ -1,11 +1,56 @@ |
|||
#include "LinearTests.h" |
|||
#include <Common.h> |
|||
|
|||
Reduce::Reduce(cl::Context* context, cl::CommandQueue* queue, cl::Program* program) |
|||
Reduce::Reduce(cl::Context* _context, cl::CommandQueue* _queue, cl::Program* _program) |
|||
{ |
|||
//TODO: Implement
|
|||
context = _context; |
|||
queue = _queue; |
|||
program = _program; |
|||
} |
|||
|
|||
void Reduce::dewIt(int n, int m, float* y, const float* A, const float* x, const float* b) |
|||
{ |
|||
//TODO: Implement
|
|||
cl_int err = CL_SUCCESS; |
|||
cl::Event _event; |
|||
|
|||
cl::Kernel kernel = cl::Kernel(*program, "reduceMV", &err); |
|||
CheckCLError(err); |
|||
|
|||
cl::Buffer ABuffer(*context, CL_MEM_READ_ONLY, sizeof(float) * n * m, NULL, &err); |
|||
CheckCLError(err); |
|||
queue->enqueueWriteBuffer(ABuffer, true, 0, sizeof(float) * n * m, A); |
|||
|
|||
cl::Buffer XBuffer(*context, CL_MEM_READ_ONLY, sizeof(float) * m, NULL, &err); |
|||
CheckCLError(err); |
|||
queue->enqueueWriteBuffer(XBuffer, true, 0, sizeof(float) * m, x); |
|||
|
|||
cl::Buffer YBuffer(*context, CL_MEM_WRITE_ONLY, sizeof(float) * n, NULL, &err); |
|||
CheckCLError(err); |
|||
queue->enqueueWriteBuffer(YBuffer, true, 0, sizeof(float) * n, y); |
|||
|
|||
cl::Buffer BBuffer(*context, CL_MEM_READ_ONLY, sizeof(float) * n, NULL, &err); |
|||
CheckCLError(err); |
|||
queue->enqueueWriteBuffer(BBuffer, true, 0, sizeof(float) * n, b); |
|||
|
|||
|
|||
|
|||
|
|||
// void reduceMV
|
|||
kernel.setArg(0, n); // (const int n,
|
|||
kernel.setArg(1, m); // const int M,
|
|||
kernel.setArg(2, YBuffer); // __global float* y,
|
|||
kernel.setArg(3, ABuffer); // __global float* A,
|
|||
kernel.setArg(4, XBuffer); // __global float* x,
|
|||
kernel.setArg(5, BBuffer); // __global float* b,
|
|||
kernel.setArg(6, sizeof(float) * n * m, NULL); // __local float* Q)
|
|||
|
|||
queue->enqueueNDRangeKernel(kernel, |
|||
cl::NullRange, // Indexek nem eloffszetelve
|
|||
cl::NDRange(n, 1), // Minden elemet egy szál
|
|||
cl::NullRange, // Workgroup méret? - ez az auto, ha nem indul, 1024-re, onnan csökkent, amig elindul
|
|||
NULL, //
|
|||
&_event); |
|||
_event.wait(); |
|||
|
|||
queue->enqueueReadBuffer(YBuffer, true, 0, sizeof(float) * n, y); |
|||
} |
Loading…
Reference in new issue