Skip to content

Commit 2876dbc

Browse files
authored
[AArch64] Don't allow mixed partial reductions without i8mm (#137602)
Partial reductions with mixed extends should only be allowed if i8mm is present.
1 parent 7ffaaf4 commit 2876dbc

File tree

2 files changed

+347
-101
lines changed

2 files changed

+347
-101
lines changed

llvm/lib/Target/AArch64/AArch64TargetTransformInfo.cpp

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -5401,11 +5401,10 @@ InstructionCost AArch64TTIImpl::getPartialReductionCost(
54015401
} else
54025402
return Invalid;
54035403

5404-
// AArch64 supports lowering mixed extensions to a usdot but only if the
5405-
// i8mm or sve/streaming features are available.
5404+
// AArch64 supports lowering mixed fixed-width extensions to a usdot but only
5405+
// if the i8mm feature is available.
54065406
if (OpAExtend == TTI::PR_None || OpBExtend == TTI::PR_None ||
5407-
(OpAExtend != OpBExtend && !ST->hasMatMulInt8() &&
5408-
!ST->isSVEorStreamingSVEAvailable()))
5407+
(OpAExtend != OpBExtend && !ST->hasMatMulInt8()))
54095408
return Invalid;
54105409

54115410
if (!BinOp || *BinOp != Instruction::Mul)

0 commit comments

Comments
 (0)