diff options
| author | Miao Wang <miaowang@google.com> | 2015-04-30 10:47:42 -0700 |
|---|---|---|
| committer | Miao Wang <miaowang@google.com> | 2015-04-30 13:37:38 -0700 |
| commit | 1e940d87c46edc672964c7674cff8da6441e9559 (patch) | |
| tree | 56a25ff984e96491360c6e7e21686eb714b83315 /rs | |
| parent | 2757fed2d612b7d7331e8ad9067df24b154ff8d7 (diff) | |
| download | frameworks_base-1e940d87c46edc672964c7674cff8da6441e9559.zip frameworks_base-1e940d87c46edc672964c7674cff8da6441e9559.tar.gz frameworks_base-1e940d87c46edc672964c7674cff8da6441e9559.tar.bz2 | |
[RenderScript] fix edges cases about Transpose for SYMM & SYR2K
Change-Id: Id86d5fe8475060a22f49b06b6e1f8f3eae7c7e9f
Diffstat (limited to 'rs')
| -rw-r--r-- | rs/java/android/renderscript/ScriptIntrinsicBLAS.java | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/rs/java/android/renderscript/ScriptIntrinsicBLAS.java b/rs/java/android/renderscript/ScriptIntrinsicBLAS.java index a387aab..149c0be 100644 --- a/rs/java/android/renderscript/ScriptIntrinsicBLAS.java +++ b/rs/java/android/renderscript/ScriptIntrinsicBLAS.java @@ -1012,7 +1012,7 @@ public final class ScriptIntrinsicBLAS extends ScriptIntrinsic { } } else { if (A != null) { - if (TransA != NO_TRANSPOSE) { + if (TransA == TRANSPOSE || TransA == CONJ_TRANSPOSE) { aN = A.getType().getY(); aM = A.getType().getX(); } else { @@ -1021,7 +1021,7 @@ public final class ScriptIntrinsicBLAS extends ScriptIntrinsic { } } if (B != null) { - if (TransB != NO_TRANSPOSE) { + if (TransB == TRANSPOSE || TransB == CONJ_TRANSPOSE) { bN = B.getType().getY(); bM = B.getType().getX(); } else { @@ -1272,7 +1272,7 @@ public final class ScriptIntrinsicBLAS extends ScriptIntrinsic { validateUplo(Uplo); validateSYR2K(Element.F32(mRS), Trans, A, B, C); int K = -1; - if (Trans == TRANSPOSE) { + if (Trans != NO_TRANSPOSE) { K = A.getType().getY(); } else { K = A.getType().getX(); @@ -1283,7 +1283,7 @@ public final class ScriptIntrinsicBLAS extends ScriptIntrinsic { validateUplo(Uplo); validateSYR2K(Element.F64(mRS), Trans, A, B, C); int K = -1; - if (Trans == TRANSPOSE) { + if (Trans != NO_TRANSPOSE) { K = A.getType().getY(); } else { K = A.getType().getX(); @@ -1294,7 +1294,7 @@ public final class ScriptIntrinsicBLAS extends ScriptIntrinsic { validateUplo(Uplo); validateSYR2K(Element.F32_2(mRS), Trans, A, B, C); int K = -1; - if (Trans == TRANSPOSE) { + if (Trans != NO_TRANSPOSE) { K = A.getType().getY(); } else { K = A.getType().getX(); @@ -1305,7 +1305,7 @@ public final class ScriptIntrinsicBLAS extends ScriptIntrinsic { validateUplo(Uplo); validateSYR2K(Element.F64_2(mRS), Trans, A, B, C); int K = -1; - if (Trans == TRANSPOSE) { + if (Trans != NO_TRANSPOSE) { K = A.getType().getY(); } else { K = A.getType().getX(); |