diff options
Diffstat (limited to 'reference_model/src/verify/verify_utils.h')
-rw-r--r-- | reference_model/src/verify/verify_utils.h | 10 |
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 |