diff --git a/src/gallium/frontends/rusticl/core/memory.rs b/src/gallium/frontends/rusticl/core/memory.rs
index fe7d59bd33c..c6754f3e7f6 100644
--- a/src/gallium/frontends/rusticl/core/memory.rs
+++ b/src/gallium/frontends/rusticl/core/memory.rs
@@ -527,8 +527,8 @@ impl Mem {
         let r = self.get_res()?.get(&q.device).unwrap();
         let ctx = q.device.helper_ctx();
 
-        // don't bother mapping directly if it's not UMA
-        let tx = if q.device.unified_memory() || bit_check(self.flags, CL_MEM_ALLOC_HOST_PTR) {
+        // don't bother mapping directly if it's not staging
+        let tx = if bit_check(self.flags, CL_MEM_ALLOC_HOST_PTR) {
             ctx.texture_map_directly(r, bx, rw)
         } else {
             None