gem_cpu_reloc: Use the mappable aperture size!

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

1 files changed, 6 insertions, 2 deletions

diff --git a/tests/gem_cpu_reloc.c b/tests/gem_cpu_reloc.c
index ceb64054..a83eea6d 100644
--- a/tests/gem_cpu_reloc.c
+++ b/tests/gem_cpu_reloc.c
@@ -146,9 +146,13 @@ int main(int argc, char **argv)
 	if (intel_gen(noop) >= 6)
 		use_blt = I915_EXEC_BLT;
 
-	aper_size = gem_aperture_size(fd);
-	count = aper_size / 4096 * 2;
+	aper_size = gem_mappable_aperture_size();
+	if (intel_get_total_ram_mb() < aper_size / (1024*1024) * 2) {
+		fprintf(stderr, "not enough mem to run test\n");
+		return 77;
+	}
 
+	count = aper_size / 4096 * 2;
 	handles = malloc (count * sizeof(uint32_t));
 	assert(handles);