@@ -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