return desc;
 }
 
-static u32 *set_offsets(u32 *regs,
+static inline unsigned int dword_in_page(void *addr)
+{
+       return offset_in_page(addr) / sizeof(u32);
+}
+
+static void set_offsets(u32 *regs,
                        const u8 *data,
-                       const struct intel_engine_cs *engine)
+                       const struct intel_engine_cs *engine,
+                       bool clear)
 #define NOP(x) (BIT(7) | (x))
 #define LRI(count, flags) ((flags) << 6 | (count) | BUILD_BUG_ON_ZERO(count >= BIT(6)))
 #define POSTED BIT(0)
 #define REG16(x) \
        (((x) >> 9) | BIT(7) | BUILD_BUG_ON_ZERO(x >= 0x10000)), \
        (((x) >> 2) & 0x7f)
-#define END() 0
+#define END(x) 0, (x)
 {
        const u32 base = engine->mmio_base;
 
                u8 count, flags;
 
                if (*data & BIT(7)) { /* skip */
-                       regs += *data++ & ~BIT(7);
+                       count = *data++ & ~BIT(7);
+                       if (clear)
+                               memset32(regs, MI_NOOP, count);
+                       regs += count;
                        continue;
                }
 
                                offset |= v & ~BIT(7);
                        } while (v & BIT(7));
 
-                       *regs = base + (offset << 2);
+                       regs[0] = base + (offset << 2);
+                       if (clear)
+                               regs[1] = 0;
                        regs += 2;
                } while (--count);
        }
 
-       return regs;
+       if (clear) {
+               u8 count = *++data;
+
+               /* Clear past the tail for HW access */
+               GEM_BUG_ON(dword_in_page(regs) > count);
+               memset32(regs, MI_NOOP, count - dword_in_page(regs));
+
+               /* Close the batch; used mainly by live_lrc_layout() */
+               *regs = MI_BATCH_BUFFER_END;
+               if (INTEL_GEN(engine->i915) >= 10)
+                       *regs |= BIT(0);
+       }
 }
 
 static const u8 gen8_xcs_offsets[] = {
        REG16(0x200),
        REG(0x028),
 
-       END(),
+       END(80)
 };
 
 static const u8 gen9_xcs_offsets[] = {
        REG16(0x67c),
        REG(0x068),
 
-       END(),
+       END(176)
 };
 
 static const u8 gen12_xcs_offsets[] = {
        REG16(0x274),
        REG16(0x270),
 
-       END(),
+       END(80)
 };
 
 static const u8 gen8_rcs_offsets[] = {
        LRI(1, 0),
        REG(0x0c8),
 
-       END(),
+       END(80)
 };
 
 static const u8 gen9_rcs_offsets[] = {
        REG16(0x67c),
        REG(0x68),
 
-       END()
+       END(176)
 };
 
 static const u8 gen11_rcs_offsets[] = {
        LRI(1, 0),
        REG(0x0c8),
 
-       END(),
+       END(80)
 };
 
 static const u8 gen12_rcs_offsets[] = {
        LRI(1, 0),
        REG(0x0c8),
 
-       END(),
+       END(80)
 };
 
 #undef END
 static void virtual_update_register_offsets(u32 *regs,
                                            struct intel_engine_cs *engine)
 {
-       set_offsets(regs, reg_offsets(engine), engine);
+       set_offsets(regs, reg_offsets(engine), engine, false);
 }
 
 static bool virtual_matches(const struct virtual_engine *ve,
 
 static void init_common_reg_state(u32 * const regs,
                                  const struct intel_engine_cs *engine,
-                                 const struct intel_ring *ring)
+                                 const struct intel_ring *ring,
+                                 bool inhibit)
 {
-       regs[CTX_CONTEXT_CONTROL] =
-               _MASKED_BIT_DISABLE(CTX_CTRL_ENGINE_CTX_RESTORE_INHIBIT) |
-               _MASKED_BIT_ENABLE(CTX_CTRL_INHIBIT_SYN_CTX_SWITCH);
+       u32 ctl;
+
+       ctl = _MASKED_BIT_ENABLE(CTX_CTRL_INHIBIT_SYN_CTX_SWITCH);
+       ctl |= _MASKED_BIT_DISABLE(CTX_CTRL_ENGINE_CTX_RESTORE_INHIBIT);
+       if (inhibit)
+               ctl |= CTX_CTRL_ENGINE_CTX_RESTORE_INHIBIT;
        if (INTEL_GEN(engine->i915) < 11)
-               regs[CTX_CONTEXT_CONTROL] |=
-                       _MASKED_BIT_DISABLE(CTX_CTRL_ENGINE_CTX_SAVE_INHIBIT |
-                                           CTX_CTRL_RS_CTX_ENABLE);
+               ctl |= _MASKED_BIT_DISABLE(CTX_CTRL_ENGINE_CTX_SAVE_INHIBIT |
+                                          CTX_CTRL_RS_CTX_ENABLE);
+       regs[CTX_CONTEXT_CONTROL] = ctl;
 
        regs[CTX_RING_CTL] = RING_CTL_SIZE(ring->size) | RING_VALID;
 }
                                     const struct intel_context *ce,
                                     const struct intel_engine_cs *engine,
                                     const struct intel_ring *ring,
-                                    bool close)
+                                    bool inhibit)
 {
        /*
         * A context is actually a big batch buffer with several
         *
         * Must keep consistent with virtual_update_register_offsets().
         */
-       u32 *bbe = set_offsets(regs, reg_offsets(engine), engine);
+       set_offsets(regs, reg_offsets(engine), engine, inhibit);
 
-       if (close) { /* Close the batch; used mainly by live_lrc_layout() */
-               *bbe = MI_BATCH_BUFFER_END;
-               if (INTEL_GEN(engine->i915) >= 10)
-                       *bbe |= BIT(0);
-       }
-
-       init_common_reg_state(regs, engine, ring);
+       init_common_reg_state(regs, engine, ring, inhibit);
        init_ppgtt_reg_state(regs, vm_alias(ce->vm));
 
        init_wa_bb_reg_state(regs, engine,
 {
        bool inhibit = true;
        void *vaddr;
-       u32 *regs;
        int ret;
 
        vaddr = i915_gem_object_pin_map(ctx_obj, I915_MAP_WB);
 
        /* The second page of the context object contains some fields which must
         * be set up prior to the first execution. */
-       regs = vaddr + LRC_STATE_PN * PAGE_SIZE;
-       execlists_init_reg_state(regs, ce, engine, ring, inhibit);
-       if (inhibit)
-               regs[CTX_CONTEXT_CONTROL] |=
-                       _MASKED_BIT_ENABLE(CTX_CTRL_ENGINE_CTX_RESTORE_INHIBIT);
+       execlists_init_reg_state(vaddr + LRC_STATE_PN * PAGE_SIZE,
+                                ce, engine, ring, inhibit);
 
        ret = 0;
 err_unpin_ctx: