|
|
@@ -746,11 +746,11 @@ static struct _starpu_chunk *_starpu_new_chunk(unsigned dst_node, int flags)
|
|
|
/* Return whether we should use our suballocator */
|
|
|
static int _starpu_malloc_should_suballoc(unsigned dst_node, size_t size, int flags)
|
|
|
{
|
|
|
- return size <= CHUNK_ALLOC_MAX &&
|
|
|
+ return (size <= CHUNK_ALLOC_MAX &&
|
|
|
(starpu_node_get_kind(dst_node) == STARPU_CUDA_RAM
|
|
|
|| (starpu_node_get_kind(dst_node) == STARPU_CPU_RAM
|
|
|
&& _starpu_malloc_should_pin(flags))
|
|
|
- )
|
|
|
+ ))
|
|
|
|| starpu_node_get_kind(dst_node) == STARPU_FPGA_RAM;
|
|
|
}
|
|
|
|