diff options
author | morgolock <pablo.tello@arm.com> | 2020-05-05 16:28:19 +0100 |
---|---|---|
committer | Georgios Pinitas <georgios.pinitas@arm.com> | 2020-05-15 13:18:28 +0000 |
commit | aba2f912a21487776e540724cf354a6cd8d89829 (patch) | |
tree | d53c28c746aa0e9eb368f0edbf9f0afbec7a840b /utils/TypePrinter.h | |
parent | 3155f77d23eb332974a6b9cb5d37a1329b8493a6 (diff) | |
download | ComputeLibrary-aba2f912a21487776e540724cf354a6cd8d89829.tar.gz |
COMPMID-3289: Test improvement CLGEMMMatrixMultiplyReshapedKernel.
Change-Id: Ia6c2f115849889baceafaf716477456e41f96037
Signed-off-by: morgolock <pablo.tello@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/3186
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Sang-Hoon Park <sang-hoon.park@arm.com>
Diffstat (limited to 'utils/TypePrinter.h')
-rw-r--r-- | utils/TypePrinter.h | 90 |
1 files changed, 90 insertions, 0 deletions
diff --git a/utils/TypePrinter.h b/utils/TypePrinter.h index 1f499dec7e..55e34c8541 100644 --- a/utils/TypePrinter.h +++ b/utils/TypePrinter.h @@ -29,6 +29,7 @@ #include "arm_compute/core/Error.h" #include "arm_compute/core/GPUTarget.h" #include "arm_compute/core/HOGInfo.h" +#include "arm_compute/core/KernelDescriptors.h" #include "arm_compute/core/Size2D.h" #include "arm_compute/core/Strides.h" #include "arm_compute/core/TensorInfo.h" @@ -235,6 +236,95 @@ inline std::string to_string(const ROIPoolingLayerInfo &pool_info) return str.str(); } +/** Formatted output of the GEMMKernelInfo type. + * + * @param[out] os Output stream. + * @param[in] gemm_info Type to output. + * + * @return Modified output stream. + */ +inline ::std::ostream &operator<<(::std::ostream &os, const GEMMKernelInfo &gemm_info) +{ + os << "( m= " << gemm_info.m; + os << " n= " << gemm_info.n; + os << " k= " << gemm_info.k; + os << " depth_output_gemm3d= " << gemm_info.depth_output_gemm3d; + os << " reinterpret_input_as_3d= " << gemm_info.reinterpret_input_as_3d; + os << " broadcast_bias= " << gemm_info.broadcast_bias; + os << " fp_mixed_precision= " << gemm_info.fp_mixed_precision; + os << " mult_transpose1xW_width= " << gemm_info.mult_transpose1xW_width; + os << " mult_interleave4x4_height= " << gemm_info.mult_interleave4x4_height; + os << " a_offset = " << gemm_info.a_offset; + os << " b_offset = " << gemm_info.b_offset; + os << ")"; + return os; +} + +/** Formatted output of the GEMMLHSMatrixInfo type. + * + * @param[out] os Output stream. + * @param[in] gemm_info Type to output. + * + * @return Modified output stream. + */ +inline ::std::ostream &operator<<(::std::ostream &os, const GEMMLHSMatrixInfo &gemm_info) +{ + os << "( m0= " << (unsigned int)gemm_info.m0 << " k0= " << gemm_info.k0 << " v0= " << gemm_info.v0 << " trans= " << gemm_info.transpose << " inter= " << gemm_info.interleave << "})"; + return os; +} + +/** Formatted output of the GEMMRHSMatrixInfo type. + * + * @param[out] os Output stream. + * @param[in] gemm_info Type to output. + * + * @return Modified output stream. + */ +inline ::std::ostream &operator<<(::std::ostream &os, const GEMMRHSMatrixInfo &gemm_info) +{ + os << "( n0= " << (unsigned int)gemm_info.n0 << " k0= " << gemm_info.k0 << " h0= " << gemm_info.h0 << " trans= " << gemm_info.transpose << " inter= " << gemm_info.interleave << "})"; + return os; +} + +/** Formatted output of the GEMMRHSMatrixInfo type. + * + * @param[in] gemm_info GEMMRHSMatrixInfo to output. + * + * @return Formatted string. + */ +inline std::string to_string(const GEMMRHSMatrixInfo &gemm_info) +{ + std::stringstream str; + str << gemm_info; + return str.str(); +} + +/** Formatted output of the GEMMLHSMatrixInfo type. + * + * @param[in] gemm_info GEMMLHSMatrixInfo to output. + * + * @return Formatted string. + */ +inline std::string to_string(const GEMMLHSMatrixInfo &gemm_info) +{ + std::stringstream str; + str << gemm_info; + return str.str(); +} + +/** Formatted output of the GEMMKernelInfo type. + * + * @param[in] gemm_info GEMMKernelInfo Type to output. + * + * @return Formatted string. + */ +inline std::string to_string(const GEMMKernelInfo &gemm_info) +{ + std::stringstream str; + str << gemm_info; + return str.str(); +} + /** Formatted output of the BoundingBoxTransformInfo type. * * @param[out] os Output stream. |