libhsakmt: add Streaming Performance Monitors APIs

Signed-off-by: Gang Ba <gaba@amd.com> Change-Id: I5c23a8dacf9bc50c740908aabe391432f2c7112e Signed-off-by: Gang Ba <gaba@amd.com> [ROCm/ROCR-Runtime commit: d675d1cce1]
2020-04-27 17:58:56 -04:00
parent efe478aed6
commit c0a0ada18b
5 changed files with 229 additions and 1 deletions
@@ -71,6 +71,9 @@ hsaKmtGetKernelDebugTrapVersionInfo;
 hsaKmtGetThunkDebugTrapVersionInfo;
 hsaKmtSetAddressWatch;
 hsaKmtClearAddressWatch;
+hsaKmtSPMAcquire;
+hsaKmtSPMRelease;
+hsaKmtSPMSetDestBuffer;

 local: *;
 };
@@ -0,0 +1,121 @@
+/*
+ * Copyright © 2014 Advanced Micro Devices, Inc.
+ *
+ * Permission is hereby granted, free of charge, to any person
+ * obtaining a copy of this software and associated documentation
+ * files (the "Software"), to deal in the Software without
+ * restriction, including without limitation the rights to use, copy,
+ * modify, merge, publish, distribute, sublicense, and/or sell copies
+ * of the Software, and to permit persons to whom the Software is
+ * furnished to do so, subject to the following conditions:
+ *
+ * The above copyright notice and this permission notice (including
+ * the next paragraph) shall be included in all copies or substantial
+ * portions of the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+ * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+ * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+ * NONINFRINGEMENT.  IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
+ * HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
+ * WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
+ * DEALINGS IN THE SOFTWARE.
+ */
+#include "libhsakmt.h"
+#include "linux/kfd_ioctl.h"
+#include <stdlib.h>
+#include <stdio.h>
+
+
+HSAKMT_STATUS HSAKMTAPI hsaKmtSPMAcquire(HSAuint32 PreferredNode)
+{
+	int ret;
+	struct kfd_ioctl_spm_args args = {0};
+	uint32_t gpu_id;
+
+	pr_debug("[%s]\n", __func__);
+
+	ret = validate_nodeid(PreferredNode, &gpu_id);
+	if (ret != HSAKMT_STATUS_SUCCESS) {
+		pr_err("[%s] invalid node ID: %d\n", __func__, PreferredNode);
+		return ret;
+	}
+
+	ret = HSAKMT_STATUS_SUCCESS;
+	args.destptr = 0;
+	args.buf_size = 0;
+	args.op = KFD_IOCTL_SPM_OP_ACQUIRE;
+	args.gpu_id = gpu_id;
+	args.bytes_copied = 0;
+
+	ret = kmtIoctl(kfd_fd, AMDKFD_IOC_RLC_SPM, &args);
+
+	return ret;
+}
+
+HSAKMT_STATUS HSAKMTAPI hsaKmtSPMSetDestBuffer(HSAuint32 PreferredNode,
+						HSAuint64 SizeInBytes,
+						HSAuint32 timeout,
+						void* DestMemoryAddress,
+						void **SPMMemoryAddress,
+						HSAuint32 *SizeCopied)
+{
+	int ret;
+	struct kfd_ioctl_spm_args args = {0};
+	uint32_t gpu_id;
+
+	ret = HSAKMT_STATUS_SUCCESS;
+
+	pr_debug("[%s]\n", __func__);
+	ret = validate_nodeid(PreferredNode, &gpu_id);
+
+	args.timeout    = timeout;
+	args.destptr    = (uint64_t)DestMemoryAddress;
+
+	args.buf_size   = SizeInBytes;
+	args.spmtptr	= 0;
+	args.op         = KFD_IOCTL_SPM_OP_SET_DEST_BUF;
+	args.gpu_id     = gpu_id;
+	args.bytes_copied = 0;
+
+	ret = kmtIoctl(kfd_fd, AMDKFD_IOC_RLC_SPM, &args);
+
+	if (ret != HSAKMT_STATUS_SUCCESS) {
+		*SPMMemoryAddress = NULL;
+		*SizeCopied = 0;
+		return ret;
+	}
+
+	*SPMMemoryAddress = (void *)args.spmtptr;
+	*SizeCopied = args.bytes_copied;
+
+	return ret;
+}
+
+HSAKMT_STATUS HSAKMTAPI hsaKmtSPMRelease(HSAuint32 PreferredNode)
+{
+	int ret = HSAKMT_STATUS_SUCCESS;
+	struct kfd_ioctl_spm_args args = {0};
+	uint32_t gpu_id;
+
+	pr_debug("[%s]\n", __func__);
+
+	ret = validate_nodeid(PreferredNode, &gpu_id);
+	if (ret != HSAKMT_STATUS_SUCCESS) {
+		pr_err("[%s] invalid node ID: %d\n", __func__, PreferredNode);
+		return ret;
+	}
+
+	args.destptr = 0;
+	args.buf_size = 0;
+	args.op = KFD_IOCTL_SPM_OP_RELEASE;
+	args.gpu_id = gpu_id;
+	args.bytes_copied = 0;
+
+	ret = kmtIoctl(kfd_fd, AMDKFD_IOC_RLC_SPM, &args);
+
+	return ret;
+}
+
+