Skip to content

Commit 605c99e

Browse files
committed
fix format
1 parent 7072bc1 commit 605c99e

File tree

1 file changed

+14
-7
lines changed

1 file changed

+14
-7
lines changed

mlir/lib/Dialect/XeGPU/IR/XeGPUOps.cpp

Lines changed: 14 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -101,7 +101,10 @@ static bool isEvenDistributed(llvm::ArrayRef<int64_t> shape,
101101
return true;
102102
}
103103

104-
static LogicalResult isValidGatherScatterParams(Type maskTy, VectorType valueTy, TensorDescType tdescTy, UnitAttr transposeAttr, function_ref<InFlightDiagnostic()> emitError) {
104+
static LogicalResult
105+
isValidGatherScatterParams(Type maskTy, VectorType valueTy,
106+
TensorDescType tdescTy, UnitAttr transposeAttr,
107+
function_ref<InFlightDiagnostic()> emitError) {
105108

106109
if (!tdescTy.isScattered())
107110
return emitError() << "Expects a scattered TensorDesc.";
@@ -115,10 +118,12 @@ static LogicalResult isValidGatherScatterParams(Type maskTy, VectorType valueTy,
115118
auto chunkSize = tdescTy.getChunkSize();
116119

117120
if (valueTy.getElementType() != tdescTy.getElementType())
118-
return emitError() << "Value should have the same element type as TensorDesc.";
121+
return emitError()
122+
<< "Value should have the same element type as TensorDesc.";
119123

120124
if (tdescShape[0] != maskShape[0])
121-
return emitError() << "dim-0 of the Mask and TensorDesc should be the same.";
125+
return emitError()
126+
<< "dim-0 of the Mask and TensorDesc should be the same.";
122127

123128
// a valid shape for SIMT case
124129
if (valueTy.getRank() == 1 && valueTy.getNumElements() == chunkSize) {
@@ -568,8 +573,9 @@ LogicalResult LoadGatherOp::verify() {
568573
if (!isReadHintOrNone(getL3HintAttr()))
569574
return emitOpError("invalid l3_hint: ") << getL3HintAttr();
570575

571-
return isValidGatherScatterParams(maskTy, valueTy, tdescTy, getTransposeAttr(),
572-
[&]() { return emitOpError(); });
576+
return isValidGatherScatterParams(maskTy, valueTy, tdescTy,
577+
getTransposeAttr(),
578+
[&]() { return emitOpError(); });
573579
}
574580

575581
//===----------------------------------------------------------------------===//
@@ -589,8 +595,9 @@ LogicalResult StoreScatterOp::verify() {
589595
if (!isWriteHintOrNone(getL3HintAttr()))
590596
return emitOpError("invalid l3_hint: ") << getL3HintAttr();
591597

592-
return isValidGatherScatterParams(maskTy, valueTy, tdescTy, getTransposeAttr(),
593-
[&]() { return emitOpError(); });
598+
return isValidGatherScatterParams(maskTy, valueTy, tdescTy,
599+
getTransposeAttr(),
600+
[&]() { return emitOpError(); });
594601
}
595602

596603
//===----------------------------------------------------------------------===//

0 commit comments

Comments
 (0)