layers: Correctly track pSetBindPoints per stage in DrawState layer
diff --git a/layers/draw_state.c b/layers/draw_state.c
index 0fa5b7c..615eaa4 100644
--- a/layers/draw_state.c
+++ b/layers/draw_state.c
@@ -1547,6 +1547,13 @@
memcpy((void*)pNewNode->pCreateInfoList, pSetLayoutInfoList, sizeof(XGL_DESCRIPTOR_SET_LAYOUT_CREATE_INFO));
pNewNode->layout = *pSetLayout;
pNewNode->stageFlags = stageFlags;
+ uint32_t i = XGL_SHADER_STAGE_COMPUTE;
+ for (uint32_t stage = XGL_SHADER_STAGE_FLAGS_COMPUTE_BIT; stage > 0; stage >>= 1) {
+ assert(i < XGL_NUM_SHADER_STAGE);
+ if (stage & stageFlags)
+ pNewNode->shaderStageBindPoints[i] = pSetBindPoints[i];
+ i--;
+ }
pNewNode->startIndex = 0;
LAYOUT_NODE* pPriorNode = getLayoutNode(priorSetLayout);
// Point to prior node or NULL if no prior node