@@ -78,35 +78,6 @@ static void reserve_simple(int fd)
 	intel_allocator_close(ialh);
 }
 
-static void purge_bb(int fd)
-{
-	struct buf_ops *bops;
-	struct intel_buf *buf;
-	struct intel_bb *ibb;
-	uint64_t offset0, offset1;
-
-	bops = buf_ops_create(fd);
-	buf = intel_buf_create(bops, 512, 512, 32, 0, I915_TILING_NONE,
-			       I915_COMPRESSION_NONE);
-	ibb = intel_bb_create(fd, 4096);
-	intel_bb_set_debug(ibb, true);
-
-	intel_bb_add_intel_buf(ibb, buf, false);
-	offset0 = buf->addr.offset;
-
-	intel_bb_reset(ibb, true);
-	buf->addr.offset = INTEL_BUF_INVALID_ADDRESS;
-
-	intel_bb_add_intel_buf(ibb, buf, false);
-	offset1 = buf->addr.offset;
-
-	igt_assert(offset0 == offset1);
-
-	intel_buf_destroy(buf);
-	intel_bb_destroy(ibb);
-	buf_ops_destroy(bops);
-}
-
 static void reserve(int fd, uint8_t type)
 {
 	struct intel_allocator *ial;

@@ -508,9 +479,6 @@ igt_main
 	igt_subtest_f("reserve-simple")
 		reserve_simple(fd);
 
-	igt_subtest_f("purge-bb")
-		purge_bb(fd);
-
 	igt_subtest_f("print")
 		basic_alloc(fd, 1UL << 2, INTEL_ALLOCATOR_RANDOM);
 

@@ -143,6 +143,33 @@ static void reset_bb(struct buf_ops *bops)
 	intel_bb_destroy(ibb);
 }
 
+static void purge_bb(struct buf_ops *bops)
+{
+	int i915 = buf_ops_get_fd(bops);
+	struct intel_buf *buf;
+	struct intel_bb *ibb;
+	uint64_t offset0, offset1;
+
+	buf = intel_buf_create(bops, 512, 512, 32, 0, I915_TILING_NONE,
+			       I915_COMPRESSION_NONE);
+	ibb = intel_bb_create(i915, 4096);
+	intel_bb_set_debug(ibb, true);
+
+	intel_bb_add_intel_buf(ibb, buf, false);
+	offset0 = buf->addr.offset;
+
+	intel_bb_reset(ibb, true);
+	buf->addr.offset = INTEL_BUF_INVALID_ADDRESS;
+
+	intel_bb_add_intel_buf(ibb, buf, false);
+	offset1 = buf->addr.offset;
+
+	igt_assert(offset0 == offset1);
+
+	intel_buf_destroy(buf);
+	intel_bb_destroy(ibb);
+}
+
 static void simple_bb(struct buf_ops *bops, bool use_context)
 {
 	int i915 = buf_ops_get_fd(bops);

@@ -240,13 +267,15 @@ static void check_canonical(struct buf_ops *bops)
 	/* Twice same addresses to verify will be unreserved in remove path */
 	uint64_t addresses[] = { 0x800000000000, 0x800000000000, 0x400000000000 };
 	uint64_t address, start, end;
-	bool supports_48bit;
+	bool supports_48bit, relocs;
 
 	ibb = intel_bb_create(i915, PAGE_SIZE);
 	supports_48bit = ibb->supports_48b_address;
-	if (!supports_48bit)
+	relocs = ibb->enforce_relocs;
+	if (!supports_48bit || relocs)
 		intel_bb_destroy(ibb);
 	igt_require_f(supports_48bit, "We need 48bit ppgtt for testing\n");
+	igt_require_f(!relocs, "We need to use allocator instead of relocations\n");
 
 	if (debug_bb)
 		intel_bb_set_debug(ibb, true);

@@ -1467,6 +1496,9 @@ igt_main_args("dpib", NULL, help_str, opt_handler, NULL)
 	igt_subtest("reset-bb")
 		reset_bb(bops);
 
+	igt_subtest_f("purge-bb")
+		purge_bb(bops);
+
 	igt_subtest("simple-bb")
 		simple_bb(bops, false);