AI/rocm-systems
2
0
Derivar 0
Código Questões Pedidos de integração Operações Pacotes Planeamentos Lançamentos Wiki Trabalho

LINT: Add cpplint, clang-format and pre-commit support

Ver a proveniência 
Change-Id: I3cbb787ef27d90486b212dfb1a8c77c460acc2ac
Signed-off-by: Galantsev, Dmitrii <dmitrii.galantsev@amd.com>


[ROCm/rdc commit: 434e40305d]
Este cometimento está contido em:
Galantsev, Dmitrii
2023-12-04 15:24:34 -06:00
ascendente 61cf14d7cc
cometimento ea624cbb7c
137 ficheiros modificados com 9122 adições e 10469 eliminações
Descarregar ficheiro patch Descarregar ficheiro diff Expandir todos os ficheiros Colapsar todos os ficheiros
projects/rdc/example/field_value_example.cc
+215 -237
Ver ficheiro
@@ -21,257 +21,235 @@ THE SOFTWARE.
*/
#include <unistd.h>
#include <iostream>
#include <iomanip>
#include <iostream>
#include "rdc/rdc.h"
int main(int, char **) {
rdc_status_t result;
rdc_handle_t rdc_handle;
bool standalone = false;
char hostIpAddress[] = {"127.0.0.1:50051"};
char group_name[] = {"group1"};
char field_group_name[] = {"fieldgroup1"};
uint64_t since_timestamp = 0;
uint64_t next_timestamp = 0;
uint64_t start_timestamp = 0;
uint32_t count = 0;
int main(int, char**) {
rdc_status_t result;
rdc_handle_t rdc_handle;
bool standalone = false;
char hostIpAddress[] = {"127.0.0.1:50051"};
char group_name[] = {"group1"};
char field_group_name[] = {"fieldgroup1"};
uint64_t since_timestamp = 0;
uint64_t next_timestamp = 0;
uint64_t start_timestamp = 0;
uint32_t count = 0;
// Select the embedded mode and standalone mode dynamically.
std::cout << "Start rdci in: \n";
std::cout << "0 - Embedded mode \n";
std::cout << "1 - Standalone mode \n";
while (!(std::cin >> standalone)) {
std::cout << "Invalid input.\n";
std::cin.clear();
std::cin.ignore();
}
std::cout << std::endl;
std::cout << (standalone ? "Standalone mode selected.\n" : "Embedded mode selected.\n");
// Select the embedded mode and standalone mode dynamically.
std::cout << "Start rdci in: \n";
std::cout << "0 - Embedded mode \n";
std::cout << "1 - Standalone mode \n";
while (!(std::cin >> standalone)) {
std::cout << "Invalid input.\n";
std::cin.clear();
std::cin.ignore();
}
std::cout << std::endl;
std::cout << (standalone?
"Standalone mode selected.\n":"Embedded mode selected.\n");
// Init the rdc
result = rdc_init(0);
// Init the rdc
result = rdc_init(0);
if (result != RDC_ST_OK) {
std::cout << "Error initializing RDC. Return: " << rdc_status_string(result) << std::endl;
goto cleanup;
} else {
std::cout << "RDC Initialized.\n";
}
if (standalone) { // standalone
result = rdc_connect(hostIpAddress, &rdc_handle, nullptr, nullptr, nullptr);
if (result != RDC_ST_OK) {
std::cout << "Error initializing RDC. Return: " <<
rdc_status_string(result) << std::endl;
goto cleanup;
} else {
std::cout << "RDC Initialized.\n";
std::cout << "Error connecting to remote rdcd. Return: " << rdc_status_string(result)
<< std::endl;
goto cleanup;
}
} else { // embedded
result = rdc_start_embedded(RDC_OPERATION_MODE_AUTO, &rdc_handle);
if (result != RDC_ST_OK) {
std::cout << "Error starting embedded RDC engine. Return: " << rdc_status_string(result)
<< std::endl;
goto cleanup;
}
}
if (standalone) { // standalone
result = rdc_connect(hostIpAddress, &rdc_handle,
nullptr, nullptr, nullptr);
if ( result != RDC_ST_OK ) {
std::cout << "Error connecting to remote rdcd. Return: "
<< rdc_status_string(result) << std::endl;
goto cleanup;
// Now we can use the same API for both standalone and embedded
// Get the list of devices in the system
uint32_t gpu_index_list[RDC_MAX_NUM_DEVICES];
result = rdc_device_get_all(rdc_handle, gpu_index_list, &count);
if (result != RDC_ST_OK) {
std::cout << "Error to find devices on the system. Return: " << rdc_status_string(result);
goto cleanup;
}
if (count == 0) {
std::cout << "No GPUs find on the sytem ";
goto cleanup;
} else {
std::cout << count << " GPUs found in the system.\n";
}
// Create the group
rdc_gpu_group_t group_id;
result = rdc_group_gpu_create(rdc_handle, RDC_GROUP_EMPTY, group_name, &group_id);
if (result != RDC_ST_OK) {
std::cout << "Error creating group. Return: " << rdc_status_string(result);
goto cleanup;
}
std::cout << "Created the GPU group " << group_id << std::endl;
// Add all GPUs to the group
for (uint32_t i = 0; i < count; i++) {
result = rdc_group_gpu_add(rdc_handle, group_id, gpu_index_list[i]); // Add GPU 0
if (result != RDC_ST_OK) {
std::cout << "Error adding group. Return: " << rdc_status_string(result);
goto cleanup;
}
rdc_device_attributes_t attribute;
result = rdc_device_get_attributes(rdc_handle, gpu_index_list[i], &attribute);
if (result != RDC_ST_OK) {
std::cout << "Error get GPU attribute. Return: " << rdc_status_string(result);
goto cleanup;
}
std::cout << "Add GPU " << gpu_index_list[i] << ":" << attribute.device_name << " to group "
<< group_id << std::endl;
}
// Create the field groups to monitor POWER and TEMP
rdc_field_grp_t field_group_id;
rdc_field_t field_ids[2];
field_ids[0] = RDC_FI_GPU_MEMORY_USAGE;
field_ids[1] = RDC_FI_POWER_USAGE;
result = rdc_group_field_create(rdc_handle, 2, &field_ids[0], field_group_name, &field_group_id);
if (result != RDC_ST_OK) {
std::cout << "Error create field group, Return: " << rdc_status_string(result);
goto cleanup;
}
std::cout << "Created the field group " << field_group_id << ": "
<< field_id_string(RDC_FI_GPU_MEMORY_USAGE) << ", "
<< field_id_string(RDC_FI_POWER_USAGE) << std::endl;
// Let the RDC to watch the fields and groups. The fields will be updated
// once per second, the max keep age is 1 minutes and only keep 10 samples.
result = rdc_field_watch(rdc_handle, group_id, field_group_id, 1000000, 60, 10);
if (result != RDC_ST_OK) {
std::cout << "Error watch group fields. Return: " << rdc_status_string(result);
goto cleanup;
}
std::cout << "Start to watch group:" << group_id << ", field_group:" << field_group_id
<< std::endl;
std::cout << "Sleep a few seconds before retreive the data ...\n";
// Since we are running the RDC_OPERATION_MODE_AUTO mode, the rdc_update_
// all_fields() will be called periodically at background. If running as
// RDC_OPERATION_MODE_MANUAL mode, we must call rdc_field_update_all()
// periodically to take samples.
usleep(5000000); // sleep 5 seconds before fetch the stats
// Retreive the field and group information from RDC
rdc_group_info_t group_info;
rdc_field_group_info_t field_info;
result = rdc_group_gpu_get_info(rdc_handle, group_id, &group_info);
if (result != RDC_ST_OK) {
std::cout << "Error get gpu group info. Return: " << rdc_status_string(result);
goto cleanup;
}
result = rdc_group_field_get_info(rdc_handle, field_group_id, &field_info);
if (result != RDC_ST_OK) {
std::cout << "Error get field group info. Return: " << rdc_status_string(result);
goto cleanup;
}
// Get the latest metrics
std::cout << "Get the latest metrics for group:" << group_id << " field_group:" << field_group_id
<< std::endl;
std::cout << "time_stamp\t"
<< "GPU_index\t"
<< "field_name\t\t"
<< "field_value\n";
for (uint32_t gindex = 0; gindex < group_info.count; gindex++) {
for (uint32_t findex = 0; findex < field_info.count; findex++) {
rdc_field_value value;
result = rdc_field_get_latest_value(rdc_handle, group_info.entity_ids[gindex],
field_info.field_ids[findex], &value);
if (result == RDC_ST_NOT_FOUND) {
continue;
}
if (result != RDC_ST_OK) {
std::cout << "Error get least value. Return: " << rdc_status_string(result);
goto cleanup;
}
// We only support the integer metrics so far
std::cout << value.ts << "\t" << group_info.entity_ids[gindex] << "\t\t" << std::left
<< std::setw(16) << field_id_string(value.field_id) << "\t" << value.value.l_int
<< std::endl;
}
}
// Stop watching the field group
result = rdc_field_unwatch(rdc_handle, group_id, field_group_id);
if (result != RDC_ST_OK) {
std::cout << "Error stop watch fields. Return: " << rdc_status_string(result);
goto cleanup;
}
std::cout << "Stop watch group:" << group_id << ", field_group:" << field_group_id << std::endl;
// Get the history data last 10 seconds
std::cout << "Get last 10 seconds metrics for group:" << group_id
<< " field_group:" << field_group_id << std::endl;
std::cout << "time_stamp\t"
<< "GPU_index\t"
<< "field_name\t\t"
<< "field_value\n";
start_timestamp = static_cast<uint64_t>(time(nullptr) - 10) * 1000;
for (uint32_t gindex = 0; gindex < group_info.count; gindex++) {
for (uint32_t findex = 0; findex < field_info.count; findex++) {
since_timestamp = start_timestamp;
while (true) {
rdc_field_value value;
result = rdc_field_get_value_since(rdc_handle, group_info.entity_ids[gindex],
field_info.field_ids[findex], since_timestamp,
&next_timestamp, &value);
if (result == RDC_ST_NOT_FOUND) {
break;
}
} else { // embedded
result = rdc_start_embedded(RDC_OPERATION_MODE_AUTO, &rdc_handle);
if (result != RDC_ST_OK) {
std::cout << "Error starting embedded RDC engine. Return: "
<< rdc_status_string(result) << std::endl;
goto cleanup;
std::cout << "Error get history data. Return: " << rdc_status_string(result);
goto cleanup;
}
}
std::cout << value.ts << "\t" << group_info.entity_ids[gindex] << "\t\t" << std::left
<< std::setw(16) << field_id_string(value.field_id) << "\t" << value.value.l_int
<< std::endl;
since_timestamp = next_timestamp;
} // while
} // for findex
} // for gindex
// Now we can use the same API for both standalone and embedded
// Get the list of devices in the system
uint32_t gpu_index_list[RDC_MAX_NUM_DEVICES];
result = rdc_device_get_all(rdc_handle, gpu_index_list, &count);
if (result != RDC_ST_OK) {
std::cout << "Error to find devices on the system. Return: "
<< rdc_status_string(result);
goto cleanup;
}
if (count == 0) {
std::cout << "No GPUs find on the sytem ";
goto cleanup;
} else {
std::cout << count << " GPUs found in the system.\n";
}
// Delete the field group and GPU group
result = rdc_group_field_destroy(rdc_handle, field_group_id);
if (result != RDC_ST_OK) {
std::cout << "Error delete field group. Return: " << rdc_status_string(result);
goto cleanup;
}
std::cout << "Deleted the field group " << field_group_id << std::endl;
// Create the group
rdc_gpu_group_t group_id;
result = rdc_group_gpu_create(rdc_handle, RDC_GROUP_EMPTY,
group_name, &group_id);
if (result != RDC_ST_OK) {
std::cout << "Error creating group. Return: "
<< rdc_status_string(result);
goto cleanup;
}
std::cout << "Created the GPU group " << group_id << std::endl;
result = rdc_group_gpu_destroy(rdc_handle, group_id);
if (result != RDC_ST_OK) {
std::cout << "Error delete GPU group. Return: " << rdc_status_string(result);
goto cleanup;
}
std::cout << "Deleted the GPU group " << group_id << std::endl;
// Add all GPUs to the group
for (uint32_t i = 0; i < count; i++) {
result = rdc_group_gpu_add(rdc_handle,
group_id, gpu_index_list[i]); // Add GPU 0
if (result != RDC_ST_OK) {
std::cout << "Error adding group. Return: "
<< rdc_status_string(result);
goto cleanup;
}
rdc_device_attributes_t attribute;
result = rdc_device_get_attributes(rdc_handle,
gpu_index_list[i], &attribute);
if (result != RDC_ST_OK) {
std::cout << "Error get GPU attribute. Return: "
<< rdc_status_string(result);
goto cleanup;
}
std::cout << "Add GPU " <<gpu_index_list[i] << ":"
<< attribute.device_name <<" to group "<< group_id << std::endl;
}
// Create the field groups to monitor POWER and TEMP
rdc_field_grp_t field_group_id;
rdc_field_t field_ids[2];
field_ids[0] = RDC_FI_GPU_MEMORY_USAGE;
field_ids[1] = RDC_FI_POWER_USAGE;
result = rdc_group_field_create(rdc_handle, 2,
&field_ids[0], field_group_name, &field_group_id);
if (result != RDC_ST_OK) {
std::cout << "Error create field group, Return: "
<< rdc_status_string(result);
goto cleanup;
}
std::cout<< "Created the field group " << field_group_id
<< ": " << field_id_string(RDC_FI_GPU_MEMORY_USAGE) << ", "
<< field_id_string(RDC_FI_POWER_USAGE) <<std::endl;
// Let the RDC to watch the fields and groups. The fields will be updated
// once per second, the max keep age is 1 minutes and only keep 10 samples.
result = rdc_field_watch(rdc_handle, group_id,
field_group_id, 1000000, 60, 10);
if (result != RDC_ST_OK) {
std::cout << "Error watch group fields. Return: "
<< rdc_status_string(result);
goto cleanup;
}
std::cout << "Start to watch group:"
<< group_id << ", field_group:" << field_group_id << std::endl;
std::cout << "Sleep a few seconds before retreive the data ...\n";
// Since we are running the RDC_OPERATION_MODE_AUTO mode, the rdc_update_
// all_fields() will be called periodically at background. If running as
// RDC_OPERATION_MODE_MANUAL mode, we must call rdc_field_update_all()
// periodically to take samples.
usleep(5000000); // sleep 5 seconds before fetch the stats
// Retreive the field and group information from RDC
rdc_group_info_t group_info;
rdc_field_group_info_t field_info;
result = rdc_group_gpu_get_info(rdc_handle, group_id, &group_info);
if (result != RDC_ST_OK) {
std::cout << "Error get gpu group info. Return: "
<< rdc_status_string(result);
goto cleanup;
}
result = rdc_group_field_get_info(rdc_handle, field_group_id, &field_info);
if (result != RDC_ST_OK) {
std::cout << "Error get field group info. Return: "
<< rdc_status_string(result);
goto cleanup;
}
// Get the latest metrics
std::cout << "Get the latest metrics for group:" << group_id
<< " field_group:" << field_group_id << std::endl;
std::cout << "time_stamp\t" << "GPU_index\t"
<< "field_name\t\t" << "field_value\n";
for (uint32_t gindex = 0; gindex < group_info.count; gindex++) {
for (uint32_t findex = 0; findex < field_info.count; findex++) {
rdc_field_value value;
result = rdc_field_get_latest_value(rdc_handle,
group_info.entity_ids[gindex], field_info.field_ids[findex], &value);
if (result == RDC_ST_NOT_FOUND) {
continue;
}
if (result != RDC_ST_OK) {
std::cout << "Error get least value. Return: "
<< rdc_status_string(result);
goto cleanup;
}
// We only support the integer metrics so far
std::cout << value.ts <<"\t" << group_info.entity_ids[gindex]
<< "\t\t" << std::left << std::setw(16)
<< field_id_string(value.field_id) << "\t"
<< value.value.l_int << std::endl;
}
}
// Stop watching the field group
result = rdc_field_unwatch(rdc_handle, group_id, field_group_id);
if (result != RDC_ST_OK) {
std::cout << "Error stop watch fields. Return: "
<< rdc_status_string(result);
goto cleanup;
}
std::cout << "Stop watch group:" << group_id << ", field_group:"
<< field_group_id << std::endl;
// Get the history data last 10 seconds
std::cout << "Get last 10 seconds metrics for group:" << group_id
<< " field_group:" << field_group_id << std::endl;
std::cout << "time_stamp\t" << "GPU_index\t"
<< "field_name\t\t" << "field_value\n";
start_timestamp = static_cast<uint64_t>(time(nullptr)-10)*1000;
for (uint32_t gindex = 0; gindex < group_info.count; gindex++) {
for (uint32_t findex = 0; findex < field_info.count; findex++) {
since_timestamp = start_timestamp;
while (true) {
rdc_field_value value;
result = rdc_field_get_value_since(rdc_handle,
group_info.entity_ids[gindex] , field_info.field_ids[findex],
since_timestamp, &next_timestamp, &value);
if (result == RDC_ST_NOT_FOUND) {
break;
}
if (result != RDC_ST_OK) {
std::cout << "Error get history data. Return: "
<< rdc_status_string(result);
goto cleanup;
}
std::cout << value.ts <<"\t" << group_info.entity_ids[gindex]
<< "\t\t" << std::left << std::setw(16)
<< field_id_string(value.field_id) << "\t"
<< value.value.l_int << std::endl;
since_timestamp = next_timestamp;
} // while
} // for findex
} // for gindex
// Delete the field group and GPU group
result = rdc_group_field_destroy(rdc_handle, field_group_id);
if (result != RDC_ST_OK) {
std::cout << "Error delete field group. Return: "
<< rdc_status_string(result);
goto cleanup;
}
std::cout << "Deleted the field group " << field_group_id << std::endl;
result = rdc_group_gpu_destroy(rdc_handle, group_id);
if (result != RDC_ST_OK) {
std::cout << "Error delete GPU group. Return: "
<< rdc_status_string(result);
goto cleanup;
}
std::cout << "Deleted the GPU group " << group_id << std::endl;
// Cleanup consists of shutting down RDC.
cleanup:
std::cout << "Cleaning up.\n";
if (standalone)
rdc_disconnect(rdc_handle);
else
rdc_stop_embedded(rdc_handle);
rdc_shutdown();
return result;
// Cleanup consists of shutting down RDC.
cleanup:
std::cout << "Cleaning up.\n";
if (standalone)
rdc_disconnect(rdc_handle);
else
rdc_stop_embedded(rdc_handle);
rdc_shutdown();
return result;
}