AI/rocm-systems
2
0
Förgrening 0
Kod Ärenden Pull-förfrågningar Actions Packages Projekt Släpp Wiki Aktiviteter

Fixing ATT outdata in test run

Bläddra i källkod 
Change-Id: If5bb2ead00179503bfa0d9a118b9e04d3f810319


[ROCm/rocprofiler commit: e88cad28b6]
This commit is contained in:
gobhardw
2023-07-12 12:18:55 +05:30
committad av Gopesh Bhardwaj
förälder 8aff2862f1
incheckning c26b32b0f7
2 ändrade filer med 12 tillägg och 15 borttagningar
Download Patch File Download Diff File Expand all files Collapse all files
projects/rocprofiler/tests-v2/featuretests/profiler/apps/hip_kernels.cpp
-2
Visa fil
@@ -88,8 +88,6 @@ int LaunchVectorAddKernel() {
} }
if (errors != 0) { if (errors != 0) {
printf("FAILED: %d errors\n", errors); printf("FAILED: %d errors\n", errors);
} else {
printf("PASSED!\n");
} }
HIP_ASSERT(hipFree(deviceA)); HIP_ASSERT(hipFree(deviceA));
projects/rocprofiler/tests-v2/featuretests/profiler/profiler_gtest.cpp
+12 -13
Visa fil
@@ -587,8 +587,9 @@ TEST_F(LoadUnloadTest, WhenLoadingSecondTimeThenToolLoadsUnloadsSuccessfully) {
class ATTCollection : public ::testing::Test { class ATTCollection : public ::testing::Test {
public: public:
virtual void SetUp(){}; virtual void SetUp(){bCollected = false;};
virtual void TearDown(){}; virtual void TearDown(){};
static bool bCollected;
static void FlushCallback(const rocprofiler_record_header_t* record, static void FlushCallback(const rocprofiler_record_header_t* record,
const rocprofiler_record_header_t* end_record, const rocprofiler_record_header_t* end_record,
@@ -606,30 +607,25 @@ class ATTCollection : public ::testing::Test {
const char* kernel_name_c = static_cast<const char*>(malloc(name_length * sizeof(char))); const char* kernel_name_c = static_cast<const char*>(malloc(name_length * sizeof(char)));
rocprofiler_query_kernel_info(ROCPROFILER_KERNEL_NAME, att_tracer_record->kernel_id, rocprofiler_query_kernel_info(ROCPROFILER_KERNEL_NAME, att_tracer_record->kernel_id,
&kernel_name_c); &kernel_name_c);
int gpu_index = att_tracer_record->gpu_id.handle;
printf("Kernel Info:\n\tGPU Index: %d\n\tKernel Name: %s\n", gpu_index, kernel_name_c);
// Get the number of shader engine traces // Get the number of shader engine traces
int se_num = att_tracer_record->shader_engine_data_count; int se_num = att_tracer_record->shader_engine_data_count;
// iterate over each shader engine att trace // iterate over each shader engine att trace
for (int i = 0; i < se_num; i++) { for (int i = 0; i < se_num; i++) {
printf("\n\n-------------- shader_engine %d --------------\n\n", i); if (!att_tracer_record->shader_engine_data)
rocprofiler_record_se_att_data_t* se_att_trace = continue;
&att_tracer_record->shader_engine_data[i]; auto se_att_trace = att_tracer_record->shader_engine_data[i];
uint32_t size = se_att_trace->buffer_size; if (!se_att_trace.buffer_ptr || !se_att_trace.buffer_size)
const unsigned short* data_buffer_ptr = continue;
reinterpret_cast<const unsigned short*>(se_att_trace->buffer_ptr); bCollected = true;
// Print the buffer in terms of shorts (16 bits)
for (uint32_t j = 0; j < (size / sizeof(short)); j++)
printf("%04x\n", data_buffer_ptr[j]);
} }
} }
rocprofiler_next_record(record, &record, session_id, buffer_id); rocprofiler_next_record(record, &record, session_id, buffer_id);
} }
} }
}; };
bool ATTCollection::bCollected = false;
TEST_F(ATTCollection, WhenRunningATTItCollectsTraceData) { TEST_F(ATTCollection, WhenRunningATTItCollectsTraceData) {
int result = ROCPROFILER_STATUS_ERROR; int result = ROCPROFILER_STATUS_ERROR;
@@ -690,6 +686,9 @@ TEST_F(ATTCollection, WhenRunningATTItCollectsTraceData) {
// finalize att tracing by destroying rocprofiler object // finalize att tracing by destroying rocprofiler object
result = rocprofiler_finalize(); result = rocprofiler_finalize();
EXPECT_EQ(ROCPROFILER_STATUS_SUCCESS, result); EXPECT_EQ(ROCPROFILER_STATUS_SUCCESS, result);
//check if we got data from any shader engine
EXPECT_EQ(bCollected, true);
} }
/* /*