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

adding support for scratch memory

Browse Source 
Signed-off-by: Moses Reuben <moses.reuben@amd.com>
This commit is contained in:
Moses Reuben
2015-06-09 05:37:05 +03:00
committed by Ben Goz
parent 2e76017278
commit 29c083f695
5 changed files with 98 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/memory.c
+9 -1
View File
@@ -124,7 +124,7 @@ hsaKmtAllocMemory(
if ((SizeInBytes & (page_size-1)) && !MemFlags.ui32.GDSMemory)
return HSAKMT_STATUS_INVALID_PARAMETER;
if (MemFlags.ui32.HostAccess && !MemFlags.ui32.NonPaged) {
if (MemFlags.ui32.HostAccess && !MemFlags.ui32.NonPaged && !MemFlags.ui32.Scratch) {
err = posix_memalign(MemoryAddress, page_size, SizeInBytes);
if (err != 0)
return HSAKMT_STATUS_NO_MEMORY;
@@ -149,6 +149,14 @@ hsaKmtAllocMemory(
return HSAKMT_STATUS_SUCCESS;
}
if (MemFlags.ui32.HostAccess && MemFlags.ui32.Scratch ) {
*MemoryAddress = fmm_allocate_scratch(gpu_id, SizeInBytes);
if (*MemoryAddress == NULL)
return HSAKMT_STATUS_NO_MEMORY;
return HSAKMT_STATUS_SUCCESS;
}
return HSAKMT_STATUS_INVALID_PARAMETER;
}