1 /*
2 * Copyright (c) Huawei Technologies Co., Ltd. 2023. All rights reserved.
3 * Licensed under the Apache License, Version 2.0 (the "License");
4 * you may not use this file except in compliance with the License.
5 * You may obtain a copy of the License at
6 *
7 * http://www.apache.org/licenses/LICENSE-2.0
8 *
9 * Unless required by applicable law or agreed to in writing, software
10 * distributed under the License is distributed on an "AS IS" BASIS,
11 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 * See the License for the specific language governing permissions and
13 * limitations under the License.
14 */
15
16 #include "memory_process_gpu_table.h"
17
18 namespace SysTuning {
19 namespace TraceStreamer {
20 enum class Index : int32_t {
21 ID = 0,
22 TS,
23 GPU_NAME_ID,
24 ALL_GPU_SIZE,
25 ADDR,
26 IPID,
27 ITID,
28 USED_GPU_SIZE,
29 };
MemoryProcessGpuTable(const TraceDataCache * dataCache)30 MemoryProcessGpuTable::MemoryProcessGpuTable(const TraceDataCache *dataCache) : TableBase(dataCache)
31 {
32 tableColumn_.push_back(TableBase::ColumnInfo("id", "INTEGER"));
33 tableColumn_.push_back(TableBase::ColumnInfo("ts", "INTEGER"));
34 tableColumn_.push_back(TableBase::ColumnInfo("gpu_name_id", "INTEGER"));
35 tableColumn_.push_back(TableBase::ColumnInfo("all_gpu_size", "INTEGER"));
36 tableColumn_.push_back(TableBase::ColumnInfo("addr", "TEXT"));
37 tableColumn_.push_back(TableBase::ColumnInfo("ipid", "INTEGER"));
38 tableColumn_.push_back(TableBase::ColumnInfo("itid", "INTEGER"));
39 tableColumn_.push_back(TableBase::ColumnInfo("used_gpu_size", "INTEGER"));
40 tablePriKey_.push_back("id");
41 }
42
~MemoryProcessGpuTable()43 MemoryProcessGpuTable::~MemoryProcessGpuTable() {}
44
CreateCursor()45 std::unique_ptr<TableBase::Cursor> MemoryProcessGpuTable::CreateCursor()
46 {
47 return std::make_unique<Cursor>(dataCache_, this);
48 }
49
Cursor(const TraceDataCache * dataCache,TableBase * table)50 MemoryProcessGpuTable::Cursor::Cursor(const TraceDataCache *dataCache, TableBase *table)
51 : TableBase::Cursor(dataCache, table, static_cast<uint32_t>(dataCache->GetConstGpuProcessMemData().Size())),
52 GpuProcessMemDataObj_(dataCache->GetConstGpuProcessMemData())
53 {
54 }
55
~Cursor()56 MemoryProcessGpuTable::Cursor::~Cursor() {}
57
Column(int32_t column) const58 int32_t MemoryProcessGpuTable::Cursor::Column(int32_t column) const
59 {
60 switch (static_cast<Index>(column)) {
61 case Index::ID:
62 sqlite3_result_int64(context_, GpuProcessMemDataObj_.IdsData()[CurrentRow()]);
63 break;
64 case Index::TS:
65 sqlite3_result_int64(context_, GpuProcessMemDataObj_.TimeStampData()[CurrentRow()]);
66 break;
67 case Index::GPU_NAME_ID:
68 sqlite3_result_int64(context_, GpuProcessMemDataObj_.GpuNameIds()[CurrentRow()]);
69 break;
70 case Index::ALL_GPU_SIZE:
71 sqlite3_result_int64(context_, GpuProcessMemDataObj_.AllGpuSizes()[CurrentRow()]);
72 break;
73 case Index::ADDR:
74 sqlite3_result_text(context_, GpuProcessMemDataObj_.Addrs()[CurrentRow()].c_str(), STR_DEFAULT_LEN,
75 nullptr);
76 break;
77 case Index::IPID:
78 sqlite3_result_int64(context_, GpuProcessMemDataObj_.Ipids()[CurrentRow()]);
79 break;
80 case Index::ITID:
81 sqlite3_result_int64(context_, GpuProcessMemDataObj_.Itids()[CurrentRow()]);
82 break;
83 case Index::USED_GPU_SIZE:
84 sqlite3_result_int64(context_, GpuProcessMemDataObj_.UsedGpuSizes()[CurrentRow()]);
85 break;
86 default:
87 TS_LOGF("Unregistered column : %d", column);
88 break;
89 }
90 return SQLITE_OK;
91 }
92 } // namespace TraceStreamer
93 } // namespace SysTuning
94