Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Fix test with clang tidy change #1034

Closed
wants to merge 2 commits into from
Closed
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: 0 additions & 4 deletions libkineto/src/Config.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -15,13 +15,9 @@
#include <fmt/ranges.h>
#include <time.h>
#include <chrono>
#include <fstream>
#include <functional>
#include <iomanip>
#include <istream>
#include <mutex>
#include <ostream>
#include <sstream>

#include "Logger.h"
#include "ThreadUtil.h"
Expand Down
6 changes: 1 addition & 5 deletions libkineto/src/ConfigLoader.h
Original file line number Diff line number Diff line change
Expand Up @@ -19,10 +19,6 @@

#include "Config.h"

// TODO(T90238193)
// @lint-ignore-every CLANGTIDY facebook-hte-RelativeInclude
#include "ILoggerObserver.h"

namespace libkineto {
class LibkinetoApi;
}
Expand Down Expand Up @@ -95,7 +91,7 @@ class ConfigLoader {
}

bool hasNewConfig(const Config& oldConfig);
int contextCountForGpu(uint32_t gpu);
int contextCountForGpu(uint32_t device);

void handleOnDemandSignal();

Expand Down
6 changes: 2 additions & 4 deletions libkineto/src/CuptiActivityApi.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -8,12 +8,10 @@

#include "CuptiActivityApi.h"

#include <assert.h>
#include <chrono>
#include <mutex>
#include <thread>

#include "Config.h"
#include "DeviceUtil.h"
#include "Logger.h"

Expand Down Expand Up @@ -214,7 +212,7 @@ std::unique_ptr<CuptiActivityBufferMap> CuptiActivityApi::activityBuffers() {
int CuptiActivityApi::processActivitiesForBuffer(
uint8_t* buf,
size_t validSize,
std::function<void(const CUpti_Activity*)> handler) {
const std::function<void(const CUpti_Activity*)>& handler) {
int count = 0;
if (buf && validSize) {
CUpti_Activity* record{nullptr};
Expand All @@ -229,7 +227,7 @@ int CuptiActivityApi::processActivitiesForBuffer(

const std::pair<int, size_t> CuptiActivityApi::processActivities(
CuptiActivityBufferMap& buffers,
std::function<void(const CUpti_Activity*)> handler) {
const std::function<void(const CUpti_Activity*)>& handler) {
std::pair<int, size_t> res{0, 0};
#ifdef HAS_CUPTI
for (auto& pair : buffers) {
Expand Down
7 changes: 3 additions & 4 deletions libkineto/src/CuptiActivityApi.h
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,6 @@
#include <atomic>
#include <condition_variable>
#include <functional>
#include <list>
#include <memory>
#include <mutex>
#include <set>
Expand Down Expand Up @@ -48,7 +47,7 @@ class CuptiActivityApi {
CuptiActivityApi(const CuptiActivityApi&) = delete;
CuptiActivityApi& operator=(const CuptiActivityApi&) = delete;

virtual ~CuptiActivityApi() {}
virtual ~CuptiActivityApi() = default;

static CuptiActivityApi& singleton();

Expand All @@ -65,7 +64,7 @@ class CuptiActivityApi {

virtual const std::pair<int, size_t> processActivities(
CuptiActivityBufferMap&,
std::function<void(const CUpti_Activity*)> handler);
const std::function<void(const CUpti_Activity*)>& handler);

void setMaxBufferSize(int size);
void setDeviceBufferSize(size_t size);
Expand All @@ -91,7 +90,7 @@ class CuptiActivityApi {
int processActivitiesForBuffer(
uint8_t* buf,
size_t validSize,
std::function<void(const CUpti_Activity*)> handler);
const std::function<void(const CUpti_Activity*)>& handler);
static void CUPTIAPI bufferRequestedTrampoline(
uint8_t** buffer,
size_t* size,
Expand Down
21 changes: 10 additions & 11 deletions libkineto/src/CuptiActivityProfiler.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -9,11 +9,9 @@
#include "CuptiActivityProfiler.h"
#include <fmt/format.h>
#include <fmt/ranges.h>
#include <time.h>
#include <atomic>
#include <cstdint>
#include <functional>
#include <iomanip>
#include <limits>
#include <optional>
#include <string>
Expand Down Expand Up @@ -504,22 +502,23 @@ static GenericTraceActivity createUserGpuSpan(
}

void CuptiActivityProfiler::GpuUserEventMap::insertOrExtendEvent(
const ITraceActivity& userActivity,
const ITraceActivity& gpuActivity) {
StreamKey key(gpuActivity.deviceId(), gpuActivity.resourceId());
const ITraceActivity& cpuTraceActivity,
const ITraceActivity& gpuTraceActivity) {
StreamKey key(gpuTraceActivity.deviceId(), gpuTraceActivity.resourceId());
CorrelationSpanMap& correlationSpanMap = streamSpanMap_[key];
auto it = correlationSpanMap.find(userActivity.correlationId());
auto it = correlationSpanMap.find(cpuTraceActivity.correlationId());
if (it == correlationSpanMap.end()) {
auto it_success = correlationSpanMap.insert(
{userActivity.correlationId(),
createUserGpuSpan(userActivity, gpuActivity)});
{cpuTraceActivity.correlationId(),
createUserGpuSpan(cpuTraceActivity, gpuTraceActivity)});
it = it_success.first;
}
GenericTraceActivity& span = it->second;
if (gpuActivity.timestamp() < span.startTime || span.startTime == 0) {
span.startTime = gpuActivity.timestamp();
if (gpuTraceActivity.timestamp() < span.startTime || span.startTime == 0) {
span.startTime = gpuTraceActivity.timestamp();
}
int64_t gpu_activity_end = gpuActivity.timestamp() + gpuActivity.duration();
int64_t gpu_activity_end =
gpuTraceActivity.timestamp() + gpuTraceActivity.duration();
if (gpu_activity_end > span.endTime) {
span.endTime = gpu_activity_end;
}
Expand Down
7 changes: 3 additions & 4 deletions libkineto/src/CuptiActivityProfiler.h
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,6 @@

#include <atomic>
#include <chrono>
#include <condition_variable>
#include <deque>
#include <list>
#include <map>
Expand Down Expand Up @@ -149,11 +148,11 @@ class CuptiActivityProfiler {
gpuActivityPresent_ = val;
}

inline bool gpuActivityPresent() {
inline bool gpuActivityPresent() const {
return gpuActivityPresent_;
}

inline bool traceNonEmpty() {
inline bool traceNonEmpty() const {
return cpuActivityPresent_ || gpuActivityPresent_;
}

Expand All @@ -172,7 +171,7 @@ class CuptiActivityProfiler {

// Collect CPU and GPU traces
void collectTrace(
bool collectionDone,
bool collection_done,
const std::chrono::time_point<std::chrono::system_clock>& now);

// Ensure collectTrace is done
Expand Down
3 changes: 2 additions & 1 deletion libkineto/src/RoctracerActivityApi.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -63,7 +63,8 @@ inline bool inRange(int64_t start, int64_t end, int64_t stamp) {
return ((stamp > start) && (stamp < end));
}

inline bool RoctracerActivityApi::isLogged(libkineto::ActivityType atype) {
inline bool RoctracerActivityApi::isLogged(
libkineto::ActivityType atype) const {
return activityMaskSnapshot_ & (1 << static_cast<uint32_t>(atype));
}

Expand Down
8 changes: 3 additions & 5 deletions libkineto/src/RoctracerActivityApi.h
Original file line number Diff line number Diff line change
Expand Up @@ -7,17 +7,14 @@
*/

#pragma once
#ifdef HAS_ROCTRACER

#include <atomic>
#include <functional>
#include <map>
#include <set>
#include <vector>

#ifdef HAS_ROCTRACER
#include <roctracer.h>
#include "RoctracerLogger.h"
#endif

#include "ActivityType.h"
#include "GenericTraceActivity.h"
Expand Down Expand Up @@ -66,9 +63,10 @@ class RoctracerActivityApi {
// Enabled Activity Filters
uint32_t activityMask_{0};
uint32_t activityMaskSnapshot_{0};
bool isLogged(libkineto::ActivityType atype);
bool isLogged(libkineto::ActivityType atype) const;

RoctracerLogger* d;
};

} // namespace KINETO_NAMESPACE
#endif
2 changes: 0 additions & 2 deletions libkineto/src/output_json.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -11,9 +11,7 @@
#include <fmt/format.h>
#include <time.h>
#include <fstream>
#include <map>
#include "Config.h"
#include "DeviceProperties.h"
#include "TraceSpan.h"

#include "Logger.h"
Expand Down
2 changes: 1 addition & 1 deletion libkineto/test/CuptiActivityProfilerTest.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -219,7 +219,7 @@ class MockCuptiActivities : public CuptiActivityApi {
public:
virtual const std::pair<int, size_t> processActivities(
CuptiActivityBufferMap&, /*unused*/
std::function<void(const CUpti_Activity*)> handler) override {
const std::function<void(const CUpti_Activity*)>& handler) override {
for (CUpti_Activity* act : activityBuffer->activities) {
handler(act);
}
Expand Down
Loading