igt/gem_wait: munmap the batch along all paths

Move the munmap() to the common path to ensure it is released at the end
of each test.

Signed-off-by: Chris Wilson <chris@chris-wilson.co.uk>

1 files changed, 1 insertions, 1 deletions

diff --git a/tests/gem_wait.c b/tests/gem_wait.c
index 0ecb92f4..b4127de8 100644
--- a/tests/gem_wait.c
+++ b/tests/gem_wait.c
@@ -169,7 +169,6 @@ static void basic(int fd, unsigned engine, unsigned flags)
 			__sync_synchronize();
 			timeout = 1;
 		}
-		munmap(batch, 4096);
 
 		memset(&tv, 0, sizeof(tv));
 		while (__gem_wait(fd, &wait) == -ETIME)
@@ -222,6 +221,7 @@ static void basic(int fd, unsigned engine, unsigned flags)
 	}
 
 	gem_close(fd, obj.handle);
+	munmap(batch, 4096);
 }
 
 igt_main