#include "xe_sriov.h"
 #include "xe_vm.h"
 
-#define LRC_VALID                              (1 << 0)
-#define LRC_PRIVILEGE                          (1 << 8)
-#define LRC_ADDRESSING_MODE_SHIFT              3
+#define LRC_VALID                              BIT_ULL(0)
+#define LRC_PRIVILEGE                          BIT_ULL(8)
+#define LRC_ADDRESSING_MODE                    GENMASK_ULL(4, 3)
 #define LRC_LEGACY_64B_CONTEXT                 3
 
-#define ENGINE_CLASS_SHIFT                     61
-#define ENGINE_INSTANCE_SHIFT                  48
+#define LRC_ENGINE_CLASS                       GENMASK_ULL(63, 61)
+#define LRC_ENGINE_INSTANCE                    GENMASK_ULL(53, 48)
 
 struct xe_lrc_snapshot {
        struct xe_bo *lrc_bo;
                xe_lrc_write_ctx_reg(lrc, PVC_CTX_ASID, vm->usm.asid);
 
        lrc->desc = LRC_VALID;
-       lrc->desc |= LRC_LEGACY_64B_CONTEXT << LRC_ADDRESSING_MODE_SHIFT;
+       lrc->desc |= FIELD_PREP(LRC_ADDRESSING_MODE, LRC_LEGACY_64B_CONTEXT);
        /* TODO: Priority */
 
        /* While this appears to have something about privileged batches or
                lrc->desc |= LRC_PRIVILEGE;
 
        if (GRAPHICS_VERx100(xe) < 1250) {
-               lrc->desc |= (u64)hwe->instance << ENGINE_INSTANCE_SHIFT;
-               lrc->desc |= (u64)hwe->class << ENGINE_CLASS_SHIFT;
+               lrc->desc |= FIELD_PREP(LRC_ENGINE_INSTANCE, hwe->instance);
+               lrc->desc |= FIELD_PREP(LRC_ENGINE_CLASS, hwe->class);
        }
 
        arb_enable = MI_ARB_ON_OFF | MI_ARB_ENABLE;