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
|
// (c) Meta Platforms, Inc. and affiliates. Confidential and proprietary.
#include <gtest/gtest.h>
#include "src/cupti_strings.h"
using namespace KINETO_NAMESPACE;
TEST(CuptiStringsTest, Valid) {
ASSERT_STREQ(
runtimeCbidName(CUPTI_RUNTIME_TRACE_CBID_INVALID), "INVALID");
ASSERT_STREQ(
runtimeCbidName(CUPTI_RUNTIME_TRACE_CBID_cudaDriverGetVersion_v3020),
"cudaDriverGetVersion");
ASSERT_STREQ(runtimeCbidName
(CUPTI_RUNTIME_TRACE_CBID_cudaDeviceSynchronize_v3020),
"cudaDeviceSynchronize");
ASSERT_STREQ(
runtimeCbidName(CUPTI_RUNTIME_TRACE_CBID_cudaStreamSetAttribute_ptsz_v11000),
"cudaStreamSetAttribute_ptsz");
}
TEST(CuptiStringsTest, Invalid) {
ASSERT_STREQ(runtimeCbidName(-1), "INVALID");
// We can't actually use CUPTI_RUNTIME_TRACE_CBID_SIZE here until we
// auto-generate the string table, since it may have more entries than
// the enum in the version used to compile.
ASSERT_STREQ(runtimeCbidName(1000), "INVALID");
}
|
