gobhardw
33 строки
1.1 KiB
C++
33 строки
1.1 KiB
C++
#include <gtest/gtest.h>
|
|
#include "src/core/hardware/hsa_info.h"
|
|
//#include "src/core/hsa/hsa_common.h"
|
|
|
|
// Entry Point for Gtests Infra
|
|
|
|
int main(int argc, char** argv) {
|
|
testing::InitGoogleTest(&argc, argv);
|
|
testing::FLAGS_gtest_death_test_style = "threadsafe";
|
|
// Add line below to disable any problematic test
|
|
testing::GTEST_FLAG(filter) =
|
|
"-OpenMPTest.*:ProfilerSPMTest*:ProfilerMQTest*:ProfilerMPTest*:MPITest*";
|
|
// Disable ATT test fir gfx10 GPUs until its supported
|
|
hsa_init();
|
|
// iterate for gpu's
|
|
hsa_iterate_agents(
|
|
[](hsa_agent_t agent, void*) {
|
|
char gpu_name[64];
|
|
hsa_agent_get_info(agent, HSA_AGENT_INFO_NAME, gpu_name);
|
|
std::string gfx_name = gpu_name;
|
|
if (gfx_name.find("gfx10") != std::string::npos) {
|
|
testing::GTEST_FLAG(filter) =
|
|
"-ATTCollection.*:OpenMPTest.*:ProfilerSPMTest*:ProfilerMQTest:*ProfilerMPTest*:"
|
|
"MPITest*";
|
|
}
|
|
return HSA_STATUS_SUCCESS;
|
|
},
|
|
nullptr);
|
|
// hsa_shut_down(); // Waiting for hsa_shutdown bug to fix
|
|
// Append filter above to disable any problematic test
|
|
return RUN_ALL_TESTS();
|
|
}
|