Apply SPIRV-Cross workarounds for Metal bugs

MoltenVK/MoltenVK/GPUObjects/MVKPipeline.h

@@ -456,6 +456,7 @@ class MVKGraphicsPipeline : public MVKPipeline {
 	bool _isRasterizingColor = false;
 	bool _sampleLocationsEnable = false;
 	bool _isTessellationPipeline = false;
+	bool _inputAttachmentIsDSAttachment = false;
 };
 
 

MoltenVK/MoltenVK/GPUObjects/MVKPipeline.mm

@@ -507,6 +507,16 @@
 	_outputControlPointCount = reflectData.numControlPoints;
 	mvkSetOrClear(&_tessInfo, _isTessellationPipeline ? pCreateInfo->pTessellationState : nullptr);
 
+	// Handles depth attachment being used as input attachment. However, it does not solve the issue when
+	// the pipeline is created without render pass (dynamic rendering) since we won't be able to know
+	// which resources will be used when rendering. Needs to be done before we do shaders
+	// Potential solution would be to generate 2 pipelines, one with the workaround for the Metal issue
+	// and one without it, and decide at bind time once we know the resources which one to use.
+	if (pCreateInfo->renderPass) {
+		MVKRenderSubpass* subpass = ((MVKRenderPass*)pCreateInfo->renderPass)->getSubpass(pCreateInfo->subpass);
+		_inputAttachmentIsDSAttachment = subpass->isInputAttachmentDepthStencilAttachment();
+	}
+
 	// Render pipeline state. Do this as early as possible, to fail fast if pipeline requires a fail on cache-miss.
 	initMTLRenderPipelineState(pCreateInfo, reflectData, pPipelineFB, pVertexSS, pVertexFB, pTessCtlSS, pTessCtlFB, pTessEvalSS, pTessEvalFB, pFragmentSS, pFragmentFB);
 	if ( !_hasValidMTLPipelineStates ) { return; }
@@ -1332,6 +1342,8 @@ static MTLVertexFormat mvkAdjustFormatVectorToSize(MTLVertexFormat format, uint3
 		shaderConfig.options.mslOptions.capture_output_to_buffer = false;
 		shaderConfig.options.mslOptions.fixed_subgroup_size = mvkIsAnyFlagEnabled(pFragmentSS->flags, VK_PIPELINE_SHADER_STAGE_CREATE_ALLOW_VARYING_SUBGROUP_SIZE_BIT_EXT) ? 0 : mtlFeats.maxSubgroupSize;
 		shaderConfig.options.mslOptions.check_discarded_frag_stores = true;
+		shaderConfig.options.mslOptions.force_fragment_with_side_effects_execution = true;
+		shaderConfig.options.mslOptions.input_attachment_is_ds_attachment = _inputAttachmentIsDSAttachment;
 		if (mtlFeats.needsSampleDrefLodArrayWorkaround) {
 			shaderConfig.options.mslOptions.sample_dref_lod_array_as_grad = true;
 		}
@@ -2580,6 +2592,8 @@ void serialize(Archive & archive, CompilerMSL::Options& opt) {
 				opt.force_sample_rate_shading,
 				opt.manual_helper_invocation_updates,
 				opt.check_discarded_frag_stores,
+				opt.force_fragment_with_side_effects_execution,
+				opt.input_attachment_is_ds_attachment,
 				opt.sample_dref_lod_array_as_grad,
 				opt.replace_recursive_inputs,
 				opt.agx_manual_cube_grad_fixup);

MoltenVK/MoltenVK/GPUObjects/MVKRenderPass.h

@@ -67,10 +67,10 @@ class MVKRenderSubpass : public MVKBaseObject {
 	bool isColorAttachmentAlsoInputAttachment(uint32_t colorAttIdx);
 
 	/** Returns whether or not the depth attachment is being used. */
-	bool isDepthAttachmentUsed() { return _depthAttachment.attachment != VK_ATTACHMENT_UNUSED; }
+	bool isDepthAttachmentUsed() const { return _depthAttachment.attachment != VK_ATTACHMENT_UNUSED; }
 
 	/** Returns whether or not the stencil attachment is being used. */
-	bool isStencilAttachmentUsed() { return _stencilAttachment.attachment != VK_ATTACHMENT_UNUSED; }
+	bool isStencilAttachmentUsed() const { return _stencilAttachment.attachment != VK_ATTACHMENT_UNUSED; }
 
 	/** Return the depth attachment format. */
 	VkFormat getDepthFormat();
@@ -90,6 +90,9 @@ class MVKRenderSubpass : public MVKBaseObject {
 	/** Returns whether or not this is a multiview subpass. */
 	bool isMultiview() const { return _pipelineRenderingCreateInfo.viewMask != 0; }
 
+	/** Returns whether or not the depth stencil is being used as input attachment */
+	bool isInputAttachmentDepthStencilAttachment() const { return _isInputAttachmentDepthStencilAttachment; }
+
 	/** Returns the multiview view mask. */
 	uint32_t getViewMask() const { return _pipelineRenderingCreateInfo.viewMask; }
 
@@ -177,6 +180,7 @@ class MVKRenderSubpass : public MVKBaseObject {
 	VkResolveModeFlagBits _stencilResolveMode = VK_RESOLVE_MODE_NONE;
 	VkSampleCountFlagBits _defaultSampleCount = VK_SAMPLE_COUNT_1_BIT;
 	uint32_t _subpassIndex;
+	bool _isInputAttachmentDepthStencilAttachment = false;
 };
 
 

MoltenVK/MoltenVK/GPUObjects/MVKRenderPass.mm

@@ -438,6 +438,21 @@
 	_pipelineRenderingCreateInfo.pColorAttachmentFormats = _colorAttachmentFormats.data();
 	_pipelineRenderingCreateInfo.depthAttachmentFormat = getDepthFormat();
 	_pipelineRenderingCreateInfo.stencilAttachmentFormat = getStencilFormat();
+
+	// Needed to understand if we need to force the depth/stencil write to post fragment execution
+	// since Metal may try to do the write pre fragment exeuction which is against Vulkan
+	bool depthAttachmentUsed = isDepthAttachmentUsed();
+	bool stencilAttachmentUsed = isStencilAttachmentUsed();
+	for (uint32_t i = 0u; i < _inputAttachments.size(); ++i) {
+		bool isDepthInput = depthAttachmentUsed && (_inputAttachments[i].attachment == _depthAttachment.attachment) &&
+							  (_inputAttachments[i].aspectMask & _depthAttachment.aspectMask);
+		bool isStencilInput = stencilAttachmentUsed && (_inputAttachments[i].attachment == _stencilAttachment.attachment) &&
+							  (_inputAttachments[i].aspectMask & _stencilAttachment.aspectMask);
+		if (isDepthInput || isStencilInput) {
+			_isInputAttachmentDepthStencilAttachment = true;
+			break;
+		}
+	}
 }
 
 static const VkAttachmentReference2 _unusedAttachment = {VK_STRUCTURE_TYPE_ATTACHMENT_REFERENCE_2, nullptr, VK_ATTACHMENT_UNUSED, VK_IMAGE_LAYOUT_UNDEFINED, 0};