From 7899d449e6526ca0947bece677894f0617eb82c3 Mon Sep 17 00:00:00 2001
From: Hans-Kristian Arntzen <maister@archlinux.us>
Date: Mon, 21 Mar 2016 11:49:27 +0100
Subject: [PATCH] Vulkan: Also invalidate fragment uniform buffer

---
 gfx/drivers/vulkan.c | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/gfx/drivers/vulkan.c b/gfx/drivers/vulkan.c
index 4e90400ff7..d5e4e786a6 100644
--- a/gfx/drivers/vulkan.c
+++ b/gfx/drivers/vulkan.c
@@ -1351,7 +1351,7 @@ static void vulkan_readback(vk_t *vk)
          VK_PIPELINE_STAGE_HOST_BIT);
 }
 
-static void vulkan_flush_vertex_caches(vk_t *vk)
+static void vulkan_flush_caches(vk_t *vk)
 {
    VkMemoryBarrier barrier = { VK_STRUCTURE_TYPE_MEMORY_BARRIER };
    barrier.srcAccessMask = VK_ACCESS_HOST_WRITE_BIT;
@@ -1359,7 +1359,9 @@ static void vulkan_flush_vertex_caches(vk_t *vk)
 
    VKFUNC(vkCmdPipelineBarrier)(vk->cmd,
          VK_PIPELINE_STAGE_HOST_BIT,
-         VK_PIPELINE_STAGE_VERTEX_INPUT_BIT | VK_PIPELINE_STAGE_VERTEX_SHADER_BIT,
+         VK_PIPELINE_STAGE_VERTEX_INPUT_BIT |
+         VK_PIPELINE_STAGE_VERTEX_SHADER_BIT |
+         VK_PIPELINE_STAGE_FRAGMENT_SHADER_BIT,
          false,
          1, &barrier,
          0, NULL, 0, NULL);
@@ -1421,7 +1423,7 @@ static bool vulkan_frame(void *data, const void *frame,
 
    memset(&vk->tracker, 0, sizeof(vk->tracker));
 
-   vulkan_flush_vertex_caches(vk);
+   vulkan_flush_caches(vk);
 
    /* Upload texture */
    retro_perf_start(&copy_frame);