From e1cf095071c27e4934522bf42043528ec1f925f4 Mon Sep 17 00:00:00 2001 From: Miao Wang Date: Thu, 30 Apr 2015 10:47:42 -0700 Subject: [PATCH] [RenderScript] fix edges cases about Transpose for SYMM & SYR2K Change-Id: Id86d5fe8475060a22f49b06b6e1f8f3eae7c7e9f --- .../android/renderscript/ScriptIntrinsicBLAS.java | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/rs/java/android/renderscript/ScriptIntrinsicBLAS.java b/rs/java/android/renderscript/ScriptIntrinsicBLAS.java index feca951d2d85a..d2adca0bb5cbc 100644 --- a/rs/java/android/renderscript/ScriptIntrinsicBLAS.java +++ b/rs/java/android/renderscript/ScriptIntrinsicBLAS.java @@ -1013,7 +1013,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 { @@ -1022,7 +1022,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 { @@ -1273,7 +1273,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(); @@ -1284,7 +1284,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(); @@ -1295,7 +1295,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(); @@ -1306,7 +1306,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();