Skip to content

Commit 47dc32e

Browse files
committed
refactor. nfc
1 parent 728686f commit 47dc32e

File tree

1 file changed

+15
-18
lines changed

1 file changed

+15
-18
lines changed

mlir/lib/Conversion/AMDGPUToROCDL/AMDGPUToROCDL.cpp

Lines changed: 15 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -1701,9 +1701,6 @@ LogicalResult ScaledExtPacked816OpLowering::matchAndRewrite(
17011701
auto targetType = cast<VectorType>(op.getResult().getType());
17021702
auto destElemType = cast<FloatType>(targetType.getElementType());
17031703
Location loc = op.getLoc();
1704-
// %scale: vector<4xf8E8M0FNU>
1705-
// ===========================
1706-
// %scale: i32
17071704
IntegerType i32 = rewriter.getI32Type();
17081705
Value castedScale =
17091706
LLVM::BitcastOp::create(rewriter, loc, i32, adaptor.getScale());
@@ -1730,31 +1727,31 @@ LogicalResult ScaledExtPacked816OpLowering::matchAndRewrite(
17301727
Value castedSource =
17311728
LLVM::BitcastOp::create(rewriter, loc, packedType, source);
17321729

1733-
if (isa<Float4E2M1FNType>(srcElemType) and destElemType.isF16()) {
1730+
if (isa<Float4E2M1FNType>(srcElemType) && destElemType.isF16()) {
17341731
rewriter.replaceOpWithNewOp<ROCDL::CvtPkScalePk8F16Fp4Op>(
17351732
op, op.getResult().getType(), castedSource, castedScale, scaleSel);
1736-
} else if (isa<Float8E4M3FNType>(srcElemType) and destElemType.isF16()) {
1733+
} else if (isa<Float8E4M3FNType>(srcElemType) && destElemType.isF16()) {
17371734
rewriter.replaceOpWithNewOp<ROCDL::CvtPkScalePk8F16Fp8Op>(
17381735
op, op.getResult().getType(), castedSource, castedScale, scaleSel);
1739-
} else if (isa<Float8E5M2Type>(srcElemType) and destElemType.isF16()) {
1736+
} else if (isa<Float8E5M2Type>(srcElemType) && destElemType.isF16()) {
17401737
rewriter.replaceOpWithNewOp<ROCDL::CvtPkScalePk8F16Bf8Op>(
17411738
op, op.getResult().getType(), castedSource, castedScale, scaleSel);
1742-
} else if (isa<Float4E2M1FNType>(srcElemType) and destElemType.isBF16()) {
1739+
} else if (isa<Float4E2M1FNType>(srcElemType) && destElemType.isBF16()) {
17431740
rewriter.replaceOpWithNewOp<ROCDL::CvtPkScalePk8Bf16Fp4Op>(
17441741
op, op.getResult().getType(), castedSource, castedScale, scaleSel);
1745-
} else if (isa<Float8E4M3FNType>(srcElemType) and destElemType.isBF16()) {
1742+
} else if (isa<Float8E4M3FNType>(srcElemType) && destElemType.isBF16()) {
17461743
rewriter.replaceOpWithNewOp<ROCDL::CvtPkScalePk8Bf16Fp8Op>(
17471744
op, op.getResult().getType(), castedSource, castedScale, scaleSel);
1748-
} else if (isa<Float8E5M2Type>(srcElemType) and destElemType.isBF16()) {
1745+
} else if (isa<Float8E5M2Type>(srcElemType) && destElemType.isBF16()) {
17491746
rewriter.replaceOpWithNewOp<ROCDL::CvtPkScalePk8Bf16Bf8Op>(
17501747
op, op.getResult().getType(), castedSource, castedScale, scaleSel);
1751-
} else if (isa<Float4E2M1FNType>(srcElemType) and destElemType.isF32()) {
1748+
} else if (isa<Float4E2M1FNType>(srcElemType) && destElemType.isF32()) {
17521749
rewriter.replaceOpWithNewOp<ROCDL::CvtPkScalePk8F32Fp4Op>(
17531750
op, op.getResult().getType(), castedSource, castedScale, scaleSel);
1754-
} else if (isa<Float8E4M3FNType>(srcElemType) and destElemType.isF32()) {
1751+
} else if (isa<Float8E4M3FNType>(srcElemType) && destElemType.isF32()) {
17551752
rewriter.replaceOpWithNewOp<ROCDL::CvtPkScalePk8F32Fp8Op>(
17561753
op, op.getResult().getType(), castedSource, castedScale, scaleSel);
1757-
} else if (isa<Float8E5M2Type>(srcElemType) and destElemType.isF32()) {
1754+
} else if (isa<Float8E5M2Type>(srcElemType) && destElemType.isF32()) {
17581755
rewriter.replaceOpWithNewOp<ROCDL::CvtPkScalePk8F32Bf8Op>(
17591756
op, op.getResult().getType(), castedSource, castedScale, scaleSel);
17601757
}
@@ -1764,22 +1761,22 @@ LogicalResult ScaledExtPacked816OpLowering::matchAndRewrite(
17641761
// largeT = [F16, Bf16, F32]
17651762
//
17661763
// CvtPkScalePk16${largeT}${smallT}
1767-
else if (isa<Float6E2M3FNType>(srcElemType) and destElemType.isF16()) {
1764+
else if (isa<Float6E2M3FNType>(srcElemType) && destElemType.isF16()) {
17681765
rewriter.replaceOpWithNewOp<ROCDL::CvtPkScalePk16F16Fp6Op>(
17691766
op, op.getResult().getType(), castedSource, castedScale, scaleSel);
1770-
} else if (isa<Float6E3M2FNType>(srcElemType) and destElemType.isF16()) {
1767+
} else if (isa<Float6E3M2FNType>(srcElemType) && destElemType.isF16()) {
17711768
rewriter.replaceOpWithNewOp<ROCDL::CvtPkScalePk16F16Bf6Op>(
17721769
op, op.getResult().getType(), castedSource, castedScale, scaleSel);
1773-
} else if (isa<Float6E2M3FNType>(srcElemType) and destElemType.isBF16()) {
1770+
} else if (isa<Float6E2M3FNType>(srcElemType) && destElemType.isBF16()) {
17741771
rewriter.replaceOpWithNewOp<ROCDL::CvtPkScalePk16Bf16Fp6Op>(
17751772
op, op.getResult().getType(), castedSource, castedScale, scaleSel);
1776-
} else if (isa<Float6E3M2FNType>(srcElemType) and destElemType.isBF16()) {
1773+
} else if (isa<Float6E3M2FNType>(srcElemType) && destElemType.isBF16()) {
17771774
rewriter.replaceOpWithNewOp<ROCDL::CvtPkScalePk16Bf16Bf6Op>(
17781775
op, op.getResult().getType(), castedSource, castedScale, scaleSel);
1779-
} else if (isa<Float6E2M3FNType>(srcElemType) and destElemType.isF32()) {
1776+
} else if (isa<Float6E2M3FNType>(srcElemType) && destElemType.isF32()) {
17801777
rewriter.replaceOpWithNewOp<ROCDL::CvtPkScalePk16F32Fp6Op>(
17811778
op, op.getResult().getType(), castedSource, castedScale, scaleSel);
1782-
} else if (isa<Float6E3M2FNType>(srcElemType) and destElemType.isF32()) {
1779+
} else if (isa<Float6E3M2FNType>(srcElemType) && destElemType.isF32()) {
17831780
rewriter.replaceOpWithNewOp<ROCDL::CvtPkScalePk16F32Bf6Op>(
17841781
op, op.getResult().getType(), castedSource, castedScale, scaleSel);
17851782
} else {

0 commit comments

Comments
 (0)