SWDEV-179105 - Stream Operations: Add swqupport for Wait and Write

Change-Id: I9d6d0665d12b62fe705ce1569a0e8264a4f23ab7


[ROCm/hip commit: e89b228eac]
This commit is contained in:
Ravi C Akkenapally
2020-12-14 15:28:01 -08:00
rodzic 0604a66dbc
commit 3d9b4cc2c2
9 zmienionych plików z 555 dodań i 2 usunięć
+12 -2
Wyświetl plik
@@ -262,11 +262,21 @@ hipError_t ihipMemcpy(void* dst, const void* src, size_t sizeBytes, hipMemcpyKin
hipError_t hipExtMallocWithFlags(void** ptr, size_t sizeBytes, unsigned int flags) {
HIP_INIT_API(hipExtMallocWithFlags, ptr, sizeBytes, flags);
if (flags != hipDeviceMallocDefault && flags != hipDeviceMallocFinegrained) {
unsigned int ihipFlags = 0;
if (flags == hipDeviceMallocDefault) {
ihipFlags = 0;
} else if (flags == hipDeviceMallocFinegrained) {
ihipFlags = CL_MEM_SVM_ATOMICS;
} else if (flags == hipMallocSignalMemory) {
ihipFlags = CL_MEM_SVM_ATOMICS | CL_MEM_SVM_FINE_GRAIN_BUFFER | ROCCLR_MEM_HSA_SIGNAL_MEMORY;
if (sizeBytes != 8) {
HIP_RETURN(hipErrorInvalidValue);
}
} else {
HIP_RETURN(hipErrorInvalidValue);
}
HIP_RETURN(ihipMalloc(ptr, sizeBytes, (flags & hipDeviceMallocFinegrained)? CL_MEM_SVM_ATOMICS: 0), (ptr != nullptr)? *ptr : nullptr);
HIP_RETURN(ihipMalloc(ptr, sizeBytes, ihipFlags), (ptr != nullptr)? *ptr : nullptr);
}
hipError_t hipMalloc(void** ptr, size_t sizeBytes) {