소스 검색

Fix testing memory allocation flags

Samuel Thibault 9 년 전
부모
커밋
57bd8f7ea5
1개의 변경된 파일2개의 추가작업 그리고 2개의 파일을 삭제
  1. 2 2
      src/datawizard/memory_manager.c

+ 2 - 2
src/datawizard/memory_manager.c

@@ -76,7 +76,7 @@ int starpu_memory_allocate(unsigned node, size_t size, int flags)
 	int ret;
 
 	STARPU_PTHREAD_MUTEX_LOCK(&lock_nodes[node]);
-	if (flags == STARPU_MEMORY_WAIT)
+	if (flags & STARPU_MEMORY_WAIT)
 	{
 		while (used_size[node] + size > global_size[node])
 		{
@@ -92,7 +92,7 @@ int starpu_memory_allocate(unsigned node, size_t size, int flags)
 		used_size[node] += size;
 		ret = 0;
 	}
-	else if (flags == STARPU_MEMORY_OVERFLOW
+	else if (flags & STARPU_MEMORY_OVERFLOW
 			|| global_size[node] == 0
 			|| used_size[node] + size <= global_size[node])
 	{