rocr: Add queries for async scratch reclaim

Add support for these 2 new queries:
- HSA_AMD_AGENT_INFO_SCRATCH_LIMIT_MAX
  Maximum amount of scratch memory allowed on this agent

- HSA_AMD_AGENT_INFO_SCRATCH_LIMIT_CURRENT
  Current limit for scratch memory on this agent
Šī revīzija ir iekļauta:
David Yat Sin
2025-02-06 16:22:52 +00:00
revīziju iesūtīja Yat Sin, David
vecāks aa2f98e6f9
revīzija 107b48fb15
3 mainīti faili ar 30 papildinājumiem un 1 dzēšanām
@@ -415,6 +415,10 @@ hsa_status_t CpuAgent::GetInfo(hsa_agent_info_t attribute, void* value) const {
case HSA_AMD_AGENT_INFO_AQL_EXTENSIONS:
memset(value, 0, sizeof(uint8_t) * 8);
break;
case HSA_AMD_AGENT_INFO_SCRATCH_LIMIT_MAX:
case HSA_AMD_AGENT_INFO_SCRATCH_LIMIT_CURRENT:
*((uint64_t*)value) = 0;
break;
default:
return HSA_STATUS_ERROR_INVALID_ARGUMENT;
break;
@@ -1630,6 +1630,12 @@ hsa_status_t GpuAgent::GetInfo(hsa_agent_info_t attribute, void* value) const {
memset(value, 0, sizeof(uint8_t) * 8);
/* Not yet implemented */
break;
case HSA_AMD_AGENT_INFO_SCRATCH_LIMIT_MAX:
*((uint64_t*)value) = MaxScratchDevice();
break;
case HSA_AMD_AGENT_INFO_SCRATCH_LIMIT_CURRENT:
*((uint64_t*)value) = scratch_limit_async_threshold_;
break;
default:
return HSA_STATUS_ERROR_INVALID_ARGUMENT;
break;