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

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

Browse Source 
Change-Id: I9d6d0665d12b62fe705ce1569a0e8264a4f23ab7
This commit is contained in:
Ravi C Akkenapally
2020-12-14 15:28:01 -08:00
parent f727a048d3
commit fd0f49503f
9 changed files with 555 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
hipamd/rocclr/hip_memory.cpp
+12 -2
View File
@@ -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) {