Method vulkan_program

compute/vulkan/vulkan_program.cpp:330–445

}

vulkan_program::vulkan_program(program_map_type&& programs_) : programs(std::move(programs_)) {
	if (programs.empty()) return;
	retrieve_unique_kernel_names(programs);
	
	// create all kernels of all device programs
	// note that this essentially reshuffles the program "device -> kernels" data to "kernels -> devices"
	kernels.reserve(kernel_names.size());
	for (const auto& func_name : kernel_names) {
		vulkan_kernel::kernel_map_type kernel_map;
		kernel_map.reserve(kernel_names.size());
		
		for (const auto& prog : programs) {
			if (!prog.second.valid) continue;
			
			for (const auto& info : prog.second.functions) {
				if (info.name == func_name) {
					vulkan_kernel::vulkan_kernel_entry entry;
					entry.info = &info;
					
					auto& prog_entry = prog.second;
					const auto& dev = prog.first.get();
					const auto& vk_dev = (const vulkan_device&)dev;
					
					const VkShaderStageFlagBits stage = (info.type == FUNCTION_TYPE::VERTEX ? VK_SHADER_STAGE_VERTEX_BIT :
														 info.type == FUNCTION_TYPE::FRAGMENT ? VK_SHADER_STAGE_FRAGMENT_BIT :
														 VK_SHADER_STAGE_COMPUTE_BIT /* should notice anything else earlier */);
					
					if (!info.has_valid_required_local_size()) {
						entry.max_local_size = dev.max_local_size;
						
						// always assume that we can execute this with the max possible work-group size,
						// i.e. use this as the initial default
						entry.max_total_local_size = dev.max_total_local_size;
					} else {
						// a required local size/dim is specified -> use it
						entry.max_local_size = info.required_local_size;
						entry.max_total_local_size = info.required_local_size.extent();
					}
					
					// TODO: make sure that _all_ of this is synchronized
					if (!create_kernel_entry_descriptor_buffer(entry, dev, stage, func_name, info)) {
						continue;
					}
					
					// find SPIR-V module index for this function
					const auto mod_iter = prog_entry.func_to_mod_map.find(func_name);
					if (mod_iter == prog_entry.func_to_mod_map.end()) {
						log_error("did not find a module mapping for function \"$\"", func_name);
						continue;
					}
					
					// stage info, can be used here or at a later point
					entry.stage_sub_group_info = VkPipelineShaderStageRequiredSubgroupSizeCreateInfo {
						.sType = VK_STRUCTURE_TYPE_PIPELINE_SHADER_STAGE_REQUIRED_SUBGROUP_SIZE_CREATE_INFO,
						.pNext = nullptr,
						// TODO: sub-group size / SIMD-width must really be stored in function info,
						// this may not work if the device supports a SIMD-width range and the program has not been compiled for the default width
						.requiredSubgroupSize = dev.simd_width,