if (ret)
                return ret;
 
-       fw->mem = ivpu_bo_alloc_internal(vdev, fw->runtime_addr, fw->runtime_size, DRM_IVPU_BO_WC);
+       fw->mem = ivpu_bo_alloc_internal(vdev, fw->runtime_addr, fw->runtime_size,
+                                        DRM_IVPU_BO_CACHED | DRM_IVPU_BO_NOSNOOP);
        if (!fw->mem) {
                ivpu_err(vdev, "Failed to allocate firmware runtime memory\n");
                return -ENOMEM;
                memset(start, 0, size);
        }
 
-       wmb(); /* Flush WC buffers after writing fw->mem */
+       clflush_cache_range(fw->mem->kvaddr, fw->mem->base.size);
 
        return 0;
 }
        if (!ivpu_fw_is_cold_boot(vdev)) {
                boot_params->save_restore_ret_address = 0;
                vdev->pm->is_warmboot = true;
+               clflush_cache_range(vdev->fw->mem->kvaddr, SZ_4K);
                return;
        }
 
        boot_params->punit_telemetry_sram_size = ivpu_hw_reg_telemetry_size_get(vdev);
        boot_params->vpu_telemetry_enable = ivpu_hw_reg_telemetry_enable_get(vdev);
 
-       wmb(); /* Flush WC buffers after writing bootparams */
+       clflush_cache_range(vdev->fw->mem->kvaddr, SZ_4K);
 
        ivpu_fw_boot_params_print(vdev, boot_params);
 }
 
 #include <drm/drm_gem.h>
 #include <drm/drm_mm.h>
 
+#define DRM_IVPU_BO_NOSNOOP       0x10000000
+
 struct dma_buf;
 struct ivpu_bo_ops;
 struct ivpu_file_priv;
 
 static inline bool ivpu_bo_is_snooped(struct ivpu_bo *bo)
 {
+       if (bo->flags & DRM_IVPU_BO_NOSNOOP)
+               return false;
+
        return ivpu_bo_cache_mode(bo) == DRM_IVPU_BO_CACHED;
 }