From 0f241d406114a392a082791c2d09809f7635c87e Mon Sep 17 00:00:00 2001 From: David Yat Sin Date: Thu, 12 Sep 2024 17:15:48 +0000 Subject: [PATCH] hsakmt: Add debug prints to trace mem allocations Add extra debug prints to trace memory alloc and register Change-Id: I03d8d7d415565916a8336db6e7063bb7d4cb9102 --- libhsakmt/src/memory.c | 13 ++++++++----- 1 file changed, 8 insertions(+), 5 deletions(-) diff --git a/libhsakmt/src/memory.c b/libhsakmt/src/memory.c index 294f209715..4e4a267a3b 100644 --- a/libhsakmt/src/memory.c +++ b/libhsakmt/src/memory.c @@ -171,6 +171,7 @@ HSAKMT_STATUS HSAKMTAPI hsaKmtAllocMemoryAlign(HSAuint32 PreferredNode, return HSAKMT_STATUS_NO_MEMORY; } + pr_debug("[%s] node %d address %p size %lu from scratch\n", __func__, PreferredNode, *MemoryAddress, SizeInBytes); return HSAKMT_STATUS_SUCCESS; } @@ -193,6 +194,7 @@ HSAKMT_STATUS HSAKMTAPI hsaKmtAllocMemoryAlign(HSAuint32 PreferredNode, return HSAKMT_STATUS_ERROR; } + pr_debug("[%s] node %d address %p size %lu from host\n", __func__, PreferredNode, *MemoryAddress, SizeInBytes); return HSAKMT_STATUS_SUCCESS; } @@ -213,6 +215,7 @@ HSAKMT_STATUS HSAKMTAPI hsaKmtAllocMemoryAlign(HSAuint32 PreferredNode, return HSAKMT_STATUS_NO_MEMORY; } + pr_debug("[%s] node %d address %p size %lu from device\n", __func__, PreferredNode, *MemoryAddress, SizeInBytes); return HSAKMT_STATUS_SUCCESS; } @@ -261,7 +264,7 @@ HSAKMT_STATUS HSAKMTAPI hsaKmtRegisterMemory(void *MemoryAddress, { CHECK_KFD_OPEN(); - pr_debug("[%s] address %p\n", __func__, MemoryAddress); + pr_debug("[%s] address %p size %lu\n", __func__, MemoryAddress, MemorySizeInBytes); if (!hsakmt_is_dgpu) /* TODO: support mixed APU and dGPU configurations */ @@ -280,8 +283,8 @@ HSAKMT_STATUS HSAKMTAPI hsaKmtRegisterMemoryToNodes(void *MemoryAddress, uint32_t *gpu_id_array; HSAKMT_STATUS ret = HSAKMT_STATUS_SUCCESS; - pr_debug("[%s] address %p number of nodes %lu\n", - __func__, MemoryAddress, NumberOfNodes); + pr_debug("[%s] address %p size %lu number of nodes %lu\n", + __func__, MemoryAddress, MemorySizeInBytes, NumberOfNodes); if (!hsakmt_is_dgpu) /* TODO: support mixed APU and dGPU configurations */ @@ -309,8 +312,8 @@ HSAKMT_STATUS HSAKMTAPI hsaKmtRegisterMemoryWithFlags(void *MemoryAddress, CHECK_KFD_OPEN(); HSAKMT_STATUS ret = HSAKMT_STATUS_SUCCESS; - pr_debug("[%s] address %p\n", - __func__, MemoryAddress); + pr_debug("[%s] address %p size %lu\n", + __func__, MemoryAddress, MemorySizeInBytes); if (MemFlags.ui32.ExtendedCoherent && MemFlags.ui32.CoarseGrain) return HSAKMT_STATUS_INVALID_PARAMETER;