Commit 3d4cf2be authored by zhugzh1's avatar zhugzh1
Browse files

Synchronize Hygon modification

parent 838f4927
......@@ -300,7 +300,7 @@ void RocmBandwidthTest::BuildBufferList() {
if ((req_copy_all_bidir_ == REQ_COPY_ALL_BIDIR) ||
(req_copy_all_unidir_ == REQ_COPY_ALL_UNIDIR)) {
if (idx == 16) {
if (idx == 19) {
size_list_.push_back(SIZE_LIST[idx]);
}
}
......
......@@ -101,6 +101,7 @@ hsa_status_t MemPoolInfo(hsa_amd_memory_pool_t pool, void* data) {
bool is_kernarg = (HSA_AMD_MEMORY_POOL_GLOBAL_FLAG_KERNARG_INIT & flag);
bool is_fine_grained = (HSA_AMD_MEMORY_POOL_GLOBAL_FLAG_FINE_GRAINED & flag);
bool is_coarse_grained =(HSA_AMD_MEMORY_POOL_GLOBAL_FLAG_COARSE_GRAINED & flag);
bool is_ext_fine_grained = (HSA_AMD_MEMORY_POOL_GLOBAL_FLAG_EXTENDED_SCOPE_FINE_GRAINED & flag);
// Update the pool handle for system memory if kernarg is true
if (is_kernarg) {
asyncDrvr->sys_pool_ = pool;
......@@ -129,6 +130,9 @@ hsa_status_t MemPoolInfo(hsa_amd_memory_pool_t pool, void* data) {
// coarse-grained memory pools if agent is GPU. Default
// is to skip fine-grained memory pools
if (agent_info.device_type_ == HSA_DEVICE_TYPE_GPU) {
if (is_ext_fine_grained == true) {
return HSA_STATUS_SUCCESS;
}
if (asyncDrvr->skip_gpu_coarse_grain_ != NULL) {
if (is_fine_grained == false) {
return HSA_STATUS_SUCCESS;
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment