Skip to content

Commit 4697fc1

Browse files
committed
Fix subtarget check
1 parent da20b2a commit 4697fc1

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

llvm/lib/Target/AArch64/AArch64ISelLowering.cpp

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -21777,7 +21777,7 @@ SDValue tryLowerPartialReductionToDot(SDNode *N,
2177721777
Intrinsic::experimental_vector_partial_reduce_add &&
2177821778
"Expected a partial reduction node");
2177921779

21780-
if (!Subtarget->isSVEAvailable() && !Subtarget->isNeonAvailable())
21780+
if (!Subtarget->isSVEorStreamingSVEAvailable())
2178121781
return SDValue();
2178221782

2178321783
SDLoc DL(N);
@@ -21819,8 +21819,6 @@ SDValue tryLowerPartialReductionToDot(SDNode *N,
2181921819
// The wide type with four times as many elements as the reduced type. Should
2182021820
// be a vector of i32 or i64, the same as the fully-reduced type
2182121821
EVT WideType = MulOp.getValueType();
21822-
if (WideType.getScalarSizeInBits() != ReducedType.getScalarSizeInBits())
21823-
return SDValue();
2182421822

2182521823
// Dot products operate on chunks of four elements so there must be four times
2182621824
// as many elements in the wide type

0 commit comments

Comments
 (0)