diff --git a/libkineto/include/GenericTraceActivity.h b/libkineto/include/GenericTraceActivity.h index 1e7bc6ba4..5d01342d5 100644 --- a/libkineto/include/GenericTraceActivity.h +++ b/libkineto/include/GenericTraceActivity.h @@ -75,7 +75,7 @@ class GenericTraceActivity : public ITraceActivity { return flow.type; } - int flowId() const override { + int64_t flowId() const override { return flow.id; } diff --git a/libkineto/include/ITraceActivity.h b/libkineto/include/ITraceActivity.h index 34a3f3248..23c8dfe28 100644 --- a/libkineto/include/ITraceActivity.h +++ b/libkineto/include/ITraceActivity.h @@ -35,7 +35,7 @@ struct ITraceActivity { virtual int64_t correlationId() const = 0; // Part of a flow, identified by flow id and type virtual int flowType() const = 0; - virtual int flowId() const = 0; + virtual int64_t flowId() const = 0; virtual bool flowStart() const = 0; virtual ActivityType type() const = 0; virtual const std::string name() const = 0; diff --git a/libkineto/src/CuptiActivity.h b/libkineto/src/CuptiActivity.h index c5ff6816b..8ba9e7717 100644 --- a/libkineto/src/CuptiActivity.h +++ b/libkineto/src/CuptiActivity.h @@ -80,7 +80,7 @@ struct CuptiActivity : public ITraceActivity { int flowType() const override { return kLinkAsyncCpuGpu; } - int flowId() const override { + int64_t flowId() const override { return correlationId(); } const T& raw() const { diff --git a/libkineto/src/RoctracerActivity.h b/libkineto/src/RoctracerActivity.h index 8dcff4995..72083ab6b 100644 --- a/libkineto/src/RoctracerActivity.h +++ b/libkineto/src/RoctracerActivity.h @@ -58,7 +58,7 @@ struct RoctracerActivity : public ITraceActivity { int flowType() const override { return kLinkAsyncCpuGpu; } - int flowId() const override { + int64_t flowId() const override { return correlationId(); } const T& raw() const {