aboutsummaryrefslogtreecommitdiff
path: root/reference_model/src/verify/verify_utils.h
diff options
context:
space:
mode:
Diffstat (limited to 'reference_model/src/verify/verify_utils.h')
-rw-r--r--reference_model/src/verify/verify_utils.h10
1 files changed, 10 insertions, 0 deletions
diff --git a/reference_model/src/verify/verify_utils.h b/reference_model/src/verify/verify_utils.h
index 5b98f5c..0afd804 100644
--- a/reference_model/src/verify/verify_utils.h
+++ b/reference_model/src/verify/verify_utils.h
@@ -64,6 +64,15 @@ struct DotProductVerifyInfo
int32_t ks;
};
+/// \brief reduce-product verification meta-data
+struct ReduceProductVerifyInfo
+{
+ ReduceProductVerifyInfo() = default;
+
+ int64_t m;
+ int64_t n;
+};
+
/// \brief Verification meta-data
struct VerifyConfig
{
@@ -73,6 +82,7 @@ struct VerifyConfig
DType dataType;
UlpInfo ulpInfo;
DotProductVerifyInfo dotProductInfo;
+ ReduceProductVerifyInfo reduceProductInfo;
};
/// \brief Parse the verification config for a tensor when given in JSON form