SWDEV-554608 - Add hipHostRegisterIoMemory for hipHostRegister (#962)

* SWDEV-554608 - Add hipHostRegisterIoMemory for hipHostRegister

* SWDEV-554608 - Add hipHostRegisterIoMemory for hipHostRegister

* SWDEV-554174 Added hipHostRegisterIoMemory flag in test cases

* SWDEV-554174 : Did formatting corrections

* SWDEV-554608 - set HSA_AMD_MEMORY_POOL_UNCACHED_FLAG if IoMemory is set

* SWDEV-554608 - set HSA_AMD_MEMORY_POOL_UNCACHED_FLAG if IoMemory is set

* SWDEV-554608 - Add hipHostRegisterIoMemory for hipHostRegister

---------

Co-authored-by: Anavena Venkatesh <Anavena.Venkatesh@amd.com>
Co-authored-by: Rambabu Swargam <rambabu.swargam@amd.com>
This commit is contained in:
Jimbo
2025-10-22 20:25:59 -04:00
committed by GitHub
parent 79076c4ad5
commit 37f2be9140
11 changed files with 70 additions and 7 deletions
+6 -1
View File
@@ -1287,7 +1287,7 @@ hipError_t hipHostGetFlags(unsigned int* flagsPtr, void* hostPtr) {
hipError_t ihipHostRegister(void* hostPtr, size_t sizeBytes, unsigned int flags) {
if (hostPtr == nullptr || sizeBytes == 0 ||
flags & ~(hipHostRegisterPortable | hipHostRegisterMapped | hipExtHostRegisterCoarseGrained |
hipExtHostRegisterUncached)) {
hipExtHostRegisterUncached | hipHostRegisterIoMemory)) {
return hipErrorInvalidValue;
} else {
unsigned int memFlags = CL_MEM_USE_HOST_PTR | CL_MEM_SVM_ATOMICS;
@@ -1296,6 +1296,11 @@ hipError_t ihipHostRegister(void* hostPtr, size_t sizeBytes, unsigned int flags)
return hipErrorInvalidValue;
}
memFlags |= ROCCLR_MEM_HSA_UNCACHED;
} else if (flags & hipHostRegisterIoMemory) {
if (IS_WINDOWS) {
return hipErrorInvalidValue;
}
memFlags |= ROCCLR_MEM_IO_MEMORY;
}
amd::Memory* mem =