Fix the implicit argument handling

bcheng0127 · igcbot · commit 374ec0c35ae8 · 2022-04-15T18:33:40.000+02:00
diff --git a/IGC/Compiler/CISACodeGen/EmitVISAPass.cpp b/IGC/Compiler/CISACodeGen/EmitVISAPass.cpp
@@ -20844,8 +20844,16 @@ void EmitPass::emitImplicitArgIntrinsic(llvm::GenIntrinsicInst* I)
         IGC_ASSERT_MESSAGE(arg, "Implicit argument not found!");
         if (arg)
         {
-            m_encoder->SetNoMask();
-            m_currShader->CopyVariable(GetSymbol(I), m_currShader->getOrCreateArgumentSymbol(arg, false));
+            if (I->getType()->isVectorTy())
+            {
+                emitVectorCopy(GetSymbol(I), m_currShader->getOrCreateArgumentSymbol(arg, false),
+                    int_cast<unsigned>(dyn_cast<IGCLLVM::FixedVectorType>(I->getType())->getNumElements()));
+            }
+            else
+            {
+                m_encoder->SetNoMask();
+                m_currShader->CopyVariable(GetSymbol(I), m_currShader->getOrCreateArgumentSymbol(arg, false));
+            }
         }
     }
     else

Original file line number	Diff line number	Diff line change
`@@ -20844,8 +20844,16 @@ void EmitPass::emitImplicitArgIntrinsic(llvm::GenIntrinsicInst* I)`
`20844`	`20844`	`IGC_ASSERT_MESSAGE(arg, "Implicit argument not found!");`
`20845`	`20845`	`if (arg)`
`20846`	`20846`	`{`
`20847`		`- m_encoder->SetNoMask();`
`20848`		`- m_currShader->CopyVariable(GetSymbol(I), m_currShader->getOrCreateArgumentSymbol(arg, false));`
	`20847`	`+ if (I->getType()->isVectorTy())`
	`20848`	`+ {`
	`20849`	`+ emitVectorCopy(GetSymbol(I), m_currShader->getOrCreateArgumentSymbol(arg, false),`
	`20850`	`+ int_cast<unsigned>(dyn_cast<IGCLLVM::FixedVectorType>(I->getType())->getNumElements()));`
	`20851`	`+ }`
	`20852`	`+ else`
	`20853`	`+ {`
	`20854`	`+ m_encoder->SetNoMask();`
	`20855`	`+ m_currShader->CopyVariable(GetSymbol(I), m_currShader->getOrCreateArgumentSymbol(arg, false));`
	`20856`	`+ }`
`20849`	`20857`	`}`
`20850`	`20858`	`}`
`20851`	`20859`	`else`