ECR #333755 - HSA samples changes for dGPU. All passed on gfx802.
[git-p4: depot-paths = "//depot/stg/hsa/drivers/hsa/runtime/": change = 1186398]
[ROCm/ROCR-Runtime commit: a32c2b9854]
This commit is contained in:
@@ -113,6 +113,13 @@ bool HSA_UTIL::HsaInit()
|
||||
base_timer.StopTimer(base_setup_time_idx);
|
||||
#endif
|
||||
|
||||
|
||||
//hsa_region_t local_kernarg_region;
|
||||
kernarg_region.handle = 0;
|
||||
hsa_agent_iterate_regions(device, get_kernarg, &kernarg_region);
|
||||
err = (kernarg_region.handle== 0) ? HSA_STATUS_ERROR : HSA_STATUS_SUCCESS;
|
||||
check(Finding a kernarg memory region, err);
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
@@ -153,18 +160,22 @@ double HSA_UTIL::Run(int dim, int group_x, int group_y, int group_z, int s_size,
|
||||
/*
|
||||
* Find a memory region that supports kernel arguments.
|
||||
*/
|
||||
hsa_region_t local_kernarg_region;
|
||||
local_kernarg_region.handle = 0;
|
||||
|
||||
hsa_agent_iterate_regions(device, get_kernarg, &local_kernarg_region);
|
||||
err = (local_kernarg_region.handle== 0) ? HSA_STATUS_ERROR : HSA_STATUS_SUCCESS;
|
||||
|
||||
/*
|
||||
kernarg_region.handle = 0;
|
||||
|
||||
hsa_agent_iterate_regions(device, get_kernarg, &kernarg_region);
|
||||
err = (kernarg_region.handle== 0) ? HSA_STATUS_ERROR : HSA_STATUS_SUCCESS;
|
||||
check(Finding a kernarg memory region, err);
|
||||
void* local_kernel_arg_buffer = NULL;
|
||||
|
||||
*/
|
||||
|
||||
void* local_kernel_arg_buffer = NULL;
|
||||
/*
|
||||
* Allocate the kernel argument buffer from the correct region.
|
||||
*/
|
||||
err = hsa_memory_allocate(local_kernarg_region, kernel_args_size, &local_kernel_arg_buffer);
|
||||
err = hsa_memory_allocate(kernarg_region, kernel_args_size, &local_kernel_arg_buffer);
|
||||
check(Allocating kernel argument memory buffer, err);
|
||||
memcpy(local_kernel_arg_buffer, kernel_args, kernel_args_size);
|
||||
local_dispatch_packet.kernarg_address = local_kernel_arg_buffer;
|
||||
|
||||
@@ -30,6 +30,7 @@ class HSA_UTIL{
|
||||
hsa_status_t err;
|
||||
uint32_t queue_size;
|
||||
hsa_agent_t device;
|
||||
hsa_region_t kernarg_region;
|
||||
|
||||
char hail_file_name[128];
|
||||
char hsa_kernel_name[128];
|
||||
|
||||
Reference in New Issue
Block a user