cuVein/cuThermo init to main repo

include/sanalyzer.h

@@ -25,6 +25,9 @@ typedef enum {
     GPU_PATCH_TIME_HOTNESS_CPU = 8,
     GPU_PATCH_ROOFLINE_FLOPS_NVBIT = 9,
     GPU_PATCH_ROOFLINE_SIZE = 10,
+    GPU_PATCH_HEATMAP_ANALYSIS = 11,
+    GPU_PATCH_BLOCK_DIVERGENCE_ANALYSIS = 12,
+    GPU_PATCH_PC_DEPENDENCY_ANALYSIS = 13,
 } AccelProfPatchName_t;
 
 

include/tools/block_divergence_analysis.h

@@ -0,0 +1,73 @@
+#ifndef YOSEMITE_TOOL_BLOCK_DIVERGENCE_ANALYSIS_H
+#define YOSEMITE_TOOL_BLOCK_DIVERGENCE_ANALYSIS_H
+
+
+#include "tools/tool.h"
+#include "utils/event.h"
+#include "gpu_patch.h"
+
+#include <map>
+#include <vector>
+#include <set>
+#include <unordered_map>
+namespace yosemite {
+
+class BlockDivergenceAnalysis final : public Tool {
+public:
+    BlockDivergenceAnalysis();
+
+    ~BlockDivergenceAnalysis();
+
+    void gpu_data_analysis(void* data, uint64_t size);
+
+    void query_ranges(void* ranges, uint32_t limit, uint32_t* count) override {};
+
+    void query_tensors(void* ranges, uint32_t limit, uint32_t* count) override {};
+
+    void evt_callback(EventPtr_t evt);
+
+    void flush();
+
+private:
+    void kernel_start_callback(std::shared_ptr<KernelLaunch_t> kernel);
+
+    void kernel_end_callback(std::shared_ptr<KernelEnd_t> kernel);
+
+    void mem_alloc_callback(std::shared_ptr<MemAlloc_t> mem);
+
+    void mem_free_callback(std::shared_ptr<MemFree_t> mem);
+
+    void ten_alloc_callback(std::shared_ptr<TenAlloc_t> ten);
+
+    void ten_free_callback(std::shared_ptr<TenFree_t> ten);
+
+    void kernel_trace_flush(std::shared_ptr<KernelLaunch_t> kernel);
+
+
+/*
+********************************* variables *********************************
+*/
+    Timer_t _timer;
+
+    std::string output_directory;
+    uint32_t kernel_id = 0;
+
+    std::map<uint64_t, std::shared_ptr<KernelLaunch_t>> kernel_events;
+    std::map<uint64_t, std::shared_ptr<MemAlloc_t>> alloc_events;
+    std::map<DevPtr, std::shared_ptr<MemAlloc_t>> active_memories;
+
+    std::map<uint64_t, std::shared_ptr<TenAlloc_t>> tensor_events;
+    std::map<DevPtr, std::shared_ptr<TenAlloc_t>> active_tensors;
+
+    struct BlockStat {
+        std::unordered_map<uint64_t, uint64_t> pc_counts;
+        uint64_t read_count = 0;
+        uint64_t write_count = 0;
+    };
+
+    std::unordered_map<uint64_t, BlockStat> _block_entries;
+    std::set<uint64_t> _unique_pcs;
+};
+
+}   // yosemite
+#endif // YOSEMITE_TOOL_BLOCK_DIVERGENCE_ANALYSIS_H

include/tools/heatmap_analysis.h

@@ -0,0 +1,78 @@
+#ifndef YOSEMITE_HEATMAP_ANALYSIS_H
+#define YOSEMITE_HEATMAP_ANALYSIS_H
+
+
+#include "tools/tool.h"
+#include "utils/event.h"
+#include "gpu_patch.h"
+
+#include <map>
+#include <unordered_map>
+#include <set>
+#include <vector>
+#include <array>
+
+#define SECTOR_TAG_SHIFT 5
+
+namespace yosemite {
+
+class HeatmapAnalysis final : public Tool {
+public:
+    HeatmapAnalysis();
+
+    ~HeatmapAnalysis();
+
+    void gpu_data_analysis(void* data, uint64_t size);
+
+    void query_ranges(void* ranges, uint32_t limit, uint32_t* count) override {};
+
+    void query_tensors(void* ranges, uint32_t limit, uint32_t* count) override {};
+
+    void evt_callback(EventPtr_t evt);
+
+    void flush();
+
+private:
+    void unit_access(uint32_t warp_id, uint64_t sector_tag, uint32_t offset, uint32_t length);
+
+    void add_sector_pc_information(uint64_t sector_tag, uint64_t pc);
+
+    void kernel_start_callback(std::shared_ptr<KernelLaunch_t> kernel);
+
+    void kernel_end_callback(std::shared_ptr<KernelEnd_t> kernel);
+
+    void mem_alloc_callback(std::shared_ptr<MemAlloc_t> mem);
+
+    void mem_free_callback(std::shared_ptr<MemFree_t> mem);
+
+    void ten_alloc_callback(std::shared_ptr<TenAlloc_t> ten);
+
+    void ten_free_callback(std::shared_ptr<TenFree_t> ten);
+
+    void kernel_trace_flush(std::shared_ptr<KernelLaunch_t> kernel);
+
+
+/*
+********************************* variables *********************************
+*/
+
+    Timer_t _timer;
+
+    std::string output_directory;
+    uint32_t kernel_id = 0;
+
+    std::map<uint64_t, std::shared_ptr<KernelLaunch_t>> kernel_events;
+    std::map<uint64_t, std::shared_ptr<MemAlloc_t>> alloc_events;
+    std::map<DevPtr, std::shared_ptr<MemAlloc_t>> active_memories;
+
+    std::map<uint64_t, std::shared_ptr<TenAlloc>> tensor_events;
+    std::map<DevPtr, std::shared_ptr<TenAlloc>> active_tensors;
+
+    std::vector<MemoryAccess> _traces;
+    std::unordered_map<uint64_t, std::array<uint32_t, 18>> _heatmap_data;
+    std::unordered_map<uint64_t, std::set<uint64_t>> _sector_pc_information;
+
+};
+
+}   // namespace yosemite
+#endif // YOSEMITE_HEATMAP_ANALYSIS_H