From 77ef24157744dc52ff5e41fe1163c43ff5e8e5c5 Mon Sep 17 00:00:00 2001
From: Ian Romanick <ian.d.romanick@intel.com>
Date: Thu, 4 Apr 2024 21:48:53 -0700
Subject: [PATCH] intel/brw/xe2+: Scale size_written by reg_unit for DPAS

Reviewed-by: Caio Oliveira <caio.oliveira@intel.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/28834>
---
 src/intel/compiler/brw_fs_builder.h | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/intel/compiler/brw_fs_builder.h b/src/intel/compiler/brw_fs_builder.h
index da4c3c9cdc7..8c95afaeb7c 100644
--- a/src/intel/compiler/brw_fs_builder.h
+++ b/src/intel/compiler/brw_fs_builder.h
@@ -852,9 +852,9 @@ namespace brw {
          inst->rcount = rcount;
 
          if (dst.type == BRW_TYPE_HF) {
-            inst->size_written = rcount * REG_SIZE / 2;
+            inst->size_written = reg_unit(shader->devinfo) * rcount * REG_SIZE / 2;
          } else {
-            inst->size_written = rcount * REG_SIZE;
+            inst->size_written = reg_unit(shader->devinfo) * rcount * REG_SIZE;
          }
 
          return inst;