diff --git a/libkineto/src/ActivityProfilerController.h b/libkineto/src/ActivityProfilerController.h index 9b23d6a88..415f107cb 100644 --- a/libkineto/src/ActivityProfilerController.h +++ b/libkineto/src/ActivityProfilerController.h @@ -3,6 +3,7 @@ #pragma once #include +#include #include #include #include @@ -76,7 +77,7 @@ class ActivityProfilerController : public ConfigLoader::ConfigHandler { std::unique_ptr logger_; std::thread* profilerThread_{nullptr}; std::atomic_bool stopRunloop_{false}; - std::atomic_int64_t iterationCount_{-1}; + std::atomic iterationCount_{-1}; ConfigLoader& configLoader_; };