gaspersic/triton
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

API: adding cuBLAS interface

Browse Source
This commit is contained in:
Philippe Tillet
2015-11-20 02:56:33 -05:00
parent da1b0a9571
commit c6333c993a
36 changed files with 17766 additions and 125 deletions
Download Patch File Download Diff File Expand all files Collapse all files

116
tests/cpp/operations/axpy.cpp
View File

@@ -2,8 +2,9 @@
#include <iostream>
#include "common.hpp"
#include "isaac/array.h"
#include "isaac/wrap/clBLAS.h"
#include "isaac/driver/common.h"
#include "clBLAS.h"
#include "cublas.h"
namespace sc = isaac;
typedef isaac::int_t int_t;
@@ -20,7 +21,7 @@ void test_impl(T epsilon, simple_vector_base<T> & cx, simple_vector_base<T>& cy,
sc::driver::CommandQueue queue = sc::driver::backend::queues::get(context,0);
int_t N = cz.size();
T aa = static_cast<T>(-4.3);
T aa = static_cast<T>(-4);
T bb = static_cast<T>(3.5);
isaac::value_scalar a(aa), b(bb);
isaac::scalar da(a, context), db(b, context);
@@ -49,65 +50,66 @@ void test_impl(T epsilon, simple_vector_base<T> & cx, simple_vector_base<T>& cy,
if(queue.device().backend()==sc::driver::OPENCL && interf==clBLAS)
{
cl_command_queue clqueue = queue.handle().cl();
RUN_TEST("AXPY", cz[i] = a*cx[i] + cz[i], BLAS<T>::F(clblasSaxpy, clblasDaxpy)(N, a, CHANDLE(x), OFF(x), INC(x), CHANDLE(z), OFF(z), INC(z), 1, &clqueue, 0, NULL, NULL));
RUN_TEST("COPY", cz[i] = cx[i], BLAS<T>::F(clblasScopy, clblasDcopy)(N, CHANDLE(x), OFF(x), INC(x), CHANDLE(z), OFF(z), INC(z), 1, &clqueue, 0, NULL, NULL));
RUN_TEST("SCAL", cz[i] = a*cz[i], BLAS<T>::F(clblasSscal, clblasDscal)(N, a, CHANDLE(z), OFF(z), INC(z), 1, &clqueue, 0, NULL, NULL));
}
if(queue.device().backend()==sc::driver::CUDA && interf == cuBLAS)
{
RUN_TEST("AXPY", cz[i] = a*cx[i] + cz[i], BLAS<T>::F(cublasSaxpy, cublasDaxpy)(N, a, (T*)CUHANDLE(x) + OFF(x), INC(x), (T*)CUHANDLE(z) + OFF(z), INC(z)));
RUN_TEST("COPY", cz[i] = cx[i], BLAS<T>::F(cublasScopy, cublasDcopy)(N, (T*)CUHANDLE(x) + OFF(x), INC(x), (T*)CUHANDLE(z) + OFF(z), INC(z)));
RUN_TEST("SCAL", cz[i] = a*cz[i], BLAS<T>::F(cublasSscal, cublasDscal)(N, a, (T*)CUHANDLE(z) + OFF(z), INC(z)));
}
if(interf == CPP)
{
RUN_TEST("z = 0", cz[i] = 0, z = zeros(N, 1, dtype, context))
RUN_TEST("z = x", cz[i] = cx[i], z = x)
RUN_TEST("z = -x", cz[i] = -cx[i], z = -x)
RUN_TEST("AXPY", cz[i] = a*cx[i] + cz[i], BLAS<T>::F(clblasSaxpy, clblasDaxpy)(N, a, CHANDLE(x), OFF(x), INC(x),
CHANDLE(z), OFF(z), INC(z),
1, &clqueue, 0, NULL, NULL));
RUN_TEST("z = x + y", cz[i] = cx[i] + cy[i], z = x + y)
RUN_TEST("z = x - y", cz[i] = cx[i] - cy[i], z = x - y)
RUN_TEST("z = x + y + z", cz[i] = cx[i] + cy[i] + cz[i], z = x + y + z)
RUN_TEST("COPY", cz[i] = cx[i], BLAS<T>::F(clblasScopy, clblasDcopy)(N, CHANDLE(x), OFF(x), INC(x),
CHANDLE(z), OFF(z), INC(z),
1, &clqueue, 0, NULL, NULL));
RUN_TEST("z = a*x", cz[i] = aa*cx[i], z = a*x)
RUN_TEST("z = da*x", cz[i] = aa*cx[i], z = da*x)
RUN_TEST("z = a*x + b*y", cz[i] = aa*cx[i] + bb*cy[i], z= a*x + b*y)
RUN_TEST("z = da*x + b*y", cz[i] = aa*cx[i] + bb*cy[i], z= da*x + b*y)
RUN_TEST("z = a*x + db*y", cz[i] = aa*cx[i] + bb*cy[i], z= a*x + db*y)
RUN_TEST("z = da*x + db*y", cz[i] = aa*cx[i] + bb*cy[i], z= da*x + db*y)
RUN_TEST("SCAL", cz[i] = a*cz[i], BLAS<T>::F(clblasSscal, clblasDscal)(N, a, CHANDLE(z), OFF(z), INC(z),
1, &clqueue, 0, NULL, NULL));
RUN_TEST("z = exp(x)", cz[i] = exp(cx[i]), z= exp(x))
RUN_TEST("z = abs(x)", cz[i] = abs(cx[i]), z= abs(x))
RUN_TEST("z = acos(x)", cz[i] = acos(cx[i]), z= acos(x))
RUN_TEST("z = asin(x)", cz[i] = asin(cx[i]), z= asin(x))
RUN_TEST("z = atan(x)", cz[i] = atan(cx[i]), z= atan(x))
RUN_TEST("z = ceil(x)", cz[i] = ceil(cx[i]), z= ceil(x))
RUN_TEST("z = cos(x)", cz[i] = cos(cx[i]), z= cos(x))
RUN_TEST("z = cosh(x)", cz[i] = cosh(cx[i]), z= cosh(x))
RUN_TEST("z = floor(x)", cz[i] = floor(cx[i]), z= floor(x))
RUN_TEST("z = log(x)", cz[i] = log(cx[i]), z= log(x))
RUN_TEST("z = log10(x)", cz[i] = log10(cx[i]), z= log10(x))
RUN_TEST("z = sin(x)", cz[i] = sin(cx[i]), z= sin(x))
RUN_TEST("z = sinh(x)", cz[i] = sinh(cx[i]), z= sinh(x))
RUN_TEST("z = sqrt(x)", cz[i] = sqrt(cx[i]), z= sqrt(x))
RUN_TEST("z = tan(x)", cz[i] = tan(cx[i]), z= tan(x))
RUN_TEST("z = tanh(x)", cz[i] = tanh(cx[i]), z= tanh(x))
RUN_TEST("z = x.*y", cz[i] = cx[i]*cy[i], z= x*y)
RUN_TEST("z = x./y", cz[i] = cx[i]/cy[i], z= x/y)
RUN_TEST("z = pow(x,y)", cz[i] = pow(cx[i], cy[i]), z= pow(x,y))
#undef CONVERT
#define CONVERT for(int_t i = 0 ; i < N ; ++i) {cz[i] = !!cz[i] ; buffer[i] = !!buffer[i];}
RUN_TEST("z = x==y", cz[i] = cx[i]==cy[i], z= cast(x==y, dtype))
RUN_TEST("z = x>=y", cz[i] = cx[i]>=cy[i], z= cast(x>=y, dtype))
RUN_TEST("z = x>y", cz[i] = cx[i]>cy[i], z= cast(x>y, dtype))
RUN_TEST("z = x<=y", cz[i] = cx[i]<=cy[i], z= cast(x<=y, dtype))
RUN_TEST("z = x<y", cz[i] = cx[i]<cy[i], z= cast(x<y, dtype))
RUN_TEST("z = x!=y", cz[i] = cx[i]!=cy[i], z= cast(x!=y, dtype))
#undef RUN_TEST
}
RUN_TEST("z = 0", cz[i] = 0, z = zeros(N, 1, dtype, context))
RUN_TEST("z = x", cz[i] = cx[i], z = x)
RUN_TEST("z = -x", cz[i] = -cx[i], z = -x)
RUN_TEST("z = x + y", cz[i] = cx[i] + cy[i], z = x + y)
RUN_TEST("z = x - y", cz[i] = cx[i] - cy[i], z = x - y)
RUN_TEST("z = x + y + z", cz[i] = cx[i] + cy[i] + cz[i], z = x + y + z)
RUN_TEST("z = a*x", cz[i] = aa*cx[i], z = a*x)
RUN_TEST("z = da*x", cz[i] = aa*cx[i], z = da*x)
RUN_TEST("z = a*x + b*y", cz[i] = aa*cx[i] + bb*cy[i], z= a*x + b*y)
RUN_TEST("z = da*x + b*y", cz[i] = aa*cx[i] + bb*cy[i], z= da*x + b*y)
RUN_TEST("z = a*x + db*y", cz[i] = aa*cx[i] + bb*cy[i], z= a*x + db*y)
RUN_TEST("z = da*x + db*y", cz[i] = aa*cx[i] + bb*cy[i], z= da*x + db*y)
RUN_TEST("z = exp(x)", cz[i] = exp(cx[i]), z= exp(x))
RUN_TEST("z = abs(x)", cz[i] = abs(cx[i]), z= abs(x))
RUN_TEST("z = acos(x)", cz[i] = acos(cx[i]), z= acos(x))
RUN_TEST("z = asin(x)", cz[i] = asin(cx[i]), z= asin(x))
RUN_TEST("z = atan(x)", cz[i] = atan(cx[i]), z= atan(x))
RUN_TEST("z = ceil(x)", cz[i] = ceil(cx[i]), z= ceil(x))
RUN_TEST("z = cos(x)", cz[i] = cos(cx[i]), z= cos(x))
RUN_TEST("z = cosh(x)", cz[i] = cosh(cx[i]), z= cosh(x))
RUN_TEST("z = floor(x)", cz[i] = floor(cx[i]), z= floor(x))
RUN_TEST("z = log(x)", cz[i] = log(cx[i]), z= log(x))
RUN_TEST("z = log10(x)", cz[i] = log10(cx[i]), z= log10(x))
RUN_TEST("z = sin(x)", cz[i] = sin(cx[i]), z= sin(x))
RUN_TEST("z = sinh(x)", cz[i] = sinh(cx[i]), z= sinh(x))
RUN_TEST("z = sqrt(x)", cz[i] = sqrt(cx[i]), z= sqrt(x))
RUN_TEST("z = tan(x)", cz[i] = tan(cx[i]), z= tan(x))
RUN_TEST("z = tanh(x)", cz[i] = tanh(cx[i]), z= tanh(x))
RUN_TEST("z = x.*y", cz[i] = cx[i]*cy[i], z= x*y)
RUN_TEST("z = x./y", cz[i] = cx[i]/cy[i], z= x/y)
RUN_TEST("z = pow(x,y)", cz[i] = pow(cx[i], cy[i]), z= pow(x,y))
#undef CONVERT
#define CONVERT for(int_t i = 0 ; i < N ; ++i) {cz[i] = !!cz[i] ; buffer[i] = !!buffer[i];}
RUN_TEST("z = x==y", cz[i] = cx[i]==cy[i], z= cast(x==y, dtype))
RUN_TEST("z = x>=y", cz[i] = cx[i]>=cy[i], z= cast(x>=y, dtype))
RUN_TEST("z = x>y", cz[i] = cx[i]>cy[i], z= cast(x>y, dtype))
RUN_TEST("z = x<=y", cz[i] = cx[i]<=cy[i], z= cast(x<=y, dtype))
RUN_TEST("z = x<y", cz[i] = cx[i]<cy[i], z= cast(x<y, dtype))
RUN_TEST("z = x!=y", cz[i] = cx[i]!=cy[i], z= cast(x!=y, dtype))
#undef RUN_TEST
if(failure_count > 0)
@@ -126,9 +128,11 @@ void test(T epsilon, sc::driver::Context const & ctx)
std::cout << "> standard..." << std::endl;
test_impl(epsilon, cx, cy, cz, x, y, z, clBLAS);
test_impl(epsilon, cx, cy, cz, x, y, z, cuBLAS);
test_impl(epsilon, cx, cy, cz, x, y, z, CPP);
std::cout << "> slice..." << std::endl;
test_impl(epsilon, cx_s, cy_s, cz_s, x_s, y_s, z_s, clBLAS);
test_impl(epsilon, cx_s, cy_s, cz_s, x_s, y_s, z_s, cuBLAS);
test_impl(epsilon, cx_s, cy_s, cz_s, x_s, y_s, z_s, CPP);
}