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

Adding support to hsaKmtMapMemoryToGPUNodes

Browse Source 
Change-Id: Iab6222402a43c3cd31b0efc5a316a6482986258e
Signed-off-by: Ben Goz <ben.goz@amd.com>


[ROCm/ROCR-Runtime commit: 7070f7ec5e]
This commit is contained in:
Ben Goz
2016-01-28 15:33:56 +02:00
parent 60bbf00fb1
commit 18aab410cc
6 changed files with 76 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files
projects/rocr-runtime/src/memory.c
+32 -13
View File
@@ -212,21 +212,10 @@ hsaKmtRegisterMemoryToNodes(
{
CHECK_KFD_OPEN();
uint32_t *gpu_id_array;
unsigned i;
HSAKMT_STATUS ret = HSAKMT_STATUS_SUCCESS;
if (NumberOfNodes == 0 || NodeArray == NULL)
return HSAKMT_STATUS_INVALID_PARAMETER;
/* Translate Node IDs to gpu_ids */
gpu_id_array = malloc(NumberOfNodes * sizeof(uint32_t));
if (gpu_id_array == NULL)
return HSAKMT_STATUS_NO_MEMORY;
for (i = 0; i < NumberOfNodes; i++) {
ret = validate_nodeid(NodeArray[i], &gpu_id_array[i]);
if (ret != HSAKMT_STATUS_SUCCESS)
break;
}
ret = validate_nodeid_array(&gpu_id_array,
NumberOfNodes, NodeArray);
if (ret == HSAKMT_STATUS_SUCCESS)
ret = fmm_register_memory(MemoryAddress, MemorySizeInBytes,
@@ -269,6 +258,36 @@ hsaKmtMapMemoryToGPU(
return HSAKMT_STATUS_ERROR;
}
HSAKMT_STATUS
HSAKMTAPI
hsaKmtMapMemoryToGPUNodes(
void* MemoryAddress, //IN (page-aligned)
HSAuint64 MemorySizeInBytes, //IN (page-aligned)
HSAuint64* AlternateVAGPU, //OUT (page-aligned)
HsaMemMapFlags MemMapFlags, //IN
HSAuint64 NumberOfNodes, //IN
HSAuint32* NodeArray //IN
)
{
uint32_t *gpu_id_array;
HSAKMT_STATUS ret;
ret = validate_nodeid_array(&gpu_id_array,
NumberOfNodes, NodeArray);
if (ret != HSAKMT_STATUS_SUCCESS) {
free(gpu_id_array);
return ret;
}
ret = fmm_register_memory(MemoryAddress, MemorySizeInBytes,
gpu_id_array, NumberOfNodes);
if (ret != HSAKMT_STATUS_SUCCESS)
return ret;
return fmm_map_to_gpu(MemoryAddress,
MemorySizeInBytes, AlternateVAGPU);
}
HSAKMT_STATUS
HSAKMTAPI
hsaKmtUnmapMemoryToGPU(