2 files changed, 15 insertions, 0 deletions
diff --git a/lib/drmtest.c b/lib/drmtest.c
index 5caf71b2..0f021902 100644
--- a/lib/drmtest.c
+++ b/lib/drmtest.c
@@ -289,6 +289,20 @@ uint64_t gem_mappable_aperture_size(void)
 	return pci_dev->regions[bar].size;
 }
 
+int gem_madvise(int fd, uint32_t handle, int state)
+{
+	struct drm_i915_gem_madvise madv;
+	int ret;
+
+	madv.handle = handle;
+	madv.madv = state;
+	madv.retained = 1;
+	ret = drmIoctl(fd, DRM_IOCTL_I915_GEM_MADVISE, &madv);
+	assert(ret == 0);
+
+	return madv.retained;
+}
+
 /* signal interrupt helpers */
 static pid_t signal_helper = -1;
 long long int sig_stat;
diff --git a/lib/drmtest.h b/lib/drmtest.h
index a8438abd..9f194448 100644
--- a/lib/drmtest.h
+++ b/lib/drmtest.h
@@ -51,6 +51,7 @@ uint32_t gem_create(int fd, int size);
 void *gem_mmap(int fd, uint32_t handle, int size, int prot);
 uint64_t gem_aperture_size(int fd);
 uint64_t gem_mappable_aperture_size(void);
+int gem_madvise(int fd, uint32_t handle, int state);
 
 /* generally useful helpers */
 void drmtest_fork_signal_helper(void);