[llvm] [SPIR-V] Fix OpenCL integer dot product lowering (PR #177258)

[Juan Manuel Martinez Caamaño via llvm-commits]

================
@@ -1198,4 +1198,58 @@ getSpirvLinkageTypeFor(const SPIRVSubtarget &ST, const GlobalValue &GV) {
   return SPIRV::LinkageType::Export;
 }
 
+bool generateIntegerDotExpansion(MachineIRBuilder &MIRBuilder, Register ResVReg,
+                                 Register Vec0, Register Vec1,
+                                 SPIRVGlobalRegistry *GR) {
+  // Expand integer dot product to element-wise multiply and sum:
+  //   dot(a, b) = a[0]*b[0] + a[1]*b[1] + ... + a[n-1]*b[n-1]
+  MachineRegisterInfo *MRI = MIRBuilder.getMRI();
+  SPIRVType *VecType = GR->getSPIRVTypeForVReg(Vec0);
+  SPIRVType *ResType = GR->getSPIRVTypeForVReg(ResVReg);
+
+  assert(VecType && VecType->getOpcode() == SPIRV::OpTypeVector &&
+         "Expected vector type for integer dot product");
+  unsigned NumComponents = GR->getScalarOrVectorComponentCount(VecType);
+  assert(NumComponents > 1 && "dot product requires vector of at least 2");
+
+  // Multiply the vectors element-wise.
+  Register TmpVec = MRI->createVirtualRegister(GR->getRegClass(VecType));
+  MIRBuilder.buildInstr(SPIRV::OpIMulV)
----------------
jmmartinez wrote:

> unsigned arithmetic instructions for unsigned dot product

Wait. I'm confused. `unsigned` does wrap (at least in C), so I'm not sure adding `nuw` applies.

https://github.com/llvm/llvm-project/pull/177258