for (d = 0; d < dword_count; d++) {
                *cs++ = cmd;
                *cs++ = i915_mmio_reg_offset(reg) + 4 * d;
-               *cs++ = intel_gt_scratch_offset(stream->engine->gt,
-                                               offset) + 4 * d;
+               *cs++ = i915_ggtt_offset(stream->noa_wait) + offset + 4 * d;
                *cs++ = 0;
        }
 
                                          MI_PREDICATE_RESULT_2_ENGINE(base) :
                                          MI_PREDICATE_RESULT_1(RENDER_RING_BASE);
 
-       bo = i915_gem_object_create_internal(i915, 4096);
+       /*
+        * gt->scratch was being used to save/restore the GPR registers, but on
+        * MTL the scratch uses stolen lmem. An MI_SRM to this memory region
+        * causes an engine hang. Instead allocate an additional page here to
+        * save/restore GPR registers
+        */
+       bo = i915_gem_object_create_internal(i915, 8192);
        if (IS_ERR(bo)) {
                drm_err(&i915->drm,
                        "Failed to allocate NOA wait batchbuffer\n");
                goto err_unpin;
        }
 
+       stream->noa_wait = vma;
+
+#define GPR_SAVE_OFFSET 4096
+#define PREDICATE_SAVE_OFFSET 4160
+
        /* Save registers. */
        for (i = 0; i < N_CS_GPR; i++)
                cs = save_restore_register(
                        stream, cs, true /* save */, CS_GPR(i),
-                       INTEL_GT_SCRATCH_FIELD_PERF_CS_GPR + 8 * i, 2);
+                       GPR_SAVE_OFFSET + 8 * i, 2);
        cs = save_restore_register(
                stream, cs, true /* save */, mi_predicate_result,
-               INTEL_GT_SCRATCH_FIELD_PERF_PREDICATE_RESULT_1, 1);
+               PREDICATE_SAVE_OFFSET, 1);
 
        /* First timestamp snapshot location. */
        ts0 = cs;
        for (i = 0; i < N_CS_GPR; i++)
                cs = save_restore_register(
                        stream, cs, false /* restore */, CS_GPR(i),
-                       INTEL_GT_SCRATCH_FIELD_PERF_CS_GPR + 8 * i, 2);
+                       GPR_SAVE_OFFSET + 8 * i, 2);
        cs = save_restore_register(
                stream, cs, false /* restore */, mi_predicate_result,
-               INTEL_GT_SCRATCH_FIELD_PERF_PREDICATE_RESULT_1, 1);
+               PREDICATE_SAVE_OFFSET, 1);
 
        /* And return to the ring. */
        *cs++ = MI_BATCH_BUFFER_END;
        i915_gem_object_flush_map(bo);
        __i915_gem_object_release_map(bo);
 
-       stream->noa_wait = vma;
        goto out_ww;
 
 err_unpin: