#include "gen8_engine_cs.h"
 #include "i915_drv.h"
 #include "intel_gpu_commands.h"
-#include "intel_gt_regs.h"
 #include "intel_lrc.h"
 #include "intel_ring.h"
 
        return MI_ARB_CHECK | 1 << 8 | state;
 }
 
-static i915_reg_t aux_inv_reg(const struct intel_engine_cs *engine)
+u32 *gen12_emit_aux_table_inv(u32 *cs, const i915_reg_t inv_reg)
 {
-       static const i915_reg_t vd[] = {
-               GEN12_VD0_AUX_NV,
-               GEN12_VD1_AUX_NV,
-               GEN12_VD2_AUX_NV,
-               GEN12_VD3_AUX_NV,
-       };
-
-       static const i915_reg_t ve[] = {
-               GEN12_VE0_AUX_NV,
-               GEN12_VE1_AUX_NV,
-       };
-
-       if (engine->class == VIDEO_DECODE_CLASS)
-               return vd[engine->instance];
-
-       if (engine->class == VIDEO_ENHANCEMENT_CLASS)
-               return ve[engine->instance];
-
-       GEM_BUG_ON("unknown aux_inv reg\n");
-       return INVALID_MMIO_REG;
-}
-
-static u32 *gen12_emit_aux_table_inv(const i915_reg_t inv_reg, u32 *cs)
-{
-       *cs++ = MI_LOAD_REGISTER_IMM(1);
+       *cs++ = MI_LOAD_REGISTER_IMM(1) | MI_LRI_MMIO_REMAP_EN;
        *cs++ = i915_mmio_reg_offset(inv_reg);
        *cs++ = AUX_INV;
        *cs++ = MI_NOOP;
 
                if (!HAS_FLAT_CCS(rq->engine->i915)) {
                        /* hsdes: 1809175790 */
-                       cs = gen12_emit_aux_table_inv(GEN12_GFX_CCS_AUX_NV, cs);
+                       cs = gen12_emit_aux_table_inv(cs, GEN12_GFX_CCS_AUX_NV);
                }
 
                *cs++ = preparser_disable(false);
        if (mode & EMIT_INVALIDATE) {
                cmd += 2;
 
-               if (!HAS_FLAT_CCS(rq->engine->i915)) {
+               if (!HAS_FLAT_CCS(rq->engine->i915) &&
+                   (rq->engine->class == VIDEO_DECODE_CLASS ||
+                    rq->engine->class == VIDEO_ENHANCEMENT_CLASS)) {
                        aux_inv = rq->engine->mask & ~BIT(BCS0);
                        if (aux_inv)
-                               cmd += 2 * hweight32(aux_inv) + 2;
+                               cmd += 4;
                }
        }
 
        *cs++ = 0; /* value */
 
        if (aux_inv) { /* hsdes: 1809175790 */
-               struct intel_engine_cs *engine;
-               unsigned int tmp;
-
-               *cs++ = MI_LOAD_REGISTER_IMM(hweight32(aux_inv));
-               for_each_engine_masked(engine, rq->engine->gt, aux_inv, tmp) {
-                       *cs++ = i915_mmio_reg_offset(aux_inv_reg(engine));
-                       *cs++ = AUX_INV;
-               }
-               *cs++ = MI_NOOP;
+               if (rq->engine->class == VIDEO_DECODE_CLASS)
+                       cs = gen12_emit_aux_table_inv(cs, GEN12_VD0_AUX_NV);
+               else
+                       cs = gen12_emit_aux_table_inv(cs, GEN12_VE0_AUX_NV);
        }
 
        if (mode & EMIT_INVALIDATE)
 
 #include <linux/types.h>
 
 #include "i915_gem.h" /* GEM_BUG_ON */
-
+#include "intel_gt_regs.h"
 #include "intel_gpu_commands.h"
 
 struct i915_request;
 u32 *gen11_emit_fini_breadcrumb_rcs(struct i915_request *rq, u32 *cs);
 u32 *gen12_emit_fini_breadcrumb_rcs(struct i915_request *rq, u32 *cs);
 
+u32 *gen12_emit_aux_table_inv(u32 *cs, const i915_reg_t inv_reg);
+
 static inline u32 *
 __gen8_emit_pipe_control(u32 *batch, u32 flags0, u32 flags1, u32 offset)
 {
 
            IS_DG2_G11(ce->engine->i915))
                cs = gen8_emit_pipe_control(cs, PIPE_CONTROL_INSTRUCTION_CACHE_INVALIDATE, 0);
 
+       /* hsdes: 1809175790 */
+       if (!HAS_FLAT_CCS(ce->engine->i915))
+               cs = gen12_emit_aux_table_inv(cs, GEN12_GFX_CCS_AUX_NV);
+
        return cs;
 }
 
                                                    PIPE_CONTROL_INSTRUCTION_CACHE_INVALIDATE,
                                                    0);
 
+       /* hsdes: 1809175790 */
+       if (!HAS_FLAT_CCS(ce->engine->i915)) {
+               if (ce->engine->class == VIDEO_DECODE_CLASS)
+                       cs = gen12_emit_aux_table_inv(cs, GEN12_VD0_AUX_NV);
+               else if (ce->engine->class == VIDEO_ENHANCEMENT_CLASS)
+                       cs = gen12_emit_aux_table_inv(cs, GEN12_VE0_AUX_NV);
+       }
+
        return cs;
 }