Android/frameworks_base
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add BNNM intrinsic.

Browse Source 
Change-Id: I05fc9835786455e28d69f2b4ffe70c3292a01404
This commit is contained in:
Tim Murray
2015-04-01 11:07:16 -07:00
parent d1af38a478
commit 9cb16a2f91
3 changed files with 60 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
rs/jni/android_renderscript_RenderScript.cpp
View File

@@ -581,6 +581,32 @@ nScriptIntrinsicBLAS_Z(JNIEnv *_env, jobject _this, jlong con, jlong id, jint fu
}
static void
nScriptIntrinsicBLAS_BNNM(JNIEnv *_env, jobject _this, jlong con, jlong id, jint M, jint N, jint K,
jlong A, jint a_offset, jlong B, jint b_offset, jlong C, jint c_offset,
jint c_mult_int) {
RsBlasCall call;
memset(&call, 0, sizeof(call));
call.func = RsBlas_bnnm;
call.M = M;
call.N = N;
call.K = K;
call.a_offset = a_offset;
call.b_offset = b_offset;
call.c_offset = c_offset;
call.c_mult_int = c_mult_int;
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)
{
@@ -2425,6 +2451,8 @@ static JNINativeMethod methods[] = {
{"rsnScriptIntrinsicBLAS_Complex", "(JJIIIIIIIIIFFJJFFJIIII)V", (void*)nScriptIntrinsicBLAS_Complex },
{"rsnScriptIntrinsicBLAS_Z", "(JJIIIIIIIIIDDJJDDJIIII)V", (void*)nScriptIntrinsicBLAS_Z },
{"rsnScriptIntrinsicBLAS_BNNM", "(JJIIIJIJIJII)V", (void*)nScriptIntrinsicBLAS_BNNM },
{"rsnProgramStoreCreate", "(JZZZZZZIII)J", (void*)nProgramStoreCreate },
{"rsnProgramBindConstants", "(JJIJ)V", (void*)nProgramBindConstants },