diff --git a/llvm/lib/Target/AArch64/AArch64TargetTransformInfo.cpp b/llvm/lib/Target/AArch64/AArch64TargetTransformInfo.cpp --- a/llvm/lib/Target/AArch64/AArch64TargetTransformInfo.cpp +++ b/llvm/lib/Target/AArch64/AArch64TargetTransformInfo.cpp @@ -1279,6 +1279,10 @@ { TTI::SK_Reverse, MVT::nxv8bf16, 1 }, { TTI::SK_Reverse, MVT::nxv4f32, 1 }, { TTI::SK_Reverse, MVT::nxv2f64, 1 }, + { TTI::SK_Reverse, MVT::nxv16i1, 1 }, + { TTI::SK_Reverse, MVT::nxv8i1, 1 }, + { TTI::SK_Reverse, MVT::nxv4i1, 1 }, + { TTI::SK_Reverse, MVT::nxv2i1, 1 }, }; std::pair LT = TLI->getTypeLegalizationCost(DL, Tp); if (const auto *Entry = CostTableLookup(ShuffleTbl, Kind, LT.second)) diff --git a/llvm/test/Analysis/CostModel/AArch64/sve-intrinsics.ll b/llvm/test/Analysis/CostModel/AArch64/sve-intrinsics.ll --- a/llvm/test/Analysis/CostModel/AArch64/sve-intrinsics.ll +++ b/llvm/test/Analysis/CostModel/AArch64/sve-intrinsics.ll @@ -133,6 +133,11 @@ ; CHECK-NEXT: Cost Model: Found an estimated cost of 2 for instruction: %reverse_nxv4f64 = call @llvm.experimental.vector.reverse.nxv4f64( undef) ; CHECK-NEXT: Cost Model: Found an estimated cost of 1 for instruction: %reverse_nxv8bf16 = call @llvm.experimental.vector.reverse.nxv8bf16( undef) ; CHECK-NEXT: Cost Model: Found an estimated cost of 2 for instruction: %reverse_nxv16bf16 = call @llvm.experimental.vector.reverse.nxv16bf16( undef) +; CHECK-NEXT: Cost Model: Found an estimated cost of 1 for instruction: %reverse_nxv16i1 = call @llvm.experimental.vector.reverse.nxv16i1( undef) +; CHECK-NEXT: Cost Model: Found an estimated cost of 1 for instruction: %reverse_nxv8i1 = call @llvm.experimental.vector.reverse.nxv8i1( undef) +; CHECK-NEXT: Cost Model: Found an estimated cost of 1 for instruction: %reverse_nxv4i1 = call @llvm.experimental.vector.reverse.nxv4i1( undef) +; CHECK-NEXT: Cost Model: Found an estimated cost of 1 for instruction: %reverse_nxv2i1 = call @llvm.experimental.vector.reverse.nxv2i1( undef) + %reverse_nxv16i8 = call @llvm.experimental.vector.reverse.nxv16i8( undef) %reverse_nxv32i8 = call @llvm.experimental.vector.reverse.nxv32i8( undef) %reverse_nxv8i16 = call @llvm.experimental.vector.reverse.nxv8i16( undef) @@ -149,6 +154,10 @@ %reverse_nxv4f64 = call @llvm.experimental.vector.reverse.nxv4f64( undef) %reverse_nxv8bf16 = call @llvm.experimental.vector.reverse.nxv8bf16( undef) %reverse_nxv16bf16 = call @llvm.experimental.vector.reverse.nxv16bf16( undef) + %reverse_nxv16i1 = call @llvm.experimental.vector.reverse.nxv16i1( undef) + %reverse_nxv8i1 = call @llvm.experimental.vector.reverse.nxv8i1( undef) + %reverse_nxv4i1 = call @llvm.experimental.vector.reverse.nxv4i1( undef) + %reverse_nxv2i1 = call @llvm.experimental.vector.reverse.nxv2i1( undef) ret void } declare @llvm.experimental.vector.reverse.nxv16i8() @@ -167,5 +176,9 @@ declare @llvm.experimental.vector.reverse.nxv4f64() declare @llvm.experimental.vector.reverse.nxv8bf16() declare @llvm.experimental.vector.reverse.nxv16bf16() +declare @llvm.experimental.vector.reverse.nxv16i1() +declare @llvm.experimental.vector.reverse.nxv8i1() +declare @llvm.experimental.vector.reverse.nxv4i1() +declare @llvm.experimental.vector.reverse.nxv2i1() attributes #0 = { "target-features"="+sve,+bf16" }