claude_code/procedural-3d-engine
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Adds specialization constant to computeheadless

Browse source
This commit is contained in:
chuggafan 2017-09-21 17:30:57 -04:00
parent 685d2849da
commit 20668189f1
3 changed files with 12 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

11
computeheadless/computeheadless.cpp
View file

@ -328,6 +328,15 @@ public:
// Create pipeline
VkComputePipelineCreateInfo computePipelineCreateInfo = vks::initializers::computePipelineCreateInfo(pipelineLayout, 0);
// Create specialization constant holding struct
struct SpecializationHolding
{
uint32_t BUFFER_ELEMENT_COUNT = BUFFER_ELEMENTS;
} specialization;
// create a specialization map entry
VkSpecializationMapEntry specializationMapEntry = vks::initializers::specializationMapEntry(0, 0, sizeof(uint32_t));
VkSpecializationInfo specializationInfo = vks::initializers::specializationInfo(1, &specializationMapEntry, sizeof(SpecializationHolding), &specialization);
VkPipelineShaderStageCreateInfo shaderStage = {};
shaderStage.sType = VK_STRUCTURE_TYPE_PIPELINE_SHADER_STAGE_CREATE_INFO;
shaderStage.stage = VK_SHADER_STAGE_COMPUTE_BIT;
@ -337,6 +346,8 @@ public:
shaderStage.module = vks::tools::loadShader(ASSET_PATH "shaders/computeheadless/headless.comp.spv", device);
#endif
shaderStage.pName = "main";
shaderStage.pSpecializationInfo = &specializationInfo;
assert(shaderStage.module != VK_NULL_HANDLE);
computePipelineCreateInfo.stage = shaderStage;
VK_CHECK_RESULT(vkCreateComputePipelines(device, pipelineCache, 1, &computePipelineCreateInfo, nullptr, &pipeline));