Added FMul

KhronosGroup · Nov 10, 2023 · 4e030ae · 4e030ae
1 parent aec34ac
commit 4e030ae
Show file tree

Hide file tree

Showing 2 changed files with 451 additions and 20 deletions.
diff --git a/lib/SPIRV/SPIRVWriter.cpp b/lib/SPIRV/SPIRVWriter.cpp
@@ -91,7 +91,6 @@
 #include "llvm/Transforms/Utils/LoopSimplify.h"
 #include "llvm/Transforms/Utils/Mem2Reg.h"
 
-#include <cstdio>
 #include <cstdlib>
 #include <functional>
 #include <iostream>
@@ -4066,6 +4065,26 @@ SPIRVValue *LLVMToSPIRVBase::transIntrinsicInst(IntrinsicInst *II,
     }
     return Extracts[0];
   }
+  case Intrinsic::vector_reduce_fadd:
+  case Intrinsic::vector_reduce_fmul: {
+    Op Op = IID == Intrinsic::vector_reduce_fadd ? OpFAdd : OpFMul;
+    SPIRVValue *VectorVal = transValue(II->getArgOperand(1), BB);
+    VectorType *VecType = cast<VectorType>(II->getArgOperand(1)->getType());
+    SPIRVValue *StartingVal = transValue(II->getArgOperand(0), BB);
+    SPIRVTypeInt *I32 = BM->addIntegerType(32);
+    unsigned ArrSize = VecType->getElementCount().getFixedValue();
+    SmallVector<SPIRVValue *, 16> Extracts(ArrSize);
+    for (unsigned Idx = 0; Idx < ArrSize; ++Idx) {
+      Extracts[Idx] = BM->addVectorExtractDynamicInst(
+          VectorVal, BM->addIntegerConstant(I32, Idx), BB);
+    }
+    SPIRVValue *V = BM->addBinaryInst(Op, StartingVal->getType(), StartingVal,
+                                      Extracts[0], BB);
+    for (unsigned Idx = 1; Idx < ArrSize; ++Idx) {
+      V = BM->addBinaryInst(Op, StartingVal->getType(), V, Extracts[Idx], BB);
+    }
+    return V;
+  }
   case Intrinsic::vector_reduce_smax:
   case Intrinsic::vector_reduce_smin:
   case Intrinsic::vector_reduce_umax:
@@ -4108,25 +4127,6 @@ SPIRVValue *LLVMToSPIRVBase::transIntrinsicInst(IntrinsicInst *II,
     }
     return Extracts[0];
   }
-  case Intrinsic::vector_reduce_fadd: {
-    SPIRVValue *VectorVal = transValue(II->getArgOperand(1), BB);
-    VectorType *VecType = cast<VectorType>(II->getArgOperand(1)->getType());
-    SPIRVValue *StartingVal = transValue(II->getArgOperand(0), BB);
-    SPIRVTypeInt *I32 = BM->addIntegerType(32);
-    unsigned ArrSize = VecType->getElementCount().getFixedValue();
-    SmallVector<SPIRVValue *, 16> Extracts(ArrSize);
-    for (unsigned Idx = 0; Idx < ArrSize; ++Idx) {
-      Extracts[Idx] = BM->addVectorExtractDynamicInst(
-          VectorVal, BM->addIntegerConstant(I32, Idx), BB);
-    }
-    SPIRVValue *V = BM->addBinaryInst(OpFAdd, StartingVal->getType(),
-                                      StartingVal, Extracts[0], BB);
-    for (unsigned Idx = 1; Idx < ArrSize; ++Idx) {
-      V = BM->addBinaryInst(OpFAdd, StartingVal->getType(), V, Extracts[Idx],
-                            BB);
-    }
-    return V;
-  }
   case Intrinsic::memset: {
     // Generally there is no direct mapping of memset to SPIR-V.  But it turns
     // out that memset is emitted by Clang for initialization in default