From 4d7b0990e404e7ef06b579bfce99cee43c0747fb Mon Sep 17 00:00:00 2001 From: Felix Kuehling Date: Fri, 12 Jul 2019 17:48:27 -0400 Subject: [PATCH] libhsakmt: Sanity check node_id for NUMA binding Ignore requests to bind to invalid NUMA nodes. This affects only legacy applications (such as KFDTest) that allocate system memory as paged memory with a GPU node ID. Change-Id: I81e514af6d0c1ab2ed5229adeeca1fa0ab2a0685 Signed-off-by: Felix Kuehling --- src/fmm.c | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/src/fmm.c b/src/fmm.c index 4aced94956..660ff234b3 100644 --- a/src/fmm.c +++ b/src/fmm.c @@ -1469,6 +1469,12 @@ static int bind_mem_to_numa(uint32_t node_id, void *mem, num_node = numa_num_task_nodes(); + /* Ignore binding requests to invalid nodes IDs */ + if (node_id >= (unsigned)num_node) { + pr_warn("node_id %d >= num_node %d\n", node_id, num_node); + return 0; + } + if (num_node > 1) { node_mask = numa_bitmask_alloc(num_node); if (!node_mask)