Adding support for mGPU

Change-Id: I5ed184e6a58b38d9dde48867f14513d161cf41a9 Signed-off-by: Ben Goz <ben.goz@amd.com>
2015-12-23 17:23:25 +02:00
parent 53b208adf2
commit ea0f9d2a0b
7 changed files with 192 additions and 19 deletions
@@ -195,8 +195,21 @@ hsaKmtRegisterMemory(
 	HSAuint64	MemorySizeInBytes	/* IN (page-aligned) */
 )
 {
+	uint32_t *NodesArray;
+	uint32_t NodesArraySize;
+
 	CHECK_KFD_OPEN();

+	/*
+	 * Build NodesArray from all dGPU nodes.
+	 */
+	NodesArraySize = fmm_build_nodes_array(&NodesArray, NULL, 0);
+	if (!NodesArray)
+		return HSAKMT_STATUS_INVALID_PARAMETER;
+
+	if (fmm_register_memory(MemoryAddress, MemorySizeInBytes,
+			NodesArray, NodesArraySize) != 0)
+		return HSAKMT_STATUS_ERROR;
 	return HSAKMT_STATUS_SUCCESS;
 }

@@ -208,6 +221,8 @@ hsaKmtDeregisterMemory(
 {
 	CHECK_KFD_OPEN();

+	fmm_deregister_memory(MemoryAddress);
+
 	return HSAKMT_STATUS_SUCCESS;
 }