Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit 41c93ffe authored by Tim Murray's avatar Tim Murray Committed by Gerrit Code Review
Browse files

Merge "Add BLAS intrinsic."

parents 627396ac 25207df6

rs/java/android/renderscript/RenderScript.java

+49 −0
Original line number Diff line number Diff line
@@ -887,6 +887,55 @@ public class RenderScript { 
        return rsnPathCreate(mContext, prim, isStatic, vtx, loop, q);

    }



    native void rsnScriptIntrinsicBLAS_Single(long con, long id, int func, int TransA,

                                              int TransB, int Side, int Uplo, int Diag, int M, int N, int K,

                                              float alpha, long A, long B, float beta, long C, int incX, int incY,

                                              int KL, int KU);

    synchronized void nScriptIntrinsicBLAS_Single(long id, int func, int TransA,

                                                  int TransB, int Side, int Uplo, int Diag, int M, int N, int K,

                                                  float alpha, long A, long B, float beta, long C, int incX, int incY,

                                                  int KL, int KU) {

        validate();

        rsnScriptIntrinsicBLAS_Single(mContext, id, func, TransA, TransB, Side, Uplo, Diag, M, N, K, alpha, A, B, beta, C, incX, incY, KL, KU);

    }



    native void rsnScriptIntrinsicBLAS_Double(long con, long id, int func, int TransA,

                                              int TransB, int Side, int Uplo, int Diag, int M, int N, int K,

                                              double alpha, long A, long B, double beta, long C, int incX, int incY,

                                              int KL, int KU);

    synchronized void nScriptIntrinsicBLAS_Double(long id, int func, int TransA,

                                                  int TransB, int Side, int Uplo, int Diag, int M, int N, int K,

                                                  double alpha, long A, long B, double beta, long C, int incX, int incY,

                                                  int KL, int KU) {

        validate();

        rsnScriptIntrinsicBLAS_Double(mContext, id, func, TransA, TransB, Side, Uplo, Diag, M, N, K, alpha, A, B, beta, C, incX, incY, KL, KU);

    }



    native void rsnScriptIntrinsicBLAS_Complex(long con, long id, int func, int TransA,

                                               int TransB, int Side, int Uplo, int Diag, int M, int N, int K,

                                               float alphaX, float alphaY, long A, long B, float betaX, float betaY, long C, int incX, int incY,

                                               int KL, int KU);

    synchronized void nScriptIntrinsicBLAS_Complex(long id, int func, int TransA,

                                                   int TransB, int Side, int Uplo, int Diag, int M, int N, int K,

                                                   float alphaX, float alphaY, long A, long B, float betaX, float betaY, long C, int incX, int incY,

                                                   int KL, int KU) {

        validate();

        rsnScriptIntrinsicBLAS_Complex(mContext, id, func, TransA, TransB, Side, Uplo, Diag, M, N, K, alphaX, alphaY, A, B, betaX, betaY, C, incX, incY, KL, KU);

    }



    native void rsnScriptIntrinsicBLAS_Z(long con, long id, int func, int TransA,

                                         int TransB, int Side, int Uplo, int Diag, int M, int N, int K,

                                         double alphaX, double alphaY, long A, long B, double betaX, double betaY, long C, int incX, int incY,

                                         int KL, int KU);

    synchronized void nScriptIntrinsicBLAS_Z(long id, int func, int TransA,

                                             int TransB, int Side, int Uplo, int Diag, int M, int N, int K,

                                             double alphaX, double alphaY, long A, long B, double betaX, double betaY, long C, int incX, int incY,

                                             int KL, int KU) {

        validate();

        rsnScriptIntrinsicBLAS_Z(mContext, id, func, TransA, TransB, Side, Uplo, Diag, M, N, K, alphaX, alphaY, A, B, betaX, betaY, C, incX, incY, KL, KU);

    }





    long     mDev;

    long     mContext;

    @SuppressWarnings({"FieldCanBeLocal"})

rs/java/android/renderscript/ScriptIntrinsicBLAS.java

0 → 100644
+1489 −0

File added.

Preview size limit exceeded, changes collapsed.

rs/jni/android_renderscript_RenderScript.cpp

+142 −0
Original line number Diff line number Diff line
@@ -309,6 +309,143 @@ nScriptGroup2Execute(JNIEnv *_env, jobject _this, jlong con, jlong groupID) { 
  rsScriptGroupExecute((RsContext)con, (RsScriptGroup2)groupID);

}



static void

nScriptIntrinsicBLAS_Single(JNIEnv *_env, jobject _this, jlong con, jlong id, jint func, jint TransA,

                            jint TransB, jint Side, jint Uplo, jint Diag, jint M, jint N, jint K,

                            jfloat alpha, jlong A, jlong B, jfloat beta, jlong C, jint incX, jint incY,

                            jint KL, jint KU) {

    RsBlasCall call;

    memset(&call, 0, sizeof(call));

    call.func = (RsBlasFunction)func;

    call.transA = (RsBlasTranspose)TransA;

    call.transB = (RsBlasTranspose)TransB;

    call.side = (RsBlasSide)Side;

    call.uplo = (RsBlasUplo)Uplo;

    call.diag = (RsBlasDiag)Diag;

    call.M = M;

    call.N = N;

    call.K = K;

    call.alpha.f = alpha;

    call.beta.f = beta;

    call.incX = incX;

    call.incY = incY;

    call.KL = KL;

    call.KU = KU;



    RsAllocation in_allocs[3];

    in_allocs[0] = (RsAllocation)A;

    in_allocs[1] = (RsAllocation)B;

    in_allocs[2] = (RsAllocation)C;



    rsScriptForEachMulti((RsContext)con, (RsScript)id, 0,

                         in_allocs, sizeof(in_allocs), nullptr,

                         &call, sizeof(call), nullptr, 0);

}



static void

nScriptIntrinsicBLAS_Double(JNIEnv *_env, jobject _this, jlong con, jlong id, jint func, jint TransA,

                            jint TransB, jint Side, jint Uplo, jint Diag, jint M, jint N, jint K,

                            jdouble alpha, jlong A, jlong B, jdouble beta, jlong C, jint incX, jint incY,

                            jint KL, jint KU) {

    RsBlasCall call;

    memset(&call, 0, sizeof(call));

    call.func = (RsBlasFunction)func;

    call.transA = (RsBlasTranspose)TransA;

    call.transB = (RsBlasTranspose)TransB;

    call.side = (RsBlasSide)Side;

    call.uplo = (RsBlasUplo)Uplo;

    call.diag = (RsBlasDiag)Diag;

    call.M = M;

    call.N = N;

    call.K = K;

    call.alpha.d = alpha;

    call.beta.d = beta;

    call.incX = incX;

    call.incY = incY;

    call.KL = KL;

    call.KU = KU;



    RsAllocation in_allocs[3];

    in_allocs[0] = (RsAllocation)A;

    in_allocs[1] = (RsAllocation)B;

    in_allocs[2] = (RsAllocation)C;



    rsScriptForEachMulti((RsContext)con, (RsScript)id, 0,

                         in_allocs, sizeof(in_allocs), nullptr,

                         &call, sizeof(call), nullptr, 0);

}



static void

nScriptIntrinsicBLAS_Complex(JNIEnv *_env, jobject _this, jlong con, jlong id, jint func, jint TransA,

                             jint TransB, jint Side, jint Uplo, jint Diag, jint M, jint N, jint K,

                             jfloat alphaX, jfloat alphaY, jlong A, jlong B, jfloat betaX,

                             jfloat betaY, jlong C, jint incX, jint incY, jint KL, jint KU) {

    RsBlasCall call;

    memset(&call, 0, sizeof(call));

    call.func = (RsBlasFunction)func;

    call.transA = (RsBlasTranspose)TransA;

    call.transB = (RsBlasTranspose)TransB;

    call.side = (RsBlasSide)Side;

    call.uplo = (RsBlasUplo)Uplo;

    call.diag = (RsBlasDiag)Diag;

    call.M = M;

    call.N = N;

    call.K = K;

    call.alpha.c.r = alphaX;

    call.alpha.c.i = alphaY;

    call.beta.c.r = betaX;

    call.beta.c.r = betaY;

    call.incX = incX;

    call.incY = incY;

    call.KL = KL;

    call.KU = KU;



    RsAllocation in_allocs[3];

    in_allocs[0] = (RsAllocation)A;

    in_allocs[1] = (RsAllocation)B;

    in_allocs[2] = (RsAllocation)C;



    rsScriptForEachMulti((RsContext)con, (RsScript)id, 0,

                         in_allocs, sizeof(in_allocs), nullptr,

                         &call, sizeof(call), nullptr, 0);

}



static void

nScriptIntrinsicBLAS_Z(JNIEnv *_env, jobject _this, jlong con, jlong id, jint func, jint TransA,

                       jint TransB, jint Side, jint Uplo, jint Diag, jint M, jint N, jint K,

                       jdouble alphaX, jdouble alphaY, jlong A, jlong B, jdouble betaX,

                       jdouble betaY, jlong C, jint incX, jint incY, jint KL, jint KU) {

    RsBlasCall call;

    memset(&call, 0, sizeof(call));

    call.func = (RsBlasFunction)func;

    call.transA = (RsBlasTranspose)TransA;

    call.transB = (RsBlasTranspose)TransB;

    call.side = (RsBlasSide)Side;

    call.uplo = (RsBlasUplo)Uplo;

    call.diag = (RsBlasDiag)Diag;

    call.M = M;

    call.N = N;

    call.K = K;

    call.alpha.z.r = alphaX;

    call.alpha.z.i = alphaY;

    call.beta.z.r = betaX;

    call.beta.z.r = betaY;

    call.incX = incX;

    call.incY = incY;

    call.KL = KL;

    call.KU = KU;



    RsAllocation in_allocs[3];

    in_allocs[0] = (RsAllocation)A;

    in_allocs[1] = (RsAllocation)B;

    in_allocs[2] = (RsAllocation)C;



    rsScriptForEachMulti((RsContext)con, (RsScript)id, 0,

                         in_allocs, sizeof(in_allocs), nullptr,

                         &call, sizeof(call), nullptr, 0);

}





static void

nAssignName(JNIEnv *_env, jobject _this, jlong con, jlong obj, jbyteArray str)

{
@@ -2094,6 +2231,11 @@ static JNINativeMethod methods[] = { 
{"rsnScriptGroupExecute",            "(JJ)V",                                 (void*)nScriptGroupExecute },

{"rsnScriptGroup2Execute",           "(JJ)V",                                 (void*)nScriptGroup2Execute },



{"rsnScriptIntrinsicBLAS_Single",    "(JJIIIIIIIIIFJJFJIIII)V",               (void*)nScriptIntrinsicBLAS_Single },

{"rsnScriptIntrinsicBLAS_Double",    "(JJIIIIIIIIIDJJDJIIII)V",               (void*)nScriptIntrinsicBLAS_Double },

{"rsnScriptIntrinsicBLAS_Complex",   "(JJIIIIIIIIIFFJJFFJIIII)V",             (void*)nScriptIntrinsicBLAS_Complex },

{"rsnScriptIntrinsicBLAS_Z",         "(JJIIIIIIIIIDDJJDDJIIII)V",             (void*)nScriptIntrinsicBLAS_Z },



{"rsnProgramStoreCreate",            "(JZZZZZZIII)J",                         (void*)nProgramStoreCreate },



{"rsnProgramBindConstants",          "(JJIJ)V",                               (void*)nProgramBindConstants },