AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

revert npkit (#748)

Browse Source
This commit is contained in:
Ziyue Yang
2023-05-24 22:41:05 +08:00
committed by GitHub
parent ed252c30f4
commit 7d6e7bcd7d
10 changed files with 167 additions and 122 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/include/npkit/npkit.h
+5 -6
View File
@@ -49,7 +49,7 @@ class NpKit {
static void CollectCpuEvent(uint8_t type, uint32_t size, uint32_t rsvd, uint64_t timestamp, int channel_id);
static uint64_t GetCpuTimestamp();
static uint64_t *GetCpuTimestamp();
private:
static void CpuTimestampUpdateThread();
@@ -65,13 +65,12 @@ class NpKit {
static NpKitEventCollectContext* gpu_collect_contexts_;
static NpKitEventCollectContext* cpu_collect_contexts_;
static uint64_t base_cpu_timestamp_global_;
static uint64_t base_cpu_timestamp_local_;
static uint64_t base_gpu_timestamp_cpu_;
static uint64_t base_gpu_timestamp_gpu_;
static uint64_t* cpu_timestamp_;
static uint64_t rank_;
static std::thread* cpu_timestamp_update_thread_;
static volatile bool cpu_timestamp_update_thread_should_stop_;
};
#endif