[llvm] [SLPVectorizer][NVPTX] Customize getBuildVectorCost for NVPTX (PR #128077)

[via llvm-commits]

github-actions[bot] wrote:

<!--LLVM CODE FORMAT COMMENT: {clang-format}-->


:warning: C/C++ code formatter, clang-format found issues in your code. :warning:

<details>
<summary>
You can test this locally with the following command:
</summary>

``````````bash
git-clang-format --diff d1dde17ab079152d8cf2d8953b90cdbf5a1dd78c dca12e57e39d560a887350959cd102ed2304b57e --extensions cpp,h -- llvm/include/llvm/Analysis/TargetTransformInfo.h llvm/include/llvm/Analysis/TargetTransformInfoImpl.h llvm/include/llvm/CodeGen/BasicTTIImpl.h llvm/lib/Analysis/TargetTransformInfo.cpp llvm/lib/Target/NVPTX/NVPTXTargetTransformInfo.h llvm/lib/Transforms/Vectorize/SLPVectorizer.cpp
``````````

</details>

<details>
<summary>
View the diff from clang-format here.
</summary>

``````````diff
diff --git a/llvm/include/llvm/Analysis/TargetTransformInfo.h b/llvm/include/llvm/Analysis/TargetTransformInfo.h
index 7fbfc58684..c3a17a6fdb 100644
--- a/llvm/include/llvm/Analysis/TargetTransformInfo.h
+++ b/llvm/include/llvm/Analysis/TargetTransformInfo.h
@@ -1481,7 +1481,9 @@ public:
 
   /// \return The cost of ISD::BUILD_VECTOR, or nullopt if the cost should be
   /// inferred from insert element and shuffle ops.
-  std::optional<InstructionCost> getBuildVectorCost(VectorType *VecTy, ArrayRef<Value*> Operands, TargetCostKind CostKind) const;
+  std::optional<InstructionCost>
+  getBuildVectorCost(VectorType *VecTy, ArrayRef<Value *> Operands,
+                     TargetCostKind CostKind) const;
 
   /// \return The cost of replication shuffle of \p VF elements typed \p EltTy
   /// \p ReplicationFactor times.
@@ -2228,7 +2230,9 @@ public:
                                              TTI::TargetCostKind CostKind,
                                              unsigned Index) = 0;
 
-  virtual std::optional<InstructionCost> getBuildVectorCost(VectorType *VecTy, ArrayRef<Value*> Operands, TargetCostKind CostKind) = 0;
+  virtual std::optional<InstructionCost>
+  getBuildVectorCost(VectorType *VecTy, ArrayRef<Value *> Operands,
+                     TargetCostKind CostKind) = 0;
 
   virtual InstructionCost
   getReplicationShuffleCost(Type *EltTy, int ReplicationFactor, int VF,
@@ -2958,7 +2962,9 @@ public:
                                      unsigned Index) override {
     return Impl.getVectorInstrCost(I, Val, CostKind, Index);
   }
-  std::optional<InstructionCost> getBuildVectorCost(VectorType *VecTy, ArrayRef<Value*> Operands, TTI::TargetCostKind CostKind) override {
+  std::optional<InstructionCost>
+  getBuildVectorCost(VectorType *VecTy, ArrayRef<Value *> Operands,
+                     TTI::TargetCostKind CostKind) override {
     return Impl.getBuildVectorCost(VecTy, Operands, CostKind);
   }
 
diff --git a/llvm/include/llvm/Analysis/TargetTransformInfoImpl.h b/llvm/include/llvm/Analysis/TargetTransformInfoImpl.h
index ab720ca0be..f7ef03bea2 100644
--- a/llvm/include/llvm/Analysis/TargetTransformInfoImpl.h
+++ b/llvm/include/llvm/Analysis/TargetTransformInfoImpl.h
@@ -739,7 +739,9 @@ public:
     return 1;
   }
 
-  std::optional<InstructionCost> getBuildVectorCost(VectorType *Val, ArrayRef<Value*> Operands, TTI::TargetCostKind CostKind) const {
+  std::optional<InstructionCost>
+  getBuildVectorCost(VectorType *Val, ArrayRef<Value *> Operands,
+                     TTI::TargetCostKind CostKind) const {
     return std::nullopt;
   }
 
diff --git a/llvm/include/llvm/CodeGen/BasicTTIImpl.h b/llvm/include/llvm/CodeGen/BasicTTIImpl.h
index a97ffdb740..a58c8dcee4 100644
--- a/llvm/include/llvm/CodeGen/BasicTTIImpl.h
+++ b/llvm/include/llvm/CodeGen/BasicTTIImpl.h
@@ -1373,7 +1373,9 @@ public:
                                        Op1);
   }
 
-  std::optional<InstructionCost> getBuildVectorCost(VectorType *VecTy, ArrayRef<Value*> Operands, TTI::TargetCostKind CostKind) {
+  std::optional<InstructionCost>
+  getBuildVectorCost(VectorType *VecTy, ArrayRef<Value *> Operands,
+                     TTI::TargetCostKind CostKind) {
     return std::nullopt;
   }
 
diff --git a/llvm/lib/Analysis/TargetTransformInfo.cpp b/llvm/lib/Analysis/TargetTransformInfo.cpp
index 65e1ca94dc..69b8f6706b 100644
--- a/llvm/lib/Analysis/TargetTransformInfo.cpp
+++ b/llvm/lib/Analysis/TargetTransformInfo.cpp
@@ -1123,7 +1123,10 @@ InstructionCost TargetTransformInfo::getInsertExtractValueCost(
   return Cost;
 }
 
-std::optional<InstructionCost> TargetTransformInfo::getBuildVectorCost(VectorType *VecTy, ArrayRef<Value*> Operands, TargetCostKind CostKind) const {
+std::optional<InstructionCost>
+TargetTransformInfo::getBuildVectorCost(VectorType *VecTy,
+                                        ArrayRef<Value *> Operands,
+                                        TargetCostKind CostKind) const {
   return TTIImpl->getBuildVectorCost(VecTy, Operands, CostKind);
 }
 
diff --git a/llvm/lib/Target/NVPTX/NVPTXTargetTransformInfo.h b/llvm/lib/Target/NVPTX/NVPTXTargetTransformInfo.h
index 25a6e2dbbc..cbe94a2e82 100644
--- a/llvm/lib/Target/NVPTX/NVPTXTargetTransformInfo.h
+++ b/llvm/lib/Target/NVPTX/NVPTXTargetTransformInfo.h
@@ -16,8 +16,8 @@
 #ifndef LLVM_LIB_TARGET_NVPTX_NVPTXTARGETTRANSFORMINFO_H
 #define LLVM_LIB_TARGET_NVPTX_NVPTXTARGETTRANSFORMINFO_H
 
-#include "NVPTXTargetMachine.h"
 #include "MCTargetDesc/NVPTXBaseInfo.h"
+#include "NVPTXTargetMachine.h"
 #include "NVPTXUtilities.h"
 #include "llvm/Analysis/TargetTransformInfo.h"
 #include "llvm/CodeGen/BasicTTIImpl.h"
@@ -101,19 +101,21 @@ public:
       TTI::OperandValueInfo Op2Info = {TTI::OK_AnyValue, TTI::OP_None},
       ArrayRef<const Value *> Args = {}, const Instruction *CxtI = nullptr);
 
-  std::optional<InstructionCost> getBuildVectorCost(VectorType *VecTy, ArrayRef<Value*> Operands, TTI::TargetCostKind CostKind) {
+  std::optional<InstructionCost>
+  getBuildVectorCost(VectorType *VecTy, ArrayRef<Value *> Operands,
+                     TTI::TargetCostKind CostKind) {
     if (CostKind != TTI::TCK_RecipThroughput)
       return std::nullopt;
     auto VT = getTLI()->getValueType(DL, VecTy);
     if (all_of(Operands, [](Value *Op) { return isa<Constant>(Op); }))
       return TTI::TCC_Free;
     if (Isv2x16VT(VT))
-      return 1;  // Single vector mov
+      return 1; // Single vector mov
     if (VT == MVT::v4i8) {
-      InstructionCost Cost = 3;  // 3 x PRMT
-      for (auto *Op: Operands)
+      InstructionCost Cost = 3; // 3 x PRMT
+      for (auto *Op : Operands)
         if (!isa<Constant>(Op))
-          Cost += 1;  // zext operand to i32
+          Cost += 1; // zext operand to i32
       return Cost;
     }
     return std::nullopt;
diff --git a/llvm/lib/Transforms/Vectorize/SLPVectorizer.cpp b/llvm/lib/Transforms/Vectorize/SLPVectorizer.cpp
index c420a2f2eb..cfe7bbc641 100644
--- a/llvm/lib/Transforms/Vectorize/SLPVectorizer.cpp
+++ b/llvm/lib/Transforms/Vectorize/SLPVectorizer.cpp
@@ -10203,7 +10203,8 @@ class BoUpSLP::ShuffleCostEstimator : public BaseShuffleAnalysis {
     if ((!Root && allConstant(VL)) || all_of(VL, IsaPred<UndefValue>))
       return TTI::TCC_Free;
     auto *VecTy = getWidenedType(ScalarTy, VL.size());
-    if (auto Cost = TTI.getBuildVectorCost(VecTy, VL, CostKind); Cost.has_value())
+    if (auto Cost = TTI.getBuildVectorCost(VecTy, VL, CostKind);
+        Cost.has_value())
       return *Cost;
     InstructionCost GatherCost = 0;
     SmallVector<Value *> Gathers(VL);

``````````

</details>


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