aboutsummaryrefslogtreecommitdiff
path: root/arm_compute/core
diff options
context:
space:
mode:
authorMohammed Suhail Munshi <MohammedSuhail.Munshi@arm.com>2023-02-16 16:22:32 +0000
committerMohmun02 <MohammedSuhail.Munshi@arm.com>2023-02-28 11:33:15 +0000
commitd538d161f6e55290bf1bb0b27cf26666379d0255 (patch)
tree82e171e42f4091a4fb168b2a79d967dbcd182c23 /arm_compute/core
parent13321f7607410cc0150ac15407eea9378dcbdfc8 (diff)
downloadComputeLibrary-d538d161f6e55290bf1bb0b27cf26666379d0255.tar.gz
Add MatMulInfo Information class to ComputeLibrary
Partially resolves : [COMPMID-5930] Signed-off-by: Mohammed Suhail Munshi <MohammedSuhail.Munshi@arm.com> Change-Id: Icb6c8a9d1b5a2c5c57e37cb7c877414ed500d0cc Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/c/VisualCompute/ComputeLibrary/+/494758 Tested-by: bsgcomp <bsgcomp@arm.com> Comments-Addressed: bsgcomp <bsgcomp@arm.com> Reviewed-by: Sicong Li <sicong.li@arm.com> Reviewed-by: Jakub Sujak <jakub.sujak@arm.com> Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/9181 Tested-by: Arm Jenkins <bsgcomp@arm.com> Reviewed-by: SiCong Li <sicong.li@arm.com> Comments-Addressed: Arm Jenkins <bsgcomp@arm.com> Benchmark: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'arm_compute/core')
-rw-r--r--arm_compute/core/Types.h46
1 files changed, 45 insertions, 1 deletions
diff --git a/arm_compute/core/Types.h b/arm_compute/core/Types.h
index 24ef931e88..71ec926483 100644
--- a/arm_compute/core/Types.h
+++ b/arm_compute/core/Types.h
@@ -1,5 +1,5 @@
/*
- * Copyright (c) 2016-2022 Arm Limited.
+ * Copyright (c) 2016-2023 Arm Limited.
*
* SPDX-License-Identifier: MIT
*
@@ -2694,5 +2694,49 @@ struct IOFormatInfo
/** Align columns */
bool align_columns;
};
+
+/** Class for holding information related to matrix multiplication function
+ */
+class MatMulInfo
+{
+public:
+ /* Get Adjoint LHS flag value */
+ bool adj_lhs() const
+ {
+ return _adj_lhs;
+ }
+ /* Get Adjoint RHS flag value */
+ bool adj_rhs() const
+ {
+ return _adj_rhs;
+ }
+ /* Get Fused Activation Layer Info */
+ ActivationLayerInfo fused_activation() const
+ {
+ return _fused_act;
+ }
+ /* Set Adjoint LHS flag */
+ MatMulInfo& adj_lhs(bool adj_lhs)
+ {
+ _adj_lhs = adj_lhs;
+ return *this;
+ }
+ /* Set Adjoint RHS flag */
+ MatMulInfo& adj_rhs(bool adj_rhs)
+ {
+ _adj_rhs = adj_rhs;
+ return *this;
+ }
+ /* Set Fused Activation Layer Info */
+ MatMulInfo& fused_activation(const ActivationLayerInfo& act_info)
+ {
+ _fused_act = act_info;
+ return *this;
+ }
+private:
+ bool _adj_lhs{false};
+ bool _adj_rhs{false};
+ ActivationLayerInfo _fused_act{}; // disabled by default
+};
} // namespace arm_compute
#endif /* ARM_COMPUTE_TYPES_H */