diff options
Diffstat (limited to 'reference_model/src/verify/verify_utils.cc')
| -rw-r--r-- | reference_model/src/verify/verify_utils.cc | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/reference_model/src/verify/verify_utils.cc b/reference_model/src/verify/verify_utils.cc index 786ab40..366238b 100644 --- a/reference_model/src/verify/verify_utils.cc +++ b/reference_model/src/verify/verify_utils.cc @@ -62,6 +62,12 @@ void from_json(const nlohmann::json& j, DotProductVerifyInfo& dotProductInfo) j.at("ks").get_to(dotProductInfo.ks); } +void from_json(const nlohmann::json& j, ReduceProductVerifyInfo& reduceProduceInfo) +{ + j.at("m").get_to(reduceProduceInfo.m); + j.at("n").get_to(reduceProduceInfo.n); +} + void from_json(const nlohmann::json& j, VerifyConfig& cfg) { j.at("mode").get_to(cfg.mode); @@ -74,6 +80,10 @@ void from_json(const nlohmann::json& j, VerifyConfig& cfg) { j.at("dot_product_info").get_to(cfg.dotProductInfo); } + if (j.contains("reduce_product_info")) + { + j.at("reduce_product_info").get_to(cfg.reduceProductInfo); + } } std::optional<VerifyConfig> parseVerifyConfig(const char* tensorName, const char* json) |