tkonolige commented on a change in pull request #8503:
URL: https://github.com/apache/tvm/pull/8503#discussion_r673284379
##########
File path: src/runtime/profiling.cc
##########
@@ -228,6 +228,71 @@ String ReportNode::AsCSV() const {
return s.str();
}
+namespace {
+void print_metric(std::ostream& os, ObjectRef o) {
+ if (o.as<StringObj>()) {
+ os << "\"" << Downcast<String>(o) << "\"";
+ } else if (const CountNode* n = o.as<CountNode>()) {
+ os << "{\"count\":" << std::to_string(n->value) << "}";
+ } else if (const DurationNode* n = o.as<DurationNode>()) {
+ os << "{\"microseconds\":" << std::to_string(n->microseconds) << "}";
+ } else if (const PercentNode* n = o.as<PercentNode>()) {
+ os << "{\"percent\":" << std::to_string(n->percent) << "}";
+ } else {
+ LOG(FATAL) << "Unprintable type " << o->GetTypeKey();
+ }
+}
+} // namespace
+
+String ReportNode::AsJSON() const {
+ std::ostringstream s;
+ // DMLC's JSONWriter does not allow us to write a key value pair without
+ // implementing Write for the value. We want a specific write for the value,
+ // so we would have to implement a custom data structure for each type of
+ // value we want to print. Instead we construct the json by hand because it
+ // is easier.
+ s << "{";
+ s << "\"calls\":[";
+ for (size_t i = 0; i < calls.size(); i++) {
+ size_t j = 0;
+ s << "{";
+ for (const auto& kv : calls[i]) {
+ s << "\"" << kv.first << "\":";
+ print_metric(s, kv.second);
+ if (j < calls[i].size() - 1) {
+ s << ",";
+ }
+ j++;
+ }
+ s << "}";
+ if (i < calls.size() - 1) {
+ s << ",";
+ }
+ }
+ s << "],";
+ s << "\"device_metrics\":{";
+ size_t i = 0;
+ for (const auto& dev_kv : device_metrics) {
+ size_t j = 0;
+ s << "\"" << dev_kv.first << "\":{";
+ for (const auto& metric_kv : dev_kv.second) {
+ s << "\"" << metric_kv.first << "\":";
Review comment:
"Duration (us)" is the name of the key. There definitely needs to be a
cleanup of the names, but I think it should wait for a different pr.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
