AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Locate kernel directory from device name.

Browse Source 
Search child directories when locating device code.

Change-Id: I51515f002ad60878a2be0b6e9ee6416c67a1d799
This commit is contained in:
Sean Keely
2021-06-16 19:54:06 -05:00
parent 9e53cab613
commit 74bcd6ee90
2 changed files with 13 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
rocrtst/common/base_rocr_utils.cc
+5
View File
@@ -306,6 +306,11 @@ hsa_status_t LoadKernelFromObjFile(BaseRocR* test, hsa_agent_t* agent) {
hsa_file_t file_handle = open(obj_file.c_str(), O_RDONLY);
if (file_handle == -1) {
obj_file = "./" + test->get_agent_name() + "/" + test->kernel_file_name();
file_handle = open(obj_file.c_str(), O_RDONLY);
}
if (file_handle == -1) {
std::cout << "failed to open " << obj_file.c_str() << " at line "
<< __LINE__ << ", file: " << __FILE__ << std::endl;
rocrtst/samples/binary_search/binary_search.cc
+8
View File
@@ -378,6 +378,14 @@ hsa_status_t LoadKernelFromObjFile(BinarySearch* bs) {
hsa_file_t file_handle = open(bs->kernel_file_name.c_str(), O_RDONLY);
if (file_handle == -1) {
char agent_name[64];
err = hsa_agent_get_info(bs->gpu_dev, HSA_AGENT_INFO_NAME, agent_name);
RET_IF_HSA_ERR(err);
std::string fileName = std::string("./") + agent_name + "/" + bs->kernel_file_name;
hsa_file_t file_handle = open(fileName.c_str(), O_RDONLY);
}
if (file_handle == -1) {
std::cout << "failed to open " << bs->kernel_file_name.c_str() <<
" at line " << __LINE__ << ", errno: " << errno << std::endl;