Преглед на файлове

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])
 	{