struct insert_pte_data {
        u64 offset;
-       bool is_lmem;
 };
 
 #define CHUNK_SZ SZ_8M /* ~1ms at 8GiB/s preemption delay */
        struct insert_pte_data *d = data;
 
        vm->insert_page(vm, px_dma(pt), d->offset, I915_CACHE_NONE,
-                       d->is_lmem ? PTE_LM : 0);
+                       i915_gem_object_is_lmem(pt->base) ? PTE_LM : 0);
        d->offset += PAGE_SIZE;
 }
 
                        goto err_vm;
 
                /* Now allow the GPU to rewrite the PTE via its own ppGTT */
-               d.is_lmem = i915_gem_object_is_lmem(vm->vm.scratch[0]);
                vm->vm.foreach(&vm->vm, base, base + sz, insert_pte, &d);
        }