X-Git-Url: https://git.kernel.dk/?a=blobdiff_plain;f=memory.c;fp=memory.c;h=b208320c5d88cc619f7167626e3b7502ded07799;hb=d01612f3ae2515eb035d0c4ce954d8cb167a0a61;hp=e06cab2900a531e1515308fb166e626b6076274f;hpb=5054e67c7d87f25a9eb7b647adaee5e1c9c7cc5b;p=fio.git

diff --git a/memory.c b/memory.c
index e06cab29..b208320c 100644
--- a/memory.c
+++ b/memory.c
@@ -209,7 +209,7 @@ int allocate_io_mem(struct thread_data *td)
 
 	total_mem = td->orig_buffer_size;
 
-	if (td->o.odirect || td->o.mem_align ||
+	if (td->o.odirect || td->o.mem_align || td->o.oatomic ||
 	    (td->io_ops->flags & FIO_MEMALIGN)) {
 		total_mem += page_mask;
 		if (td->o.mem_align && td->o.mem_align > page_size)
@@ -240,7 +240,7 @@ void free_io_mem(struct thread_data *td)
 	unsigned int total_mem;
 
 	total_mem = td->orig_buffer_size;
-	if (td->o.odirect)
+	if (td->o.odirect || td->o.oatomic)
 		total_mem += page_mask;
 
 	if (td->o.mem_type == MEM_MALLOC)