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

Update Private Segment Size parameter of the dispatch packet

Browse Source 
[git-p4: depot-paths = "//depot/stg/hsa/drivers/hsa/runtime/": change = 1254638]


[ROCm/ROCR-Runtime commit: b93946790d]
This commit is contained in:
Ramesh Errabolu (xN/A) TX
2016-04-05 14:03:33 -05:00
parent 2caa405817
commit cae56040bb
4 changed files with 22 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
projects/rocr-runtime/samples/common/hsa_test.cpp
+11 -3
View File
@@ -151,16 +151,16 @@ HsaTest::Kernel::Kernel(hsa_agent_t agent, std::string hsail_text)
HsaTest::Kernel::~Kernel() { Cleanup(); }
uint64_t HsaTest::Kernel::GetCodeHandle(const char* kernel_name) {
hsa_executable_symbol_t kernel_symbol = {0};
kernel_symbol_ = {0};
if (HSA_STATUS_SUCCESS != hsa_executable_get_symbol(executable_, NULL,
kernel_name, agent_, 0,
&kernel_symbol)) {
&kernel_symbol_)) {
return 0;
}
uint64_t code_handle = 0;
if (HSA_STATUS_SUCCESS !=
hsa_executable_symbol_get_info(kernel_symbol,
hsa_executable_symbol_get_info(kernel_symbol_,
HSA_EXECUTABLE_SYMBOL_INFO_KERNEL_OBJECT,
&code_handle)) {
return 0;
@@ -169,6 +169,14 @@ uint64_t HsaTest::Kernel::GetCodeHandle(const char* kernel_name) {
return code_handle;
}
hsa_status_t HsaTest::Kernel::GetScratchSize(uint32_t* size) {
hsa_status_t status;
status = hsa_executable_symbol_get_info(kernel_symbol_,
HSA_EXECUTABLE_SYMBOL_INFO_KERNEL_PRIVATE_SEGMENT_SIZE, size);
return status;
}
void HsaTest::Kernel::Initialize() {
CreateProgramFromHsailFile();
CreateCodeObjectAndExecutable();