rusticl/mem: only write pitch when required

Signed-off-by: Karol Herbst <kherbst@redhat.com>
Acked-by: Alyssa Rosenzweig <alyssa.rosenzweig@collabora.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/15439>

src/gallium/frontends/rusticl/core/memory.rs

@@ -880,8 +880,12 @@ impl Mem {
         } else {
             let tx = self.map(q, ctx, &mut lock)?;
 
-            *row_pitch = tx.row_pitch() as usize;
+            if self.image_desc.dims() > 1 {
-            *slice_pitch = tx.slice_pitch() as usize;
+                *row_pitch = tx.row_pitch() as usize;
+            }
+            if self.image_desc.dims() > 2 || self.image_desc.is_array() {
+                *slice_pitch = tx.slice_pitch() as usize;
+            }
 
             tx.ptr()
         };