From 27ef460546d013903b3d489e791cf6a4c05563fd Mon Sep 17 00:00:00 2001
From: Alex Smith <asmith@feralinteractive.com>
Date: Wed, 25 Oct 2017 15:33:46 +0100
Subject: [PATCH] Set shader stage in reflection for compute shaders

Missing this meant an incorrect stage flag would be passed to
vkGetShaderInfoAMD for compute shaders when getting a disassembly.
---
 renderdoc/driver/vulkan/vk_info.cpp | 1 +
 1 file changed, 1 insertion(+)

diff --git a/renderdoc/driver/vulkan/vk_info.cpp b/renderdoc/driver/vulkan/vk_info.cpp
index 1323209f9..5802cd71d 100644
--- a/renderdoc/driver/vulkan/vk_info.cpp
+++ b/renderdoc/driver/vulkan/vk_info.cpp
@@ -376,6 +376,7 @@ void VulkanCreationInfo::Pipeline::Init(VulkanResourceManager *resourceMan, Vulk
     if(reflData.entryPoint.empty())
     {
       reflData.entryPoint = shad.entryPoint;
+      reflData.stage = StageIndex(pCreateInfo->stage.stage);
       SPVModule &spv = info.m_ShaderModule[id].spirv;
       spv.MakeReflection(ShaderStage::Compute, reflData.entryPoint, reflData.refl, reflData.mapping,
                          reflData.patchData);