Add support for hsaKmtRegisterGraphicsHandleToNodes

Change-Id: I6fd7154dea78188480d5cb89ac237bad572356c4 [ROCm/ROCR-Runtime commit: 61ec3df2f9]
2016-03-05 01:12:40 -05:00
parent c32a504b59
commit 029002d073
6 changed files with 208 additions and 20 deletions
@@ -233,6 +233,33 @@ hsaKmtRegisterMemoryToNodes(
 	return ret;
 }

+HSAKMT_STATUS
+HSAKMTAPI
+hsaKmtRegisterGraphicsHandleToNodes(
+	HSAuint64	GraphicsResourceHandle,	       /* IN */
+	HsaGraphicsResourceInfo *GraphicsResourceInfo, /* OUT */
+	HSAuint64	NumberOfNodes,		       /* IN */
+	HSAuint32*	NodeArray		       /* IN */
+)
+{
+	CHECK_KFD_OPEN();
+	uint32_t *gpu_id_array;
+	HSAKMT_STATUS ret = HSAKMT_STATUS_SUCCESS;
+
+	ret = validate_nodeid_array(&gpu_id_array,
+			NumberOfNodes, NodeArray);
+
+	if (ret == HSAKMT_STATUS_SUCCESS) {
+		ret = fmm_register_graphics_handle(
+			GraphicsResourceHandle, GraphicsResourceInfo,
+			gpu_id_array,NumberOfNodes*sizeof(uint32_t));
+		if (ret != HSAKMT_STATUS_SUCCESS)
+			free(gpu_id_array);
+	}
+
+	return ret;
+}
+
 HSAKMT_STATUS
 HSAKMTAPI
 hsaKmtDeregisterMemory(