浏览代码

We can push data outside CPU memory when there are several NUMA nodes

Samuel Thibault 5 年之前
父节点
当前提交
9c500272a9
共有 2 个文件被更改,包括 3 次插入1 次删除
  1. 2 1
      src/datawizard/memalloc.c
  2. 1 0
      tests/Makefile.am

+ 2 - 1
src/datawizard/memalloc.c

@@ -556,7 +556,8 @@ static size_t try_to_throw_mem_chunk(struct _starpu_mem_chunk *mc, unsigned node
 		return 0;
 
 	/* This data cannnot be pushed outside CPU memory */
-	if (!handle->ooc && starpu_node_get_kind(node) == STARPU_CPU_RAM)
+	if (!handle->ooc && starpu_node_get_kind(node) == STARPU_CPU_RAM
+		&& starpu_memory_nodes_get_numa_count() == 1)
 		return 0;
 
 	if (diduse_barrier && !mc->diduse)

+ 1 - 0
tests/Makefile.am

@@ -155,6 +155,7 @@ myPROGRAMS +=					\
 	datawizard/deps				\
 	datawizard/user_interaction_implicit	\
 	datawizard/interfaces/copy_interfaces	\
+	datawizard/numa_overflow		\
 	datawizard/locality			\
 	datawizard/variable_size		\
 	errorcheck/starpu_init_noworker		\