1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
|
// (c) Meta Platforms, Inc. and affiliates. Confidential and proprietary.
#include "include/Config.h"
#include "src/CuptiRangeProfilerConfig.h"
#include <fmt/format.h>
#include <gtest/gtest.h>
#include <time.h>
#include <chrono>
using namespace std::chrono;
using namespace KINETO_NAMESPACE;
class CuptiRangeProfilerConfigTest : public ::testing::Test {
protected:
void SetUp() override {
CuptiRangeProfilerConfig::registerFactory();
}
};
TEST_F(CuptiRangeProfilerConfigTest, ConfigureProfiler) {
Config cfg;
std::vector<std::string> metrics = {
"kineto__cuda_core_flops",
"sm__inst_executed.sum",
"l1tex__data_bank_conflicts_pipe_lsu.sum",
};
auto metricsConfigStr =
fmt::format("CUPTI_PROFILER_METRICS = {}", fmt::join(metrics, ","));
EXPECT_TRUE(cfg.parse(metricsConfigStr));
EXPECT_TRUE(cfg.parse("CUPTI_PROFILER_ENABLE_PER_KERNEL = true"));
EXPECT_TRUE(cfg.parse("CUPTI_PROFILER_MAX_RANGES = 42"));
const CuptiRangeProfilerConfig& cupti_cfg =
CuptiRangeProfilerConfig::get(cfg);
EXPECT_EQ(cupti_cfg.activitiesCuptiMetrics(), metrics);
EXPECT_EQ(cupti_cfg.cuptiProfilerPerKernel(), true);
EXPECT_EQ(cupti_cfg.cuptiProfilerMaxRanges(), 42);
}
TEST_F(CuptiRangeProfilerConfigTest, RangesDefaults) {
Config cfg, cfg_auto;
// do not set max ranges in config, check defaults are sane
EXPECT_TRUE(cfg.parse("CUPTI_PROFILER_METRICS = kineto__cuda_core_flops"));
EXPECT_TRUE(cfg.parse("CUPTI_PROFILER_ENABLE_PER_KERNEL = false"));
cfg.setSignalDefaults();
EXPECT_TRUE(cfg_auto.parse("CUPTI_PROFILER_METRICS = kineto__cuda_core_flops"));
EXPECT_TRUE(cfg_auto.parse("CUPTI_PROFILER_ENABLE_PER_KERNEL = true"));
cfg_auto.setClientDefaults();
int user_ranges, auto_ranges;
user_ranges = CuptiRangeProfilerConfig::get(cfg).cuptiProfilerMaxRanges();
auto_ranges = CuptiRangeProfilerConfig::get(cfg_auto).cuptiProfilerMaxRanges();
EXPECT_GE(user_ranges, 1) << " in user range mode default to at least 1 ranges";
EXPECT_GE(auto_ranges, 1000) << " in auto range mode default to at least 1000 ranges";
EXPECT_GT(auto_ranges, user_ranges);
}
|
