gen8/cmd_buffer: Properly return flushed push constant stages

This is required on SKL so that we can properly re-emit binding table
pointers commands.

1 files changed, 3 insertions, 1 deletions

diff --git a/src/intel/vulkan/gen8_cmd_buffer.c b/src/intel/vulkan/gen8_cmd_buffer.c
index 884152da207..9d4926f86cd 100644
--- a/src/intel/vulkan/gen8_cmd_buffer.c
+++ b/src/intel/vulkan/gen8_cmd_buffer.c
@@ -63,9 +63,11 @@ cmd_buffer_flush_push_constants(struct anv_cmd_buffer *cmd_buffer)
                            .ConstantBuffer2ReadLength = DIV_ROUND_UP(state.alloc_size, 32),
                         });
       }
+
+      flushed |= mesa_to_vk_shader_stage(stage);
    }
 
-   cmd_buffer->state.push_constants_dirty &= ~VK_SHADER_STAGE_ALL_GRAPHICS;
+   cmd_buffer->state.push_constants_dirty &= ~flushed;
 
    return flushed;
 }