Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 4 additions & 0 deletions onnxruntime/core/platform/telemetry.cc
Original file line number Diff line number Diff line change
Expand Up @@ -81,6 +81,10 @@ void Telemetry::LogSessionCreation(uint32_t session_id, int64_t ir_version, cons
ORT_UNUSED_PARAMETER(captureState);
}

void Telemetry::LogCompileModel(uint32_t session_id) const {
ORT_UNUSED_PARAMETER(session_id);
}

void Telemetry::LogRuntimeError(uint32_t session_id, const common::Status& status, const char* file,
const char* function, uint32_t line) const {
ORT_UNUSED_PARAMETER(session_id);
Expand Down
2 changes: 2 additions & 0 deletions onnxruntime/core/platform/telemetry.h
Original file line number Diff line number Diff line change
Expand Up @@ -66,6 +66,8 @@ class Telemetry {
const std::string& loadedFrom, const std::vector<std::string>& execution_provider_ids,
bool use_fp16, bool captureState) const;

virtual void LogCompileModel(uint32_t session_id) const;

virtual void LogRuntimeError(uint32_t session_id, const common::Status& status, const char* file,
const char* function, uint32_t line) const;

Expand Down
14 changes: 14 additions & 0 deletions onnxruntime/core/platform/windows/telemetry.cc
Original file line number Diff line number Diff line change
Expand Up @@ -334,6 +334,20 @@ void WindowsTelemetry::LogSessionCreation(uint32_t session_id, int64_t ir_versio
}
}

void WindowsTelemetry::LogCompileModel(uint32_t session_id) const {
if (global_register_count_ == 0 || enabled_ == false)
return;

TraceLoggingWrite(telemetry_provider_handle,
"CompileModel",
TraceLoggingBool(true, "UTCReplace_AppSessionGuid"),
TelemetryPrivacyDataTag(PDT_ProductAndServiceUsage),
TraceLoggingKeyword(MICROSOFT_KEYWORD_MEASURES),
// Telemetry info
TraceLoggingUInt8(0, "schemaVersion"),
TraceLoggingUInt32(session_id, "sessionId"));
}

void WindowsTelemetry::LogRuntimeError(uint32_t session_id, const common::Status& status, const char* file,
const char* function, uint32_t line) const {
if (global_register_count_ == 0 || enabled_ == false)
Expand Down
2 changes: 2 additions & 0 deletions onnxruntime/core/platform/windows/telemetry.h
Original file line number Diff line number Diff line change
Expand Up @@ -59,6 +59,8 @@ class WindowsTelemetry : public Telemetry {
const std::string& loadedFrom, const std::vector<std::string>& execution_provider_ids,
bool use_fp16, bool captureState) const override;

void LogCompileModel(uint32_t session_id) const override;

void LogRuntimeError(uint32_t session_id, const common::Status& status, const char* file,
const char* function, uint32_t line) const override;

Expand Down
1 change: 1 addition & 0 deletions onnxruntime/core/session/utils.cc
Original file line number Diff line number Diff line change
Expand Up @@ -404,6 +404,7 @@ Status CompileModel(const Environment& env, const ModelCompilationOptions& model
session)));
}

Env::Default().GetTelemetryProvider().LogCompileModel(session->GetCurrentSessionId());
ORT_RETURN_IF_ERROR(ToStatusAndRelease(InitializeSession(session_options, *session)));
return Status::OK();
}
Expand Down
Loading