diff options
| author | Gian Marco Iodice <gianmarco.iodice@arm.com> | 2020-05-28 10:22:03 +0100 |
|---|---|---|
| committer | Gian Marco Iodice <gianmarco.iodice@arm.com> | 2020-05-28 14:00:46 +0000 |
| commit | dd717c3150b813da403fbfd38e1200936998824a (patch) | |
| tree | 30c13f8cc428b03fae14999d18f8b4c2c1f21517 /arm_compute | |
| parent | 04c8e636c1083411e2f49fb0c80e2d1be498aa25 (diff) | |
| download | ComputeLibrary-dd717c3150b813da403fbfd38e1200936998824a.tar.gz | |
COMPMID-3318: Add flag to export to cl_image
- Added flag to export to cl_image in GEMMRHSMatrixInfo
- Returned an error in the GEMM/Lowp kernels without this support
Change-Id: I4a523d93c0984626bbf23e2efeb114f9c7c20a24
Signed-off-by: Gian Marco Iodice <gianmarco.iodice@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/3274
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'arm_compute')
| -rw-r--r-- | arm_compute/core/Types.h | 11 |
1 files changed, 6 insertions, 5 deletions
diff --git a/arm_compute/core/Types.h b/arm_compute/core/Types.h index 759ff07822..4e73edba4b 100644 --- a/arm_compute/core/Types.h +++ b/arm_compute/core/Types.h @@ -1916,11 +1916,12 @@ struct GEMMRHSMatrixInfo : n0(n), k0(k), h0(h), transpose(trans), interleave(inter) { } - unsigned int n0{ 1 }; /**< Number of columns processed by the matrix multiplication */ - unsigned int k0{ 1 }; /**< Number of partial accumulations performed by the matrix multiplication */ - unsigned int h0{ 1 }; /**< Number of horizontal blocks of size (k0xn0) stored on the same output row */ - bool transpose{ true }; /**< True if the (k0xn0) block has to be transposed before been stored */ - bool interleave{ true }; /**< True if the h0 (k0xn0) blocks have to be interleaved in the output row */ + unsigned int n0{ 1 }; /**< Number of columns processed by the matrix multiplication */ + unsigned int k0{ 1 }; /**< Number of partial accumulations performed by the matrix multiplication */ + unsigned int h0{ 1 }; /**< Number of horizontal blocks of size (k0xn0) stored on the same output row */ + bool transpose{ true }; /**< True if the (k0xn0) block has to be transposed before been stored */ + bool interleave{ true }; /**< True if the h0 (k0xn0) blocks have to be interleaved in the output row */ + bool export_to_cl_image{ false }; /**< True if the reshaped rhs has to be exported to cl_image. n0 must be equal to 4 */ }; /** GEMM information class. This class stores the necessary information to compute GEMM functions |