Setup correct values of URB Info

If URB Info was not defined, set all properties based on existing instructions (SGVs and SIVs).

Author: krystian-andrzejewski

IGC/Compiler/CISACodeGen/CollectGeometryShaderProperties.cpp

@@ -147,9 +147,19 @@ void CollectGeometryShaderProperties::HandleSystemInput(llvm::GenIntrinsicInst&
     case CLIP_DISTANCE_Y:
     case CLIP_DISTANCE_Z:
     case CLIP_DISTANCE_W:
+        m_gsProps.Input().PerVertex().HasVertexHeader(true);
         m_gsProps.Input().PerVertex().HasClipCullDistances(true);
         break;
 
+    case POSITION_X:
+    case POSITION_Y:
+    case POSITION_Z:
+    case POSITION_W:
+        m_gsProps.Input().PerVertex().HasVertexHeader(true);
+
+    case POINT_WIDTH:
+        m_gsProps.Input().PerVertex().HasVertexHeader(true);
+
     case GS_INSTANCEID:
         m_gsProps.Input().HasInstanceID(true);
         break;

IGC/Compiler/CISACodeGen/DomainShaderLowering.cpp

@@ -385,9 +385,22 @@ namespace IGC
                         if (usage == SHADER_OUTPUT_TYPE_CLIPDISTANCE_LO ||
                             usage == SHADER_OUTPUT_TYPE_CLIPDISTANCE_HI)
                         {
+                            context->getModuleMetaData()->URBInfo.hasVertexHeader = true;
+                            context->getModuleMetaData()->URBInfo.has64BVertexHeaderOutput = true;
                             m_headerSize = QuadEltUnit(4);
                             m_dsPropsPass->DeclareClipDistance();
                         }
+                        else if (usage == SHADER_OUTPUT_TYPE_POSITION ||
+                            usage == SHADER_OUTPUT_TYPE_POINTWIDTH ||
+                            usage == SHADER_OUTPUT_TYPE_RENDER_TARGET_ARRAY_INDEX ||
+                            usage == SHADER_OUTPUT_TYPE_COARSE_PIXEL_SIZE)
+                        {
+                            context->getModuleMetaData()->URBInfo.hasVertexHeader = true;
+                            if (m_headerSize < QuadEltUnit(2))
+                            {
+                                m_headerSize = QuadEltUnit(2);
+                            }
+                        }
                     }
                 }
             }

IGC/Compiler/CISACodeGen/HullShaderLowering.cpp

@@ -758,8 +758,10 @@ namespace IGC
 
         auto clipCullAsInput = false;
         IGC::CodeGenContext* ctx = getAnalysis<CodeGenContextWrapper>().getCodeGenContext();
-        if (ctx->getModuleMetaData()->URBInfo.has64BVertexHeaderInput) {
+        if (ctx->getModuleMetaData()->URBInfo.has64BVertexHeaderInput)
+        {
             // In case we have no linking information we need the URB header to have a fixed size
+            IGC_ASSERT(ctx->getModuleMetaData()->URBInfo.hasVertexHeader);
             clipCullAsInput = true;
         }
 

IGC/Compiler/CISACodeGen/VertexShaderLowering.cpp

@@ -556,9 +556,22 @@ namespace IGC
                         if (usage == SHADER_OUTPUT_TYPE_CLIPDISTANCE_LO ||
                             usage == SHADER_OUTPUT_TYPE_CLIPDISTANCE_HI)
                         {
+                            m_context->getModuleMetaData()->URBInfo.hasVertexHeader = true;
+                            m_context->getModuleMetaData()->URBInfo.has64BVertexHeaderOutput = true;
                             m_headerSize = QuadEltUnit(4);
                             m_vsPropsPass->DeclareClipDistance();
                         }
+                        else if (usage == SHADER_OUTPUT_TYPE_POSITION ||
+                            usage == SHADER_OUTPUT_TYPE_POINTWIDTH ||
+                            usage == SHADER_OUTPUT_TYPE_RENDER_TARGET_ARRAY_INDEX ||
+                            usage == SHADER_OUTPUT_TYPE_COARSE_PIXEL_SIZE)
+                        {
+                            m_context->getModuleMetaData()->URBInfo.hasVertexHeader = true;
+                            if (m_headerSize < QuadEltUnit(2))
+                            {
+                                m_headerSize = QuadEltUnit(2);
+                            }
+                        }
                     }
                 }
             }