diff --git a/llvm/lib/Transforms/Scalar/LowerMatrixIntrinsics.cpp b/llvm/lib/Transforms/Scalar/LowerMatrixIntrinsics.cpp --- a/llvm/lib/Transforms/Scalar/LowerMatrixIntrinsics.cpp +++ b/llvm/lib/Transforms/Scalar/LowerMatrixIntrinsics.cpp @@ -43,6 +43,11 @@ static cl::opt EnableShapePropagation("matrix-propagate-shape", cl::init(true)); +static cl::opt AllowContraction( + "matrix-allow-contraction", cl::init(false), cl::Hidden, + cl::desc("Allow the use of FMAs if available and profitable. This may " + "result in different results, due to less rounding error.")); + namespace { // Given an element poitner \p BasePtr to the start of a (sub) matrix, compute @@ -653,11 +658,24 @@ Value *createMulAdd(Value *Sum, Value *A, Value *B, bool UseFPOp, IRBuilder<> &Builder) { - Value *Mul = UseFPOp ? Builder.CreateFMul(A, B) : Builder.CreateMul(A, B); + if (!Sum) - return Mul; + return UseFPOp ? Builder.CreateFMul(A, B) : Builder.CreateMul(A, B); + + if (UseFPOp) { + if (AllowContraction) { + // Use fmuladd for floating point operations and let the backend decide + // if that's profitable. + Value *FMulAdd = Intrinsic::getDeclaration( + Func.getParent(), Intrinsic::fmuladd, A->getType()); + return Builder.CreateCall(FMulAdd, {A, B, Sum}); + } + Value *Mul = Builder.CreateFMul(A, B); + return Builder.CreateFAdd(Sum, Mul); + } - return UseFPOp ? Builder.CreateFAdd(Sum, Mul) : Builder.CreateAdd(Sum, Mul); + Value *Mul = Builder.CreateMul(A, B); + return Builder.CreateAdd(Sum, Mul); } /// Cache \p Matrix as result of \p Inst and update the uses of \p Inst. For diff --git a/llvm/test/Transforms/LowerMatrixIntrinsics/bigger-expressions-double-contraction.ll b/llvm/test/Transforms/LowerMatrixIntrinsics/bigger-expressions-double-contraction.ll new file mode 100644 --- /dev/null +++ b/llvm/test/Transforms/LowerMatrixIntrinsics/bigger-expressions-double-contraction.ll @@ -0,0 +1,476 @@ +; NOTE: Assertions have been autogenerated by utils/update_test_checks.py +; RUN: opt -lower-matrix-intrinsics -matrix-allow-contraction -S < %s | FileCheck %s +; RUN: opt -passes='lower-matrix-intrinsics' -matrix-allow-contraction -S < %s | FileCheck %s + + +define void @transpose_multiply(<9 x double>* %A.Ptr, <9 x double>* %B.Ptr, <9 x double>* %C.Ptr) { +; CHECK-LABEL: @transpose_multiply( +; CHECK-NEXT: entry: + +; Load columns of input matrixes %A and %B. + +; CHECK-NEXT: [[TMP0:%.*]] = bitcast <9 x double>* [[A_PTR:%.*]] to double* +; CHECK-NEXT: [[TMP1:%.*]] = bitcast double* [[TMP0]] to <3 x double>* +; CHECK-NEXT: [[TMP2:%.*]] = load <3 x double>, <3 x double>* [[TMP1]], align 8 +; CHECK-NEXT: [[TMP3:%.*]] = getelementptr double, double* [[TMP0]], i32 3 +; CHECK-NEXT: [[TMP4:%.*]] = bitcast double* [[TMP3]] to <3 x double>* +; CHECK-NEXT: [[TMP5:%.*]] = load <3 x double>, <3 x double>* [[TMP4]], align 8 +; CHECK-NEXT: [[TMP6:%.*]] = getelementptr double, double* [[TMP0]], i32 6 +; CHECK-NEXT: [[TMP7:%.*]] = bitcast double* [[TMP6]] to <3 x double>* +; CHECK-NEXT: [[TMP8:%.*]] = load <3 x double>, <3 x double>* [[TMP7]], align 8 +; CHECK-NEXT: [[TMP9:%.*]] = bitcast <9 x double>* [[B_PTR:%.*]] to double* +; CHECK-NEXT: [[TMP10:%.*]] = bitcast double* [[TMP9]] to <3 x double>* +; CHECK-NEXT: [[TMP11:%.*]] = load <3 x double>, <3 x double>* [[TMP10]], align 8 +; CHECK-NEXT: [[TMP12:%.*]] = getelementptr double, double* [[TMP9]], i32 3 +; CHECK-NEXT: [[TMP13:%.*]] = bitcast double* [[TMP12]] to <3 x double>* +; CHECK-NEXT: [[TMP14:%.*]] = load <3 x double>, <3 x double>* [[TMP13]], align 8 +; CHECK-NEXT: [[TMP15:%.*]] = getelementptr double, double* [[TMP9]], i32 6 +; CHECK-NEXT: [[TMP16:%.*]] = bitcast double* [[TMP15]] to <3 x double>* +; CHECK-NEXT: [[TMP17:%.*]] = load <3 x double>, <3 x double>* [[TMP16]], align 8 + +; Transpose %A. + +; CHECK-NEXT: [[TMP18:%.*]] = extractelement <3 x double> [[TMP2]], i64 0 +; CHECK-NEXT: [[TMP19:%.*]] = insertelement <3 x double> undef, double [[TMP18]], i64 0 +; CHECK-NEXT: [[TMP20:%.*]] = extractelement <3 x double> [[TMP5]], i64 0 +; CHECK-NEXT: [[TMP21:%.*]] = insertelement <3 x double> [[TMP19]], double [[TMP20]], i64 1 +; CHECK-NEXT: [[TMP22:%.*]] = extractelement <3 x double> [[TMP8]], i64 0 +; CHECK-NEXT: [[TMP23:%.*]] = insertelement <3 x double> [[TMP21]], double [[TMP22]], i64 2 +; CHECK-NEXT: [[TMP24:%.*]] = extractelement <3 x double> [[TMP2]], i64 1 +; CHECK-NEXT: [[TMP25:%.*]] = insertelement <3 x double> undef, double [[TMP24]], i64 0 +; CHECK-NEXT: [[TMP26:%.*]] = extractelement <3 x double> [[TMP5]], i64 1 +; CHECK-NEXT: [[TMP27:%.*]] = insertelement <3 x double> [[TMP25]], double [[TMP26]], i64 1 +; CHECK-NEXT: [[TMP28:%.*]] = extractelement <3 x double> [[TMP8]], i64 1 +; CHECK-NEXT: [[TMP29:%.*]] = insertelement <3 x double> [[TMP27]], double [[TMP28]], i64 2 +; CHECK-NEXT: [[TMP30:%.*]] = extractelement <3 x double> [[TMP2]], i64 2 +; CHECK-NEXT: [[TMP31:%.*]] = insertelement <3 x double> undef, double [[TMP30]], i64 0 +; CHECK-NEXT: [[TMP32:%.*]] = extractelement <3 x double> [[TMP5]], i64 2 +; CHECK-NEXT: [[TMP33:%.*]] = insertelement <3 x double> [[TMP31]], double [[TMP32]], i64 1 +; CHECK-NEXT: [[TMP34:%.*]] = extractelement <3 x double> [[TMP8]], i64 2 +; CHECK-NEXT: [[TMP35:%.*]] = insertelement <3 x double> [[TMP33]], double [[TMP34]], i64 2 +; CHECK-NEXT: [[BLOCK:%.*]] = shufflevector <3 x double> [[TMP23]], <3 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP36:%.*]] = extractelement <3 x double> [[TMP11]], i64 0 + +; Lower multiply(transpose(%A), %B) + +; CHECK-NEXT: [[SPLAT_SPLATINSERT:%.*]] = insertelement <1 x double> undef, double [[TMP36]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP37:%.*]] = fmul <1 x double> [[BLOCK]], [[SPLAT_SPLAT]] +; CHECK-NEXT: [[BLOCK1:%.*]] = shufflevector <3 x double> [[TMP29]], <3 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP38:%.*]] = extractelement <3 x double> [[TMP11]], i64 1 +; CHECK-NEXT: [[SPLAT_SPLATINSERT2:%.*]] = insertelement <1 x double> undef, double [[TMP38]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT3:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT2]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP39:%.*]] = call <1 x double> @llvm.fmuladd.v1f64(<1 x double> [[BLOCK1]], <1 x double> [[SPLAT_SPLAT3]], <1 x double> [[TMP37]]) +; CHECK-NEXT: [[BLOCK4:%.*]] = shufflevector <3 x double> [[TMP35]], <3 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP40:%.*]] = extractelement <3 x double> [[TMP11]], i64 2 +; CHECK-NEXT: [[SPLAT_SPLATINSERT5:%.*]] = insertelement <1 x double> undef, double [[TMP40]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT6:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT5]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP41:%.*]] = call <1 x double> @llvm.fmuladd.v1f64(<1 x double> [[BLOCK4]], <1 x double> [[SPLAT_SPLAT6]], <1 x double> [[TMP39]]) +; CHECK-NEXT: [[TMP42:%.*]] = shufflevector <1 x double> [[TMP41]], <1 x double> undef, <3 x i32> +; CHECK-NEXT: [[TMP43:%.*]] = shufflevector <3 x double> undef, <3 x double> [[TMP42]], <3 x i32> +; CHECK-NEXT: [[BLOCK7:%.*]] = shufflevector <3 x double> [[TMP23]], <3 x double> undef, <1 x i32> +; CHECK-NEXT: [[TMP44:%.*]] = extractelement <3 x double> [[TMP11]], i64 0 +; CHECK-NEXT: [[SPLAT_SPLATINSERT8:%.*]] = insertelement <1 x double> undef, double [[TMP44]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT9:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT8]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP45:%.*]] = fmul <1 x double> [[BLOCK7]], [[SPLAT_SPLAT9]] +; CHECK-NEXT: [[BLOCK10:%.*]] = shufflevector <3 x double> [[TMP29]], <3 x double> undef, <1 x i32> +; CHECK-NEXT: [[TMP46:%.*]] = extractelement <3 x double> [[TMP11]], i64 1 +; CHECK-NEXT: [[SPLAT_SPLATINSERT11:%.*]] = insertelement <1 x double> undef, double [[TMP46]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT12:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT11]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP47:%.*]] = call <1 x double> @llvm.fmuladd.v1f64(<1 x double> [[BLOCK10]], <1 x double> [[SPLAT_SPLAT12]], <1 x double> [[TMP45]]) +; CHECK-NEXT: [[BLOCK13:%.*]] = shufflevector <3 x double> [[TMP35]], <3 x double> undef, <1 x i32> +; CHECK-NEXT: [[TMP48:%.*]] = extractelement <3 x double> [[TMP11]], i64 2 +; CHECK-NEXT: [[SPLAT_SPLATINSERT14:%.*]] = insertelement <1 x double> undef, double [[TMP48]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT15:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT14]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP49:%.*]] = call <1 x double> @llvm.fmuladd.v1f64(<1 x double> [[BLOCK13]], <1 x double> [[SPLAT_SPLAT15]], <1 x double> [[TMP47]]) +; CHECK-NEXT: [[TMP50:%.*]] = shufflevector <1 x double> [[TMP49]], <1 x double> undef, <3 x i32> +; CHECK-NEXT: [[TMP51:%.*]] = shufflevector <3 x double> [[TMP43]], <3 x double> [[TMP50]], <3 x i32> +; CHECK-NEXT: [[BLOCK16:%.*]] = shufflevector <3 x double> [[TMP23]], <3 x double> undef, <1 x i32> +; CHECK-NEXT: [[TMP52:%.*]] = extractelement <3 x double> [[TMP11]], i64 0 +; CHECK-NEXT: [[SPLAT_SPLATINSERT17:%.*]] = insertelement <1 x double> undef, double [[TMP52]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT18:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT17]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP53:%.*]] = fmul <1 x double> [[BLOCK16]], [[SPLAT_SPLAT18]] +; CHECK-NEXT: [[BLOCK19:%.*]] = shufflevector <3 x double> [[TMP29]], <3 x double> undef, <1 x i32> +; CHECK-NEXT: [[TMP54:%.*]] = extractelement <3 x double> [[TMP11]], i64 1 +; CHECK-NEXT: [[SPLAT_SPLATINSERT20:%.*]] = insertelement <1 x double> undef, double [[TMP54]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT21:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT20]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP55:%.*]] = call <1 x double> @llvm.fmuladd.v1f64(<1 x double> [[BLOCK19]], <1 x double> [[SPLAT_SPLAT21]], <1 x double> [[TMP53]]) +; CHECK-NEXT: [[BLOCK22:%.*]] = shufflevector <3 x double> [[TMP35]], <3 x double> undef, <1 x i32> +; CHECK-NEXT: [[TMP56:%.*]] = extractelement <3 x double> [[TMP11]], i64 2 +; CHECK-NEXT: [[SPLAT_SPLATINSERT23:%.*]] = insertelement <1 x double> undef, double [[TMP56]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT24:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT23]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP57:%.*]] = call <1 x double> @llvm.fmuladd.v1f64(<1 x double> [[BLOCK22]], <1 x double> [[SPLAT_SPLAT24]], <1 x double> [[TMP55]]) +; CHECK-NEXT: [[TMP58:%.*]] = shufflevector <1 x double> [[TMP57]], <1 x double> undef, <3 x i32> +; CHECK-NEXT: [[TMP59:%.*]] = shufflevector <3 x double> [[TMP51]], <3 x double> [[TMP58]], <3 x i32> +; CHECK-NEXT: [[BLOCK25:%.*]] = shufflevector <3 x double> [[TMP23]], <3 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP60:%.*]] = extractelement <3 x double> [[TMP14]], i64 0 +; CHECK-NEXT: [[SPLAT_SPLATINSERT26:%.*]] = insertelement <1 x double> undef, double [[TMP60]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT27:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT26]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP61:%.*]] = fmul <1 x double> [[BLOCK25]], [[SPLAT_SPLAT27]] +; CHECK-NEXT: [[BLOCK28:%.*]] = shufflevector <3 x double> [[TMP29]], <3 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP62:%.*]] = extractelement <3 x double> [[TMP14]], i64 1 +; CHECK-NEXT: [[SPLAT_SPLATINSERT29:%.*]] = insertelement <1 x double> undef, double [[TMP62]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT30:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT29]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP63:%.*]] = call <1 x double> @llvm.fmuladd.v1f64(<1 x double> [[BLOCK28]], <1 x double> [[SPLAT_SPLAT30]], <1 x double> [[TMP61]]) +; CHECK-NEXT: [[BLOCK31:%.*]] = shufflevector <3 x double> [[TMP35]], <3 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP64:%.*]] = extractelement <3 x double> [[TMP14]], i64 2 +; CHECK-NEXT: [[SPLAT_SPLATINSERT32:%.*]] = insertelement <1 x double> undef, double [[TMP64]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT33:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT32]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP65:%.*]] = call <1 x double> @llvm.fmuladd.v1f64(<1 x double> [[BLOCK31]], <1 x double> [[SPLAT_SPLAT33]], <1 x double> [[TMP63]]) +; CHECK-NEXT: [[TMP66:%.*]] = shufflevector <1 x double> [[TMP65]], <1 x double> undef, <3 x i32> +; CHECK-NEXT: [[TMP67:%.*]] = shufflevector <3 x double> undef, <3 x double> [[TMP66]], <3 x i32> +; CHECK-NEXT: [[BLOCK34:%.*]] = shufflevector <3 x double> [[TMP23]], <3 x double> undef, <1 x i32> +; CHECK-NEXT: [[TMP68:%.*]] = extractelement <3 x double> [[TMP14]], i64 0 +; CHECK-NEXT: [[SPLAT_SPLATINSERT35:%.*]] = insertelement <1 x double> undef, double [[TMP68]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT36:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT35]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP69:%.*]] = fmul <1 x double> [[BLOCK34]], [[SPLAT_SPLAT36]] +; CHECK-NEXT: [[BLOCK37:%.*]] = shufflevector <3 x double> [[TMP29]], <3 x double> undef, <1 x i32> +; CHECK-NEXT: [[TMP70:%.*]] = extractelement <3 x double> [[TMP14]], i64 1 +; CHECK-NEXT: [[SPLAT_SPLATINSERT38:%.*]] = insertelement <1 x double> undef, double [[TMP70]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT39:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT38]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP71:%.*]] = call <1 x double> @llvm.fmuladd.v1f64(<1 x double> [[BLOCK37]], <1 x double> [[SPLAT_SPLAT39]], <1 x double> [[TMP69]]) +; CHECK-NEXT: [[BLOCK40:%.*]] = shufflevector <3 x double> [[TMP35]], <3 x double> undef, <1 x i32> +; CHECK-NEXT: [[TMP72:%.*]] = extractelement <3 x double> [[TMP14]], i64 2 +; CHECK-NEXT: [[SPLAT_SPLATINSERT41:%.*]] = insertelement <1 x double> undef, double [[TMP72]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT42:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT41]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP73:%.*]] = call <1 x double> @llvm.fmuladd.v1f64(<1 x double> [[BLOCK40]], <1 x double> [[SPLAT_SPLAT42]], <1 x double> [[TMP71]]) +; CHECK-NEXT: [[TMP74:%.*]] = shufflevector <1 x double> [[TMP73]], <1 x double> undef, <3 x i32> +; CHECK-NEXT: [[TMP75:%.*]] = shufflevector <3 x double> [[TMP67]], <3 x double> [[TMP74]], <3 x i32> +; CHECK-NEXT: [[BLOCK43:%.*]] = shufflevector <3 x double> [[TMP23]], <3 x double> undef, <1 x i32> +; CHECK-NEXT: [[TMP76:%.*]] = extractelement <3 x double> [[TMP14]], i64 0 +; CHECK-NEXT: [[SPLAT_SPLATINSERT44:%.*]] = insertelement <1 x double> undef, double [[TMP76]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT45:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT44]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP77:%.*]] = fmul <1 x double> [[BLOCK43]], [[SPLAT_SPLAT45]] +; CHECK-NEXT: [[BLOCK46:%.*]] = shufflevector <3 x double> [[TMP29]], <3 x double> undef, <1 x i32> +; CHECK-NEXT: [[TMP78:%.*]] = extractelement <3 x double> [[TMP14]], i64 1 +; CHECK-NEXT: [[SPLAT_SPLATINSERT47:%.*]] = insertelement <1 x double> undef, double [[TMP78]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT48:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT47]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP79:%.*]] = call <1 x double> @llvm.fmuladd.v1f64(<1 x double> [[BLOCK46]], <1 x double> [[SPLAT_SPLAT48]], <1 x double> [[TMP77]]) +; CHECK-NEXT: [[BLOCK49:%.*]] = shufflevector <3 x double> [[TMP35]], <3 x double> undef, <1 x i32> +; CHECK-NEXT: [[TMP80:%.*]] = extractelement <3 x double> [[TMP14]], i64 2 +; CHECK-NEXT: [[SPLAT_SPLATINSERT50:%.*]] = insertelement <1 x double> undef, double [[TMP80]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT51:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT50]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP81:%.*]] = call <1 x double> @llvm.fmuladd.v1f64(<1 x double> [[BLOCK49]], <1 x double> [[SPLAT_SPLAT51]], <1 x double> [[TMP79]]) +; CHECK-NEXT: [[TMP82:%.*]] = shufflevector <1 x double> [[TMP81]], <1 x double> undef, <3 x i32> +; CHECK-NEXT: [[TMP83:%.*]] = shufflevector <3 x double> [[TMP75]], <3 x double> [[TMP82]], <3 x i32> +; CHECK-NEXT: [[BLOCK52:%.*]] = shufflevector <3 x double> [[TMP23]], <3 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP84:%.*]] = extractelement <3 x double> [[TMP17]], i64 0 +; CHECK-NEXT: [[SPLAT_SPLATINSERT53:%.*]] = insertelement <1 x double> undef, double [[TMP84]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT54:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT53]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP85:%.*]] = fmul <1 x double> [[BLOCK52]], [[SPLAT_SPLAT54]] +; CHECK-NEXT: [[BLOCK55:%.*]] = shufflevector <3 x double> [[TMP29]], <3 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP86:%.*]] = extractelement <3 x double> [[TMP17]], i64 1 +; CHECK-NEXT: [[SPLAT_SPLATINSERT56:%.*]] = insertelement <1 x double> undef, double [[TMP86]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT57:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT56]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP87:%.*]] = call <1 x double> @llvm.fmuladd.v1f64(<1 x double> [[BLOCK55]], <1 x double> [[SPLAT_SPLAT57]], <1 x double> [[TMP85]]) +; CHECK-NEXT: [[BLOCK58:%.*]] = shufflevector <3 x double> [[TMP35]], <3 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP88:%.*]] = extractelement <3 x double> [[TMP17]], i64 2 +; CHECK-NEXT: [[SPLAT_SPLATINSERT59:%.*]] = insertelement <1 x double> undef, double [[TMP88]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT60:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT59]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP89:%.*]] = call <1 x double> @llvm.fmuladd.v1f64(<1 x double> [[BLOCK58]], <1 x double> [[SPLAT_SPLAT60]], <1 x double> [[TMP87]]) +; CHECK-NEXT: [[TMP90:%.*]] = shufflevector <1 x double> [[TMP89]], <1 x double> undef, <3 x i32> +; CHECK-NEXT: [[TMP91:%.*]] = shufflevector <3 x double> undef, <3 x double> [[TMP90]], <3 x i32> +; CHECK-NEXT: [[BLOCK61:%.*]] = shufflevector <3 x double> [[TMP23]], <3 x double> undef, <1 x i32> +; CHECK-NEXT: [[TMP92:%.*]] = extractelement <3 x double> [[TMP17]], i64 0 +; CHECK-NEXT: [[SPLAT_SPLATINSERT62:%.*]] = insertelement <1 x double> undef, double [[TMP92]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT63:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT62]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP93:%.*]] = fmul <1 x double> [[BLOCK61]], [[SPLAT_SPLAT63]] +; CHECK-NEXT: [[BLOCK64:%.*]] = shufflevector <3 x double> [[TMP29]], <3 x double> undef, <1 x i32> +; CHECK-NEXT: [[TMP94:%.*]] = extractelement <3 x double> [[TMP17]], i64 1 +; CHECK-NEXT: [[SPLAT_SPLATINSERT65:%.*]] = insertelement <1 x double> undef, double [[TMP94]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT66:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT65]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP95:%.*]] = call <1 x double> @llvm.fmuladd.v1f64(<1 x double> [[BLOCK64]], <1 x double> [[SPLAT_SPLAT66]], <1 x double> [[TMP93]]) +; CHECK-NEXT: [[BLOCK67:%.*]] = shufflevector <3 x double> [[TMP35]], <3 x double> undef, <1 x i32> +; CHECK-NEXT: [[TMP96:%.*]] = extractelement <3 x double> [[TMP17]], i64 2 +; CHECK-NEXT: [[SPLAT_SPLATINSERT68:%.*]] = insertelement <1 x double> undef, double [[TMP96]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT69:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT68]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP97:%.*]] = call <1 x double> @llvm.fmuladd.v1f64(<1 x double> [[BLOCK67]], <1 x double> [[SPLAT_SPLAT69]], <1 x double> [[TMP95]]) +; CHECK-NEXT: [[TMP98:%.*]] = shufflevector <1 x double> [[TMP97]], <1 x double> undef, <3 x i32> +; CHECK-NEXT: [[TMP99:%.*]] = shufflevector <3 x double> [[TMP91]], <3 x double> [[TMP98]], <3 x i32> +; CHECK-NEXT: [[BLOCK70:%.*]] = shufflevector <3 x double> [[TMP23]], <3 x double> undef, <1 x i32> +; CHECK-NEXT: [[TMP100:%.*]] = extractelement <3 x double> [[TMP17]], i64 0 +; CHECK-NEXT: [[SPLAT_SPLATINSERT71:%.*]] = insertelement <1 x double> undef, double [[TMP100]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT72:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT71]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP101:%.*]] = fmul <1 x double> [[BLOCK70]], [[SPLAT_SPLAT72]] +; CHECK-NEXT: [[BLOCK73:%.*]] = shufflevector <3 x double> [[TMP29]], <3 x double> undef, <1 x i32> +; CHECK-NEXT: [[TMP102:%.*]] = extractelement <3 x double> [[TMP17]], i64 1 +; CHECK-NEXT: [[SPLAT_SPLATINSERT74:%.*]] = insertelement <1 x double> undef, double [[TMP102]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT75:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT74]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP103:%.*]] = call <1 x double> @llvm.fmuladd.v1f64(<1 x double> [[BLOCK73]], <1 x double> [[SPLAT_SPLAT75]], <1 x double> [[TMP101]]) +; CHECK-NEXT: [[BLOCK76:%.*]] = shufflevector <3 x double> [[TMP35]], <3 x double> undef, <1 x i32> +; CHECK-NEXT: [[TMP104:%.*]] = extractelement <3 x double> [[TMP17]], i64 2 +; CHECK-NEXT: [[SPLAT_SPLATINSERT77:%.*]] = insertelement <1 x double> undef, double [[TMP104]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT78:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT77]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP105:%.*]] = call <1 x double> @llvm.fmuladd.v1f64(<1 x double> [[BLOCK76]], <1 x double> [[SPLAT_SPLAT78]], <1 x double> [[TMP103]]) +; CHECK-NEXT: [[TMP106:%.*]] = shufflevector <1 x double> [[TMP105]], <1 x double> undef, <3 x i32> +; CHECK-NEXT: [[TMP107:%.*]] = shufflevector <3 x double> [[TMP99]], <3 x double> [[TMP106]], <3 x i32> + +; Store result columns. + +; CHECK-NEXT: [[TMP108:%.*]] = bitcast <9 x double>* [[C_PTR:%.*]] to double* +; CHECK-NEXT: [[TMP109:%.*]] = bitcast double* [[TMP108]] to <3 x double>* +; CHECK-NEXT: store <3 x double> [[TMP59]], <3 x double>* [[TMP109]], align 8 +; CHECK-NEXT: [[TMP110:%.*]] = getelementptr double, double* [[TMP108]], i32 3 +; CHECK-NEXT: [[TMP111:%.*]] = bitcast double* [[TMP110]] to <3 x double>* +; CHECK-NEXT: store <3 x double> [[TMP83]], <3 x double>* [[TMP111]], align 8 +; CHECK-NEXT: [[TMP112:%.*]] = getelementptr double, double* [[TMP108]], i32 6 +; CHECK-NEXT: [[TMP113:%.*]] = bitcast double* [[TMP112]] to <3 x double>* +; CHECK-NEXT: store <3 x double> [[TMP107]], <3 x double>* [[TMP113]], align 8 +; CHECK-NEXT: ret void +; +entry: + %a = load <9 x double>, <9 x double>* %A.Ptr + %b = load <9 x double>, <9 x double>* %B.Ptr + %a.trans = call <9 x double> @llvm.matrix.transpose(<9 x double> %a, i32 3, i32 3) + %c = call <9 x double> @llvm.matrix.multiply.v9f64.v9f64.v9f64(<9 x double> %a.trans, <9 x double> %b, i32 3, i32 3, i32 3) + store <9 x double> %c, <9 x double>* %C.Ptr + ret void +} + +declare <9 x double> @llvm.matrix.transpose(<9 x double>, i32, i32) +declare <9 x double> @llvm.matrix.multiply.v9f64.v9f64.v9f64(<9 x double>, <9 x double>, i32, i32, i32) + +define void @transpose_multiply_add(<9 x double>* %A.Ptr, <9 x double>* %B.Ptr, <9 x double>* %C.Ptr) { +; CHECK-LABEL: @transpose_multiply_add( +; CHECK-NEXT: entry: + +; Load columns of input matrixes %A and %B. + +; CHECK-NEXT: [[TMP0:%.*]] = bitcast <9 x double>* [[A_PTR:%.*]] to double* +; CHECK-NEXT: [[TMP1:%.*]] = bitcast double* [[TMP0]] to <3 x double>* +; CHECK-NEXT: [[TMP2:%.*]] = load <3 x double>, <3 x double>* [[TMP1]], align 8 +; CHECK-NEXT: [[TMP3:%.*]] = getelementptr double, double* [[TMP0]], i32 3 +; CHECK-NEXT: [[TMP4:%.*]] = bitcast double* [[TMP3]] to <3 x double>* +; CHECK-NEXT: [[TMP5:%.*]] = load <3 x double>, <3 x double>* [[TMP4]], align 8 +; CHECK-NEXT: [[TMP6:%.*]] = getelementptr double, double* [[TMP0]], i32 6 +; CHECK-NEXT: [[TMP7:%.*]] = bitcast double* [[TMP6]] to <3 x double>* +; CHECK-NEXT: [[TMP8:%.*]] = load <3 x double>, <3 x double>* [[TMP7]], align 8 +; CHECK-NEXT: [[TMP9:%.*]] = bitcast <9 x double>* [[B_PTR:%.*]] to double* +; CHECK-NEXT: [[TMP10:%.*]] = bitcast double* [[TMP9]] to <3 x double>* +; CHECK-NEXT: [[TMP11:%.*]] = load <3 x double>, <3 x double>* [[TMP10]], align 8 +; CHECK-NEXT: [[TMP12:%.*]] = getelementptr double, double* [[TMP9]], i32 3 +; CHECK-NEXT: [[TMP13:%.*]] = bitcast double* [[TMP12]] to <3 x double>* +; CHECK-NEXT: [[TMP14:%.*]] = load <3 x double>, <3 x double>* [[TMP13]], align 8 +; CHECK-NEXT: [[TMP15:%.*]] = getelementptr double, double* [[TMP9]], i32 6 +; CHECK-NEXT: [[TMP16:%.*]] = bitcast double* [[TMP15]] to <3 x double>* +; CHECK-NEXT: [[TMP17:%.*]] = load <3 x double>, <3 x double>* [[TMP16]], align 8 + +; Transpose %A. + +; CHECK-NEXT: [[TMP18:%.*]] = extractelement <3 x double> [[TMP2]], i64 0 +; CHECK-NEXT: [[TMP19:%.*]] = insertelement <3 x double> undef, double [[TMP18]], i64 0 +; CHECK-NEXT: [[TMP20:%.*]] = extractelement <3 x double> [[TMP5]], i64 0 +; CHECK-NEXT: [[TMP21:%.*]] = insertelement <3 x double> [[TMP19]], double [[TMP20]], i64 1 +; CHECK-NEXT: [[TMP22:%.*]] = extractelement <3 x double> [[TMP8]], i64 0 +; CHECK-NEXT: [[TMP23:%.*]] = insertelement <3 x double> [[TMP21]], double [[TMP22]], i64 2 +; CHECK-NEXT: [[TMP24:%.*]] = extractelement <3 x double> [[TMP2]], i64 1 +; CHECK-NEXT: [[TMP25:%.*]] = insertelement <3 x double> undef, double [[TMP24]], i64 0 +; CHECK-NEXT: [[TMP26:%.*]] = extractelement <3 x double> [[TMP5]], i64 1 +; CHECK-NEXT: [[TMP27:%.*]] = insertelement <3 x double> [[TMP25]], double [[TMP26]], i64 1 +; CHECK-NEXT: [[TMP28:%.*]] = extractelement <3 x double> [[TMP8]], i64 1 +; CHECK-NEXT: [[TMP29:%.*]] = insertelement <3 x double> [[TMP27]], double [[TMP28]], i64 2 +; CHECK-NEXT: [[TMP30:%.*]] = extractelement <3 x double> [[TMP2]], i64 2 +; CHECK-NEXT: [[TMP31:%.*]] = insertelement <3 x double> undef, double [[TMP30]], i64 0 +; CHECK-NEXT: [[TMP32:%.*]] = extractelement <3 x double> [[TMP5]], i64 2 +; CHECK-NEXT: [[TMP33:%.*]] = insertelement <3 x double> [[TMP31]], double [[TMP32]], i64 1 +; CHECK-NEXT: [[TMP34:%.*]] = extractelement <3 x double> [[TMP8]], i64 2 +; CHECK-NEXT: [[TMP35:%.*]] = insertelement <3 x double> [[TMP33]], double [[TMP34]], i64 2 +; CHECK-NEXT: [[BLOCK:%.*]] = shufflevector <3 x double> [[TMP23]], <3 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP36:%.*]] = extractelement <3 x double> [[TMP11]], i64 0 + +; Lower multiply(transpose(%A), %B) + +; CHECK-NEXT: [[SPLAT_SPLATINSERT:%.*]] = insertelement <1 x double> undef, double [[TMP36]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP37:%.*]] = fmul <1 x double> [[BLOCK]], [[SPLAT_SPLAT]] +; CHECK-NEXT: [[BLOCK1:%.*]] = shufflevector <3 x double> [[TMP29]], <3 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP38:%.*]] = extractelement <3 x double> [[TMP11]], i64 1 +; CHECK-NEXT: [[SPLAT_SPLATINSERT2:%.*]] = insertelement <1 x double> undef, double [[TMP38]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT3:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT2]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP39:%.*]] = call <1 x double> @llvm.fmuladd.v1f64(<1 x double> [[BLOCK1]], <1 x double> [[SPLAT_SPLAT3]], <1 x double> [[TMP37]]) +; CHECK-NEXT: [[BLOCK4:%.*]] = shufflevector <3 x double> [[TMP35]], <3 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP40:%.*]] = extractelement <3 x double> [[TMP11]], i64 2 +; CHECK-NEXT: [[SPLAT_SPLATINSERT5:%.*]] = insertelement <1 x double> undef, double [[TMP40]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT6:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT5]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP41:%.*]] = call <1 x double> @llvm.fmuladd.v1f64(<1 x double> [[BLOCK4]], <1 x double> [[SPLAT_SPLAT6]], <1 x double> [[TMP39]]) +; CHECK-NEXT: [[TMP42:%.*]] = shufflevector <1 x double> [[TMP41]], <1 x double> undef, <3 x i32> +; CHECK-NEXT: [[TMP43:%.*]] = shufflevector <3 x double> undef, <3 x double> [[TMP42]], <3 x i32> +; CHECK-NEXT: [[BLOCK7:%.*]] = shufflevector <3 x double> [[TMP23]], <3 x double> undef, <1 x i32> +; CHECK-NEXT: [[TMP44:%.*]] = extractelement <3 x double> [[TMP11]], i64 0 +; CHECK-NEXT: [[SPLAT_SPLATINSERT8:%.*]] = insertelement <1 x double> undef, double [[TMP44]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT9:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT8]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP45:%.*]] = fmul <1 x double> [[BLOCK7]], [[SPLAT_SPLAT9]] +; CHECK-NEXT: [[BLOCK10:%.*]] = shufflevector <3 x double> [[TMP29]], <3 x double> undef, <1 x i32> +; CHECK-NEXT: [[TMP46:%.*]] = extractelement <3 x double> [[TMP11]], i64 1 +; CHECK-NEXT: [[SPLAT_SPLATINSERT11:%.*]] = insertelement <1 x double> undef, double [[TMP46]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT12:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT11]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP47:%.*]] = call <1 x double> @llvm.fmuladd.v1f64(<1 x double> [[BLOCK10]], <1 x double> [[SPLAT_SPLAT12]], <1 x double> [[TMP45]]) +; CHECK-NEXT: [[BLOCK13:%.*]] = shufflevector <3 x double> [[TMP35]], <3 x double> undef, <1 x i32> +; CHECK-NEXT: [[TMP48:%.*]] = extractelement <3 x double> [[TMP11]], i64 2 +; CHECK-NEXT: [[SPLAT_SPLATINSERT14:%.*]] = insertelement <1 x double> undef, double [[TMP48]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT15:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT14]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP49:%.*]] = call <1 x double> @llvm.fmuladd.v1f64(<1 x double> [[BLOCK13]], <1 x double> [[SPLAT_SPLAT15]], <1 x double> [[TMP47]]) +; CHECK-NEXT: [[TMP50:%.*]] = shufflevector <1 x double> [[TMP49]], <1 x double> undef, <3 x i32> +; CHECK-NEXT: [[TMP51:%.*]] = shufflevector <3 x double> [[TMP43]], <3 x double> [[TMP50]], <3 x i32> +; CHECK-NEXT: [[BLOCK16:%.*]] = shufflevector <3 x double> [[TMP23]], <3 x double> undef, <1 x i32> +; CHECK-NEXT: [[TMP52:%.*]] = extractelement <3 x double> [[TMP11]], i64 0 +; CHECK-NEXT: [[SPLAT_SPLATINSERT17:%.*]] = insertelement <1 x double> undef, double [[TMP52]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT18:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT17]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP53:%.*]] = fmul <1 x double> [[BLOCK16]], [[SPLAT_SPLAT18]] +; CHECK-NEXT: [[BLOCK19:%.*]] = shufflevector <3 x double> [[TMP29]], <3 x double> undef, <1 x i32> +; CHECK-NEXT: [[TMP54:%.*]] = extractelement <3 x double> [[TMP11]], i64 1 +; CHECK-NEXT: [[SPLAT_SPLATINSERT20:%.*]] = insertelement <1 x double> undef, double [[TMP54]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT21:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT20]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP55:%.*]] = call <1 x double> @llvm.fmuladd.v1f64(<1 x double> [[BLOCK19]], <1 x double> [[SPLAT_SPLAT21]], <1 x double> [[TMP53]]) +; CHECK-NEXT: [[BLOCK22:%.*]] = shufflevector <3 x double> [[TMP35]], <3 x double> undef, <1 x i32> +; CHECK-NEXT: [[TMP56:%.*]] = extractelement <3 x double> [[TMP11]], i64 2 +; CHECK-NEXT: [[SPLAT_SPLATINSERT23:%.*]] = insertelement <1 x double> undef, double [[TMP56]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT24:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT23]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP57:%.*]] = call <1 x double> @llvm.fmuladd.v1f64(<1 x double> [[BLOCK22]], <1 x double> [[SPLAT_SPLAT24]], <1 x double> [[TMP55]]) +; CHECK-NEXT: [[TMP58:%.*]] = shufflevector <1 x double> [[TMP57]], <1 x double> undef, <3 x i32> +; CHECK-NEXT: [[TMP59:%.*]] = shufflevector <3 x double> [[TMP51]], <3 x double> [[TMP58]], <3 x i32> +; CHECK-NEXT: [[BLOCK25:%.*]] = shufflevector <3 x double> [[TMP23]], <3 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP60:%.*]] = extractelement <3 x double> [[TMP14]], i64 0 +; CHECK-NEXT: [[SPLAT_SPLATINSERT26:%.*]] = insertelement <1 x double> undef, double [[TMP60]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT27:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT26]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP61:%.*]] = fmul <1 x double> [[BLOCK25]], [[SPLAT_SPLAT27]] +; CHECK-NEXT: [[BLOCK28:%.*]] = shufflevector <3 x double> [[TMP29]], <3 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP62:%.*]] = extractelement <3 x double> [[TMP14]], i64 1 +; CHECK-NEXT: [[SPLAT_SPLATINSERT29:%.*]] = insertelement <1 x double> undef, double [[TMP62]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT30:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT29]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP63:%.*]] = call <1 x double> @llvm.fmuladd.v1f64(<1 x double> [[BLOCK28]], <1 x double> [[SPLAT_SPLAT30]], <1 x double> [[TMP61]]) +; CHECK-NEXT: [[BLOCK31:%.*]] = shufflevector <3 x double> [[TMP35]], <3 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP64:%.*]] = extractelement <3 x double> [[TMP14]], i64 2 +; CHECK-NEXT: [[SPLAT_SPLATINSERT32:%.*]] = insertelement <1 x double> undef, double [[TMP64]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT33:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT32]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP65:%.*]] = call <1 x double> @llvm.fmuladd.v1f64(<1 x double> [[BLOCK31]], <1 x double> [[SPLAT_SPLAT33]], <1 x double> [[TMP63]]) +; CHECK-NEXT: [[TMP66:%.*]] = shufflevector <1 x double> [[TMP65]], <1 x double> undef, <3 x i32> +; CHECK-NEXT: [[TMP67:%.*]] = shufflevector <3 x double> undef, <3 x double> [[TMP66]], <3 x i32> +; CHECK-NEXT: [[BLOCK34:%.*]] = shufflevector <3 x double> [[TMP23]], <3 x double> undef, <1 x i32> +; CHECK-NEXT: [[TMP68:%.*]] = extractelement <3 x double> [[TMP14]], i64 0 +; CHECK-NEXT: [[SPLAT_SPLATINSERT35:%.*]] = insertelement <1 x double> undef, double [[TMP68]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT36:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT35]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP69:%.*]] = fmul <1 x double> [[BLOCK34]], [[SPLAT_SPLAT36]] +; CHECK-NEXT: [[BLOCK37:%.*]] = shufflevector <3 x double> [[TMP29]], <3 x double> undef, <1 x i32> +; CHECK-NEXT: [[TMP70:%.*]] = extractelement <3 x double> [[TMP14]], i64 1 +; CHECK-NEXT: [[SPLAT_SPLATINSERT38:%.*]] = insertelement <1 x double> undef, double [[TMP70]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT39:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT38]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP71:%.*]] = call <1 x double> @llvm.fmuladd.v1f64(<1 x double> [[BLOCK37]], <1 x double> [[SPLAT_SPLAT39]], <1 x double> [[TMP69]]) +; CHECK-NEXT: [[BLOCK40:%.*]] = shufflevector <3 x double> [[TMP35]], <3 x double> undef, <1 x i32> +; CHECK-NEXT: [[TMP72:%.*]] = extractelement <3 x double> [[TMP14]], i64 2 +; CHECK-NEXT: [[SPLAT_SPLATINSERT41:%.*]] = insertelement <1 x double> undef, double [[TMP72]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT42:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT41]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP73:%.*]] = call <1 x double> @llvm.fmuladd.v1f64(<1 x double> [[BLOCK40]], <1 x double> [[SPLAT_SPLAT42]], <1 x double> [[TMP71]]) +; CHECK-NEXT: [[TMP74:%.*]] = shufflevector <1 x double> [[TMP73]], <1 x double> undef, <3 x i32> +; CHECK-NEXT: [[TMP75:%.*]] = shufflevector <3 x double> [[TMP67]], <3 x double> [[TMP74]], <3 x i32> +; CHECK-NEXT: [[BLOCK43:%.*]] = shufflevector <3 x double> [[TMP23]], <3 x double> undef, <1 x i32> +; CHECK-NEXT: [[TMP76:%.*]] = extractelement <3 x double> [[TMP14]], i64 0 +; CHECK-NEXT: [[SPLAT_SPLATINSERT44:%.*]] = insertelement <1 x double> undef, double [[TMP76]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT45:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT44]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP77:%.*]] = fmul <1 x double> [[BLOCK43]], [[SPLAT_SPLAT45]] +; CHECK-NEXT: [[BLOCK46:%.*]] = shufflevector <3 x double> [[TMP29]], <3 x double> undef, <1 x i32> +; CHECK-NEXT: [[TMP78:%.*]] = extractelement <3 x double> [[TMP14]], i64 1 +; CHECK-NEXT: [[SPLAT_SPLATINSERT47:%.*]] = insertelement <1 x double> undef, double [[TMP78]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT48:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT47]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP79:%.*]] = call <1 x double> @llvm.fmuladd.v1f64(<1 x double> [[BLOCK46]], <1 x double> [[SPLAT_SPLAT48]], <1 x double> [[TMP77]]) +; CHECK-NEXT: [[BLOCK49:%.*]] = shufflevector <3 x double> [[TMP35]], <3 x double> undef, <1 x i32> +; CHECK-NEXT: [[TMP80:%.*]] = extractelement <3 x double> [[TMP14]], i64 2 +; CHECK-NEXT: [[SPLAT_SPLATINSERT50:%.*]] = insertelement <1 x double> undef, double [[TMP80]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT51:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT50]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP81:%.*]] = call <1 x double> @llvm.fmuladd.v1f64(<1 x double> [[BLOCK49]], <1 x double> [[SPLAT_SPLAT51]], <1 x double> [[TMP79]]) +; CHECK-NEXT: [[TMP82:%.*]] = shufflevector <1 x double> [[TMP81]], <1 x double> undef, <3 x i32> +; CHECK-NEXT: [[TMP83:%.*]] = shufflevector <3 x double> [[TMP75]], <3 x double> [[TMP82]], <3 x i32> +; CHECK-NEXT: [[BLOCK52:%.*]] = shufflevector <3 x double> [[TMP23]], <3 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP84:%.*]] = extractelement <3 x double> [[TMP17]], i64 0 +; CHECK-NEXT: [[SPLAT_SPLATINSERT53:%.*]] = insertelement <1 x double> undef, double [[TMP84]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT54:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT53]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP85:%.*]] = fmul <1 x double> [[BLOCK52]], [[SPLAT_SPLAT54]] +; CHECK-NEXT: [[BLOCK55:%.*]] = shufflevector <3 x double> [[TMP29]], <3 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP86:%.*]] = extractelement <3 x double> [[TMP17]], i64 1 +; CHECK-NEXT: [[SPLAT_SPLATINSERT56:%.*]] = insertelement <1 x double> undef, double [[TMP86]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT57:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT56]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP87:%.*]] = call <1 x double> @llvm.fmuladd.v1f64(<1 x double> [[BLOCK55]], <1 x double> [[SPLAT_SPLAT57]], <1 x double> [[TMP85]]) +; CHECK-NEXT: [[BLOCK58:%.*]] = shufflevector <3 x double> [[TMP35]], <3 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP88:%.*]] = extractelement <3 x double> [[TMP17]], i64 2 +; CHECK-NEXT: [[SPLAT_SPLATINSERT59:%.*]] = insertelement <1 x double> undef, double [[TMP88]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT60:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT59]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP89:%.*]] = call <1 x double> @llvm.fmuladd.v1f64(<1 x double> [[BLOCK58]], <1 x double> [[SPLAT_SPLAT60]], <1 x double> [[TMP87]]) +; CHECK-NEXT: [[TMP90:%.*]] = shufflevector <1 x double> [[TMP89]], <1 x double> undef, <3 x i32> +; CHECK-NEXT: [[TMP91:%.*]] = shufflevector <3 x double> undef, <3 x double> [[TMP90]], <3 x i32> +; CHECK-NEXT: [[BLOCK61:%.*]] = shufflevector <3 x double> [[TMP23]], <3 x double> undef, <1 x i32> +; CHECK-NEXT: [[TMP92:%.*]] = extractelement <3 x double> [[TMP17]], i64 0 +; CHECK-NEXT: [[SPLAT_SPLATINSERT62:%.*]] = insertelement <1 x double> undef, double [[TMP92]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT63:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT62]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP93:%.*]] = fmul <1 x double> [[BLOCK61]], [[SPLAT_SPLAT63]] +; CHECK-NEXT: [[BLOCK64:%.*]] = shufflevector <3 x double> [[TMP29]], <3 x double> undef, <1 x i32> +; CHECK-NEXT: [[TMP94:%.*]] = extractelement <3 x double> [[TMP17]], i64 1 +; CHECK-NEXT: [[SPLAT_SPLATINSERT65:%.*]] = insertelement <1 x double> undef, double [[TMP94]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT66:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT65]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP95:%.*]] = call <1 x double> @llvm.fmuladd.v1f64(<1 x double> [[BLOCK64]], <1 x double> [[SPLAT_SPLAT66]], <1 x double> [[TMP93]]) +; CHECK-NEXT: [[BLOCK67:%.*]] = shufflevector <3 x double> [[TMP35]], <3 x double> undef, <1 x i32> +; CHECK-NEXT: [[TMP96:%.*]] = extractelement <3 x double> [[TMP17]], i64 2 +; CHECK-NEXT: [[SPLAT_SPLATINSERT68:%.*]] = insertelement <1 x double> undef, double [[TMP96]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT69:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT68]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP97:%.*]] = call <1 x double> @llvm.fmuladd.v1f64(<1 x double> [[BLOCK67]], <1 x double> [[SPLAT_SPLAT69]], <1 x double> [[TMP95]]) +; CHECK-NEXT: [[TMP98:%.*]] = shufflevector <1 x double> [[TMP97]], <1 x double> undef, <3 x i32> +; CHECK-NEXT: [[TMP99:%.*]] = shufflevector <3 x double> [[TMP91]], <3 x double> [[TMP98]], <3 x i32> +; CHECK-NEXT: [[BLOCK70:%.*]] = shufflevector <3 x double> [[TMP23]], <3 x double> undef, <1 x i32> +; CHECK-NEXT: [[TMP100:%.*]] = extractelement <3 x double> [[TMP17]], i64 0 +; CHECK-NEXT: [[SPLAT_SPLATINSERT71:%.*]] = insertelement <1 x double> undef, double [[TMP100]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT72:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT71]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP101:%.*]] = fmul <1 x double> [[BLOCK70]], [[SPLAT_SPLAT72]] +; CHECK-NEXT: [[BLOCK73:%.*]] = shufflevector <3 x double> [[TMP29]], <3 x double> undef, <1 x i32> +; CHECK-NEXT: [[TMP102:%.*]] = extractelement <3 x double> [[TMP17]], i64 1 +; CHECK-NEXT: [[SPLAT_SPLATINSERT74:%.*]] = insertelement <1 x double> undef, double [[TMP102]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT75:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT74]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP103:%.*]] = call <1 x double> @llvm.fmuladd.v1f64(<1 x double> [[BLOCK73]], <1 x double> [[SPLAT_SPLAT75]], <1 x double> [[TMP101]]) +; CHECK-NEXT: [[BLOCK76:%.*]] = shufflevector <3 x double> [[TMP35]], <3 x double> undef, <1 x i32> +; CHECK-NEXT: [[TMP104:%.*]] = extractelement <3 x double> [[TMP17]], i64 2 +; CHECK-NEXT: [[SPLAT_SPLATINSERT77:%.*]] = insertelement <1 x double> undef, double [[TMP104]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT78:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT77]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP105:%.*]] = call <1 x double> @llvm.fmuladd.v1f64(<1 x double> [[BLOCK76]], <1 x double> [[SPLAT_SPLAT78]], <1 x double> [[TMP103]]) +; CHECK-NEXT: [[TMP106:%.*]] = shufflevector <1 x double> [[TMP105]], <1 x double> undef, <3 x i32> +; CHECK-NEXT: [[TMP107:%.*]] = shufflevector <3 x double> [[TMP99]], <3 x double> [[TMP106]], <3 x i32> + +; Load columns of matrix %C. + +; CHECK-NEXT: [[TMP108:%.*]] = bitcast <9 x double>* [[C_PTR:%.*]] to double* +; CHECK-NEXT: [[TMP109:%.*]] = bitcast double* [[TMP108]] to <3 x double>* +; CHECK-NEXT: [[TMP110:%.*]] = load <3 x double>, <3 x double>* [[TMP109]], align 8 +; CHECK-NEXT: [[TMP111:%.*]] = getelementptr double, double* [[TMP108]], i32 3 +; CHECK-NEXT: [[TMP112:%.*]] = bitcast double* [[TMP111]] to <3 x double>* +; CHECK-NEXT: [[TMP113:%.*]] = load <3 x double>, <3 x double>* [[TMP112]], align 8 +; CHECK-NEXT: [[TMP114:%.*]] = getelementptr double, double* [[TMP108]], i32 6 +; CHECK-NEXT: [[TMP115:%.*]] = bitcast double* [[TMP114]] to <3 x double>* +; CHECK-NEXT: [[TMP116:%.*]] = load <3 x double>, <3 x double>* [[TMP115]], align 8 + +; Add column vectors. + +; CHECK-NEXT: [[TMP117:%.*]] = fadd <3 x double> [[TMP110]], [[TMP59]] +; CHECK-NEXT: [[TMP118:%.*]] = fadd <3 x double> [[TMP113]], [[TMP83]] +; CHECK-NEXT: [[TMP119:%.*]] = fadd <3 x double> [[TMP116]], [[TMP107]] + +; Store result columns. + +; CHECK-NEXT: [[TMP120:%.*]] = bitcast <9 x double>* [[C_PTR]] to double* +; CHECK-NEXT: [[TMP121:%.*]] = bitcast double* [[TMP120]] to <3 x double>* +; CHECK-NEXT: store <3 x double> [[TMP117]], <3 x double>* [[TMP121]], align 8 +; CHECK-NEXT: [[TMP122:%.*]] = getelementptr double, double* [[TMP120]], i32 3 +; CHECK-NEXT: [[TMP123:%.*]] = bitcast double* [[TMP122]] to <3 x double>* +; CHECK-NEXT: store <3 x double> [[TMP118]], <3 x double>* [[TMP123]], align 8 +; CHECK-NEXT: [[TMP124:%.*]] = getelementptr double, double* [[TMP120]], i32 6 +; CHECK-NEXT: [[TMP125:%.*]] = bitcast double* [[TMP124]] to <3 x double>* +; CHECK-NEXT: store <3 x double> [[TMP119]], <3 x double>* [[TMP125]], align 8 +; CHECK-NEXT: ret void +; +entry: + %a = load <9 x double>, <9 x double>* %A.Ptr + %b = load <9 x double>, <9 x double>* %B.Ptr + %a.trans = call <9 x double> @llvm.matrix.transpose(<9 x double> %a, i32 3, i32 3) + %mult = call <9 x double> @llvm.matrix.multiply.v9f64.v9f64.v9f64(<9 x double> %a.trans, <9 x double> %b, i32 3, i32 3, i32 3) + %c = load <9 x double>, <9 x double>* %C.Ptr + %res = fadd <9 x double> %c, %mult + + store <9 x double> %res, <9 x double>* %C.Ptr + ret void +} diff --git a/llvm/test/Transforms/LowerMatrixIntrinsics/multiply-double-contraction.ll b/llvm/test/Transforms/LowerMatrixIntrinsics/multiply-double-contraction.ll new file mode 100644 --- /dev/null +++ b/llvm/test/Transforms/LowerMatrixIntrinsics/multiply-double-contraction.ll @@ -0,0 +1,241 @@ +; NOTE: Assertions have been autogenerated by utils/update_test_checks.py +; RUN: opt -lower-matrix-intrinsics -matrix-allow-contraction -S < %s | FileCheck %s +; RUN: opt -passes='lower-matrix-intrinsics' -matrix-allow-contraction -S < %s | FileCheck %s + + +define <4 x double> @multiply_2x2(<4 x double> %a, <4 x double> %b) { +; CHECK-LABEL: @multiply_2x2( +; CHECK-NEXT: entry: +; CHECK-NEXT: [[SPLIT:%.*]] = shufflevector <4 x double> [[A:%.*]], <4 x double> undef, <2 x i32> +; CHECK-NEXT: [[SPLIT1:%.*]] = shufflevector <4 x double> [[A]], <4 x double> undef, <2 x i32> +; CHECK-NEXT: [[SPLIT2:%.*]] = shufflevector <4 x double> [[B:%.*]], <4 x double> undef, <2 x i32> +; CHECK-NEXT: [[SPLIT3:%.*]] = shufflevector <4 x double> [[B]], <4 x double> undef, <2 x i32> +; CHECK-NEXT: [[BLOCK:%.*]] = shufflevector <2 x double> [[SPLIT]], <2 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP0:%.*]] = extractelement <2 x double> [[SPLIT2]], i64 0 +; CHECK-NEXT: [[SPLAT_SPLATINSERT:%.*]] = insertelement <1 x double> undef, double [[TMP0]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP1:%.*]] = fmul <1 x double> [[BLOCK]], [[SPLAT_SPLAT]] +; CHECK-NEXT: [[BLOCK4:%.*]] = shufflevector <2 x double> [[SPLIT1]], <2 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP2:%.*]] = extractelement <2 x double> [[SPLIT2]], i64 1 +; CHECK-NEXT: [[SPLAT_SPLATINSERT5:%.*]] = insertelement <1 x double> undef, double [[TMP2]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT6:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT5]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP3:%.*]] = call <1 x double> @llvm.fmuladd.v1f64(<1 x double> [[BLOCK4]], <1 x double> [[SPLAT_SPLAT6]], <1 x double> [[TMP1]]) +; CHECK-NEXT: [[TMP4:%.*]] = shufflevector <1 x double> [[TMP3]], <1 x double> undef, <2 x i32> +; CHECK-NEXT: [[TMP5:%.*]] = shufflevector <2 x double> undef, <2 x double> [[TMP4]], <2 x i32> +; CHECK-NEXT: [[BLOCK7:%.*]] = shufflevector <2 x double> [[SPLIT]], <2 x double> undef, <1 x i32> +; CHECK-NEXT: [[TMP6:%.*]] = extractelement <2 x double> [[SPLIT2]], i64 0 +; CHECK-NEXT: [[SPLAT_SPLATINSERT8:%.*]] = insertelement <1 x double> undef, double [[TMP6]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT9:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT8]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP7:%.*]] = fmul <1 x double> [[BLOCK7]], [[SPLAT_SPLAT9]] +; CHECK-NEXT: [[BLOCK10:%.*]] = shufflevector <2 x double> [[SPLIT1]], <2 x double> undef, <1 x i32> +; CHECK-NEXT: [[TMP8:%.*]] = extractelement <2 x double> [[SPLIT2]], i64 1 +; CHECK-NEXT: [[SPLAT_SPLATINSERT11:%.*]] = insertelement <1 x double> undef, double [[TMP8]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT12:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT11]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP9:%.*]] = call <1 x double> @llvm.fmuladd.v1f64(<1 x double> [[BLOCK10]], <1 x double> [[SPLAT_SPLAT12]], <1 x double> [[TMP7]]) +; CHECK-NEXT: [[TMP10:%.*]] = shufflevector <1 x double> [[TMP9]], <1 x double> undef, <2 x i32> +; CHECK-NEXT: [[TMP11:%.*]] = shufflevector <2 x double> [[TMP5]], <2 x double> [[TMP10]], <2 x i32> +; CHECK-NEXT: [[BLOCK13:%.*]] = shufflevector <2 x double> [[SPLIT]], <2 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP12:%.*]] = extractelement <2 x double> [[SPLIT3]], i64 0 +; CHECK-NEXT: [[SPLAT_SPLATINSERT14:%.*]] = insertelement <1 x double> undef, double [[TMP12]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT15:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT14]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP13:%.*]] = fmul <1 x double> [[BLOCK13]], [[SPLAT_SPLAT15]] +; CHECK-NEXT: [[BLOCK16:%.*]] = shufflevector <2 x double> [[SPLIT1]], <2 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP14:%.*]] = extractelement <2 x double> [[SPLIT3]], i64 1 +; CHECK-NEXT: [[SPLAT_SPLATINSERT17:%.*]] = insertelement <1 x double> undef, double [[TMP14]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT18:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT17]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP15:%.*]] = call <1 x double> @llvm.fmuladd.v1f64(<1 x double> [[BLOCK16]], <1 x double> [[SPLAT_SPLAT18]], <1 x double> [[TMP13]]) +; CHECK-NEXT: [[TMP16:%.*]] = shufflevector <1 x double> [[TMP15]], <1 x double> undef, <2 x i32> +; CHECK-NEXT: [[TMP17:%.*]] = shufflevector <2 x double> undef, <2 x double> [[TMP16]], <2 x i32> +; CHECK-NEXT: [[BLOCK19:%.*]] = shufflevector <2 x double> [[SPLIT]], <2 x double> undef, <1 x i32> +; CHECK-NEXT: [[TMP18:%.*]] = extractelement <2 x double> [[SPLIT3]], i64 0 +; CHECK-NEXT: [[SPLAT_SPLATINSERT20:%.*]] = insertelement <1 x double> undef, double [[TMP18]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT21:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT20]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP19:%.*]] = fmul <1 x double> [[BLOCK19]], [[SPLAT_SPLAT21]] +; CHECK-NEXT: [[BLOCK22:%.*]] = shufflevector <2 x double> [[SPLIT1]], <2 x double> undef, <1 x i32> +; CHECK-NEXT: [[TMP20:%.*]] = extractelement <2 x double> [[SPLIT3]], i64 1 +; CHECK-NEXT: [[SPLAT_SPLATINSERT23:%.*]] = insertelement <1 x double> undef, double [[TMP20]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT24:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT23]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP21:%.*]] = call <1 x double> @llvm.fmuladd.v1f64(<1 x double> [[BLOCK22]], <1 x double> [[SPLAT_SPLAT24]], <1 x double> [[TMP19]]) +; CHECK-NEXT: [[TMP22:%.*]] = shufflevector <1 x double> [[TMP21]], <1 x double> undef, <2 x i32> +; CHECK-NEXT: [[TMP23:%.*]] = shufflevector <2 x double> [[TMP17]], <2 x double> [[TMP22]], <2 x i32> +; CHECK-NEXT: [[TMP24:%.*]] = shufflevector <2 x double> [[TMP11]], <2 x double> [[TMP23]], <4 x i32> +; CHECK-NEXT: ret <4 x double> [[TMP24]] +; +entry: + %c = call <4 x double> @llvm.matrix.multiply.v4f64.v4f64.v4f64(<4 x double> %a, <4 x double> %b, i32 2, i32 2, i32 2) + ret <4 x double> %c +} + +declare <4 x double> @llvm.matrix.multiply.v4f64.v4f64.v4f64(<4 x double>, <4 x double>, i32, i32, i32) + +define <4 x double> @multiply_1x2(<2 x double> %a, <2 x double> %b) { +; CHECK-LABEL: @multiply_1x2( +; CHECK-NEXT: entry: +; CHECK-NEXT: [[SPLIT:%.*]] = shufflevector <2 x double> [[A:%.*]], <2 x double> undef, <2 x i32> +; CHECK-NEXT: [[SPLIT1:%.*]] = shufflevector <2 x double> [[B:%.*]], <2 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[SPLIT2:%.*]] = shufflevector <2 x double> [[B]], <2 x double> undef, <1 x i32> +; CHECK-NEXT: [[BLOCK:%.*]] = shufflevector <2 x double> [[SPLIT]], <2 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP0:%.*]] = extractelement <1 x double> [[SPLIT1]], i64 0 +; CHECK-NEXT: [[SPLAT_SPLATINSERT:%.*]] = insertelement <1 x double> undef, double [[TMP0]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP1:%.*]] = fmul <1 x double> [[BLOCK]], [[SPLAT_SPLAT]] +; CHECK-NEXT: [[TMP2:%.*]] = shufflevector <1 x double> [[TMP1]], <1 x double> undef, <2 x i32> +; CHECK-NEXT: [[TMP3:%.*]] = shufflevector <2 x double> undef, <2 x double> [[TMP2]], <2 x i32> +; CHECK-NEXT: [[BLOCK3:%.*]] = shufflevector <2 x double> [[SPLIT]], <2 x double> undef, <1 x i32> +; CHECK-NEXT: [[TMP4:%.*]] = extractelement <1 x double> [[SPLIT1]], i64 0 +; CHECK-NEXT: [[SPLAT_SPLATINSERT4:%.*]] = insertelement <1 x double> undef, double [[TMP4]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT5:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT4]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP5:%.*]] = fmul <1 x double> [[BLOCK3]], [[SPLAT_SPLAT5]] +; CHECK-NEXT: [[TMP6:%.*]] = shufflevector <1 x double> [[TMP5]], <1 x double> undef, <2 x i32> +; CHECK-NEXT: [[TMP7:%.*]] = shufflevector <2 x double> [[TMP3]], <2 x double> [[TMP6]], <2 x i32> +; CHECK-NEXT: [[BLOCK6:%.*]] = shufflevector <2 x double> [[SPLIT]], <2 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP8:%.*]] = extractelement <1 x double> [[SPLIT2]], i64 0 +; CHECK-NEXT: [[SPLAT_SPLATINSERT7:%.*]] = insertelement <1 x double> undef, double [[TMP8]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT8:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT7]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP9:%.*]] = fmul <1 x double> [[BLOCK6]], [[SPLAT_SPLAT8]] +; CHECK-NEXT: [[TMP10:%.*]] = shufflevector <1 x double> [[TMP9]], <1 x double> undef, <2 x i32> +; CHECK-NEXT: [[TMP11:%.*]] = shufflevector <2 x double> undef, <2 x double> [[TMP10]], <2 x i32> +; CHECK-NEXT: [[BLOCK9:%.*]] = shufflevector <2 x double> [[SPLIT]], <2 x double> undef, <1 x i32> +; CHECK-NEXT: [[TMP12:%.*]] = extractelement <1 x double> [[SPLIT2]], i64 0 +; CHECK-NEXT: [[SPLAT_SPLATINSERT10:%.*]] = insertelement <1 x double> undef, double [[TMP12]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT11:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT10]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP13:%.*]] = fmul <1 x double> [[BLOCK9]], [[SPLAT_SPLAT11]] +; CHECK-NEXT: [[TMP14:%.*]] = shufflevector <1 x double> [[TMP13]], <1 x double> undef, <2 x i32> +; CHECK-NEXT: [[TMP15:%.*]] = shufflevector <2 x double> [[TMP11]], <2 x double> [[TMP14]], <2 x i32> +; CHECK-NEXT: [[TMP16:%.*]] = shufflevector <2 x double> [[TMP7]], <2 x double> [[TMP15]], <4 x i32> +; CHECK-NEXT: ret <4 x double> [[TMP16]] +; +entry: + %c = call <4 x double> @llvm.matrix.multiply.v4f64.v2f64.v2f64(<2 x double> %a, <2 x double> %b, i32 2, i32 1, i32 2) + ret <4 x double> %c +} + +declare <4 x double> @llvm.matrix.multiply.v4f64.v2f64.v2f64(<2 x double>, <2 x double>, i32, i32, i32) + +define <9 x double> @multiply_2x3(<6 x double> %a, <6 x double> %b) { +; CHECK-LABEL: @multiply_2x3( +; CHECK-NEXT: entry: +; CHECK-NEXT: [[SPLIT:%.*]] = shufflevector <6 x double> [[A:%.*]], <6 x double> undef, <3 x i32> +; CHECK-NEXT: [[SPLIT1:%.*]] = shufflevector <6 x double> [[A]], <6 x double> undef, <3 x i32> +; CHECK-NEXT: [[SPLIT2:%.*]] = shufflevector <6 x double> [[B:%.*]], <6 x double> undef, <2 x i32> +; CHECK-NEXT: [[SPLIT3:%.*]] = shufflevector <6 x double> [[B]], <6 x double> undef, <2 x i32> +; CHECK-NEXT: [[SPLIT4:%.*]] = shufflevector <6 x double> [[B]], <6 x double> undef, <2 x i32> +; CHECK-NEXT: [[BLOCK:%.*]] = shufflevector <3 x double> [[SPLIT]], <3 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP0:%.*]] = extractelement <2 x double> [[SPLIT2]], i64 0 +; CHECK-NEXT: [[SPLAT_SPLATINSERT:%.*]] = insertelement <1 x double> undef, double [[TMP0]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP1:%.*]] = fmul <1 x double> [[BLOCK]], [[SPLAT_SPLAT]] +; CHECK-NEXT: [[BLOCK5:%.*]] = shufflevector <3 x double> [[SPLIT1]], <3 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP2:%.*]] = extractelement <2 x double> [[SPLIT2]], i64 1 +; CHECK-NEXT: [[SPLAT_SPLATINSERT6:%.*]] = insertelement <1 x double> undef, double [[TMP2]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT7:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT6]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP3:%.*]] = call <1 x double> @llvm.fmuladd.v1f64(<1 x double> [[BLOCK5]], <1 x double> [[SPLAT_SPLAT7]], <1 x double> [[TMP1]]) +; CHECK-NEXT: [[TMP4:%.*]] = shufflevector <1 x double> [[TMP3]], <1 x double> undef, <3 x i32> +; CHECK-NEXT: [[TMP5:%.*]] = shufflevector <3 x double> undef, <3 x double> [[TMP4]], <3 x i32> +; CHECK-NEXT: [[BLOCK8:%.*]] = shufflevector <3 x double> [[SPLIT]], <3 x double> undef, <1 x i32> +; CHECK-NEXT: [[TMP6:%.*]] = extractelement <2 x double> [[SPLIT2]], i64 0 +; CHECK-NEXT: [[SPLAT_SPLATINSERT9:%.*]] = insertelement <1 x double> undef, double [[TMP6]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT10:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT9]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP7:%.*]] = fmul <1 x double> [[BLOCK8]], [[SPLAT_SPLAT10]] +; CHECK-NEXT: [[BLOCK11:%.*]] = shufflevector <3 x double> [[SPLIT1]], <3 x double> undef, <1 x i32> +; CHECK-NEXT: [[TMP8:%.*]] = extractelement <2 x double> [[SPLIT2]], i64 1 +; CHECK-NEXT: [[SPLAT_SPLATINSERT12:%.*]] = insertelement <1 x double> undef, double [[TMP8]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT13:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT12]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP9:%.*]] = call <1 x double> @llvm.fmuladd.v1f64(<1 x double> [[BLOCK11]], <1 x double> [[SPLAT_SPLAT13]], <1 x double> [[TMP7]]) +; CHECK-NEXT: [[TMP10:%.*]] = shufflevector <1 x double> [[TMP9]], <1 x double> undef, <3 x i32> +; CHECK-NEXT: [[TMP11:%.*]] = shufflevector <3 x double> [[TMP5]], <3 x double> [[TMP10]], <3 x i32> +; CHECK-NEXT: [[BLOCK14:%.*]] = shufflevector <3 x double> [[SPLIT]], <3 x double> undef, <1 x i32> +; CHECK-NEXT: [[TMP12:%.*]] = extractelement <2 x double> [[SPLIT2]], i64 0 +; CHECK-NEXT: [[SPLAT_SPLATINSERT15:%.*]] = insertelement <1 x double> undef, double [[TMP12]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT16:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT15]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP13:%.*]] = fmul <1 x double> [[BLOCK14]], [[SPLAT_SPLAT16]] +; CHECK-NEXT: [[BLOCK17:%.*]] = shufflevector <3 x double> [[SPLIT1]], <3 x double> undef, <1 x i32> +; CHECK-NEXT: [[TMP14:%.*]] = extractelement <2 x double> [[SPLIT2]], i64 1 +; CHECK-NEXT: [[SPLAT_SPLATINSERT18:%.*]] = insertelement <1 x double> undef, double [[TMP14]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT19:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT18]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP15:%.*]] = call <1 x double> @llvm.fmuladd.v1f64(<1 x double> [[BLOCK17]], <1 x double> [[SPLAT_SPLAT19]], <1 x double> [[TMP13]]) +; CHECK-NEXT: [[TMP16:%.*]] = shufflevector <1 x double> [[TMP15]], <1 x double> undef, <3 x i32> +; CHECK-NEXT: [[TMP17:%.*]] = shufflevector <3 x double> [[TMP11]], <3 x double> [[TMP16]], <3 x i32> +; CHECK-NEXT: [[BLOCK20:%.*]] = shufflevector <3 x double> [[SPLIT]], <3 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP18:%.*]] = extractelement <2 x double> [[SPLIT3]], i64 0 +; CHECK-NEXT: [[SPLAT_SPLATINSERT21:%.*]] = insertelement <1 x double> undef, double [[TMP18]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT22:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT21]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP19:%.*]] = fmul <1 x double> [[BLOCK20]], [[SPLAT_SPLAT22]] +; CHECK-NEXT: [[BLOCK23:%.*]] = shufflevector <3 x double> [[SPLIT1]], <3 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP20:%.*]] = extractelement <2 x double> [[SPLIT3]], i64 1 +; CHECK-NEXT: [[SPLAT_SPLATINSERT24:%.*]] = insertelement <1 x double> undef, double [[TMP20]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT25:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT24]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP21:%.*]] = call <1 x double> @llvm.fmuladd.v1f64(<1 x double> [[BLOCK23]], <1 x double> [[SPLAT_SPLAT25]], <1 x double> [[TMP19]]) +; CHECK-NEXT: [[TMP22:%.*]] = shufflevector <1 x double> [[TMP21]], <1 x double> undef, <3 x i32> +; CHECK-NEXT: [[TMP23:%.*]] = shufflevector <3 x double> undef, <3 x double> [[TMP22]], <3 x i32> +; CHECK-NEXT: [[BLOCK26:%.*]] = shufflevector <3 x double> [[SPLIT]], <3 x double> undef, <1 x i32> +; CHECK-NEXT: [[TMP24:%.*]] = extractelement <2 x double> [[SPLIT3]], i64 0 +; CHECK-NEXT: [[SPLAT_SPLATINSERT27:%.*]] = insertelement <1 x double> undef, double [[TMP24]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT28:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT27]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP25:%.*]] = fmul <1 x double> [[BLOCK26]], [[SPLAT_SPLAT28]] +; CHECK-NEXT: [[BLOCK29:%.*]] = shufflevector <3 x double> [[SPLIT1]], <3 x double> undef, <1 x i32> +; CHECK-NEXT: [[TMP26:%.*]] = extractelement <2 x double> [[SPLIT3]], i64 1 +; CHECK-NEXT: [[SPLAT_SPLATINSERT30:%.*]] = insertelement <1 x double> undef, double [[TMP26]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT31:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT30]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP27:%.*]] = call <1 x double> @llvm.fmuladd.v1f64(<1 x double> [[BLOCK29]], <1 x double> [[SPLAT_SPLAT31]], <1 x double> [[TMP25]]) +; CHECK-NEXT: [[TMP28:%.*]] = shufflevector <1 x double> [[TMP27]], <1 x double> undef, <3 x i32> +; CHECK-NEXT: [[TMP29:%.*]] = shufflevector <3 x double> [[TMP23]], <3 x double> [[TMP28]], <3 x i32> +; CHECK-NEXT: [[BLOCK32:%.*]] = shufflevector <3 x double> [[SPLIT]], <3 x double> undef, <1 x i32> +; CHECK-NEXT: [[TMP30:%.*]] = extractelement <2 x double> [[SPLIT3]], i64 0 +; CHECK-NEXT: [[SPLAT_SPLATINSERT33:%.*]] = insertelement <1 x double> undef, double [[TMP30]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT34:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT33]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP31:%.*]] = fmul <1 x double> [[BLOCK32]], [[SPLAT_SPLAT34]] +; CHECK-NEXT: [[BLOCK35:%.*]] = shufflevector <3 x double> [[SPLIT1]], <3 x double> undef, <1 x i32> +; CHECK-NEXT: [[TMP32:%.*]] = extractelement <2 x double> [[SPLIT3]], i64 1 +; CHECK-NEXT: [[SPLAT_SPLATINSERT36:%.*]] = insertelement <1 x double> undef, double [[TMP32]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT37:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT36]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP33:%.*]] = call <1 x double> @llvm.fmuladd.v1f64(<1 x double> [[BLOCK35]], <1 x double> [[SPLAT_SPLAT37]], <1 x double> [[TMP31]]) +; CHECK-NEXT: [[TMP34:%.*]] = shufflevector <1 x double> [[TMP33]], <1 x double> undef, <3 x i32> +; CHECK-NEXT: [[TMP35:%.*]] = shufflevector <3 x double> [[TMP29]], <3 x double> [[TMP34]], <3 x i32> +; CHECK-NEXT: [[BLOCK38:%.*]] = shufflevector <3 x double> [[SPLIT]], <3 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP36:%.*]] = extractelement <2 x double> [[SPLIT4]], i64 0 +; CHECK-NEXT: [[SPLAT_SPLATINSERT39:%.*]] = insertelement <1 x double> undef, double [[TMP36]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT40:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT39]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP37:%.*]] = fmul <1 x double> [[BLOCK38]], [[SPLAT_SPLAT40]] +; CHECK-NEXT: [[BLOCK41:%.*]] = shufflevector <3 x double> [[SPLIT1]], <3 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP38:%.*]] = extractelement <2 x double> [[SPLIT4]], i64 1 +; CHECK-NEXT: [[SPLAT_SPLATINSERT42:%.*]] = insertelement <1 x double> undef, double [[TMP38]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT43:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT42]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP39:%.*]] = call <1 x double> @llvm.fmuladd.v1f64(<1 x double> [[BLOCK41]], <1 x double> [[SPLAT_SPLAT43]], <1 x double> [[TMP37]]) +; CHECK-NEXT: [[TMP40:%.*]] = shufflevector <1 x double> [[TMP39]], <1 x double> undef, <3 x i32> +; CHECK-NEXT: [[TMP41:%.*]] = shufflevector <3 x double> undef, <3 x double> [[TMP40]], <3 x i32> +; CHECK-NEXT: [[BLOCK44:%.*]] = shufflevector <3 x double> [[SPLIT]], <3 x double> undef, <1 x i32> +; CHECK-NEXT: [[TMP42:%.*]] = extractelement <2 x double> [[SPLIT4]], i64 0 +; CHECK-NEXT: [[SPLAT_SPLATINSERT45:%.*]] = insertelement <1 x double> undef, double [[TMP42]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT46:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT45]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP43:%.*]] = fmul <1 x double> [[BLOCK44]], [[SPLAT_SPLAT46]] +; CHECK-NEXT: [[BLOCK47:%.*]] = shufflevector <3 x double> [[SPLIT1]], <3 x double> undef, <1 x i32> +; CHECK-NEXT: [[TMP44:%.*]] = extractelement <2 x double> [[SPLIT4]], i64 1 +; CHECK-NEXT: [[SPLAT_SPLATINSERT48:%.*]] = insertelement <1 x double> undef, double [[TMP44]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT49:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT48]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP45:%.*]] = call <1 x double> @llvm.fmuladd.v1f64(<1 x double> [[BLOCK47]], <1 x double> [[SPLAT_SPLAT49]], <1 x double> [[TMP43]]) +; CHECK-NEXT: [[TMP46:%.*]] = shufflevector <1 x double> [[TMP45]], <1 x double> undef, <3 x i32> +; CHECK-NEXT: [[TMP47:%.*]] = shufflevector <3 x double> [[TMP41]], <3 x double> [[TMP46]], <3 x i32> +; CHECK-NEXT: [[BLOCK50:%.*]] = shufflevector <3 x double> [[SPLIT]], <3 x double> undef, <1 x i32> +; CHECK-NEXT: [[TMP48:%.*]] = extractelement <2 x double> [[SPLIT4]], i64 0 +; CHECK-NEXT: [[SPLAT_SPLATINSERT51:%.*]] = insertelement <1 x double> undef, double [[TMP48]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT52:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT51]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP49:%.*]] = fmul <1 x double> [[BLOCK50]], [[SPLAT_SPLAT52]] +; CHECK-NEXT: [[BLOCK53:%.*]] = shufflevector <3 x double> [[SPLIT1]], <3 x double> undef, <1 x i32> +; CHECK-NEXT: [[TMP50:%.*]] = extractelement <2 x double> [[SPLIT4]], i64 1 +; CHECK-NEXT: [[SPLAT_SPLATINSERT54:%.*]] = insertelement <1 x double> undef, double [[TMP50]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT55:%.*]] = shufflevector <1 x double> [[SPLAT_SPLATINSERT54]], <1 x double> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP51:%.*]] = call <1 x double> @llvm.fmuladd.v1f64(<1 x double> [[BLOCK53]], <1 x double> [[SPLAT_SPLAT55]], <1 x double> [[TMP49]]) +; CHECK-NEXT: [[TMP52:%.*]] = shufflevector <1 x double> [[TMP51]], <1 x double> undef, <3 x i32> +; CHECK-NEXT: [[TMP53:%.*]] = shufflevector <3 x double> [[TMP47]], <3 x double> [[TMP52]], <3 x i32> +; CHECK-NEXT: [[TMP54:%.*]] = shufflevector <3 x double> [[TMP17]], <3 x double> [[TMP35]], <6 x i32> +; CHECK-NEXT: [[TMP55:%.*]] = shufflevector <3 x double> [[TMP53]], <3 x double> undef, <6 x i32> +; CHECK-NEXT: [[TMP56:%.*]] = shufflevector <6 x double> [[TMP54]], <6 x double> [[TMP55]], <9 x i32> +; CHECK-NEXT: ret <9 x double> [[TMP56]] +; +entry: + %c = call <9 x double> @llvm.matrix.multiply.v6f64.v6f64.v6f64(<6 x double> %a, <6 x double> %b, i32 3, i32 2, i32 3) + ret <9 x double> %c +} + +declare <9 x double> @llvm.matrix.multiply.v6f64.v6f64.v6f64(<6 x double>, <6 x double>, i32, i32, i32) diff --git a/llvm/test/Transforms/LowerMatrixIntrinsics/multiply-float-contraction.ll b/llvm/test/Transforms/LowerMatrixIntrinsics/multiply-float-contraction.ll new file mode 100644 --- /dev/null +++ b/llvm/test/Transforms/LowerMatrixIntrinsics/multiply-float-contraction.ll @@ -0,0 +1,241 @@ +; NOTE: Assertions have been autogenerated by utils/update_test_checks.py +; RUN: opt -lower-matrix-intrinsics -matrix-allow-contraction -S < %s | FileCheck %s +; RUN: opt -passes='lower-matrix-intrinsics' -matrix-allow-contraction -S < %s | FileCheck %s + + +define <4 x float> @multiply_2x2(<4 x float> %a, <4 x float> %b) { +; CHECK-LABEL: @multiply_2x2( +; CHECK-NEXT: entry: +; CHECK-NEXT: [[SPLIT:%.*]] = shufflevector <4 x float> [[A:%.*]], <4 x float> undef, <2 x i32> +; CHECK-NEXT: [[SPLIT1:%.*]] = shufflevector <4 x float> [[A]], <4 x float> undef, <2 x i32> +; CHECK-NEXT: [[SPLIT2:%.*]] = shufflevector <4 x float> [[B:%.*]], <4 x float> undef, <2 x i32> +; CHECK-NEXT: [[SPLIT3:%.*]] = shufflevector <4 x float> [[B]], <4 x float> undef, <2 x i32> +; CHECK-NEXT: [[BLOCK:%.*]] = shufflevector <2 x float> [[SPLIT]], <2 x float> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP0:%.*]] = extractelement <2 x float> [[SPLIT2]], i64 0 +; CHECK-NEXT: [[SPLAT_SPLATINSERT:%.*]] = insertelement <1 x float> undef, float [[TMP0]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT:%.*]] = shufflevector <1 x float> [[SPLAT_SPLATINSERT]], <1 x float> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP1:%.*]] = fmul <1 x float> [[BLOCK]], [[SPLAT_SPLAT]] +; CHECK-NEXT: [[BLOCK4:%.*]] = shufflevector <2 x float> [[SPLIT1]], <2 x float> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP2:%.*]] = extractelement <2 x float> [[SPLIT2]], i64 1 +; CHECK-NEXT: [[SPLAT_SPLATINSERT5:%.*]] = insertelement <1 x float> undef, float [[TMP2]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT6:%.*]] = shufflevector <1 x float> [[SPLAT_SPLATINSERT5]], <1 x float> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP3:%.*]] = call <1 x float> @llvm.fmuladd.v1f32(<1 x float> [[BLOCK4]], <1 x float> [[SPLAT_SPLAT6]], <1 x float> [[TMP1]]) +; CHECK-NEXT: [[TMP4:%.*]] = shufflevector <1 x float> [[TMP3]], <1 x float> undef, <2 x i32> +; CHECK-NEXT: [[TMP5:%.*]] = shufflevector <2 x float> undef, <2 x float> [[TMP4]], <2 x i32> +; CHECK-NEXT: [[BLOCK7:%.*]] = shufflevector <2 x float> [[SPLIT]], <2 x float> undef, <1 x i32> +; CHECK-NEXT: [[TMP6:%.*]] = extractelement <2 x float> [[SPLIT2]], i64 0 +; CHECK-NEXT: [[SPLAT_SPLATINSERT8:%.*]] = insertelement <1 x float> undef, float [[TMP6]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT9:%.*]] = shufflevector <1 x float> [[SPLAT_SPLATINSERT8]], <1 x float> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP7:%.*]] = fmul <1 x float> [[BLOCK7]], [[SPLAT_SPLAT9]] +; CHECK-NEXT: [[BLOCK10:%.*]] = shufflevector <2 x float> [[SPLIT1]], <2 x float> undef, <1 x i32> +; CHECK-NEXT: [[TMP8:%.*]] = extractelement <2 x float> [[SPLIT2]], i64 1 +; CHECK-NEXT: [[SPLAT_SPLATINSERT11:%.*]] = insertelement <1 x float> undef, float [[TMP8]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT12:%.*]] = shufflevector <1 x float> [[SPLAT_SPLATINSERT11]], <1 x float> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP9:%.*]] = call <1 x float> @llvm.fmuladd.v1f32(<1 x float> [[BLOCK10]], <1 x float> [[SPLAT_SPLAT12]], <1 x float> [[TMP7]]) +; CHECK-NEXT: [[TMP10:%.*]] = shufflevector <1 x float> [[TMP9]], <1 x float> undef, <2 x i32> +; CHECK-NEXT: [[TMP11:%.*]] = shufflevector <2 x float> [[TMP5]], <2 x float> [[TMP10]], <2 x i32> +; CHECK-NEXT: [[BLOCK13:%.*]] = shufflevector <2 x float> [[SPLIT]], <2 x float> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP12:%.*]] = extractelement <2 x float> [[SPLIT3]], i64 0 +; CHECK-NEXT: [[SPLAT_SPLATINSERT14:%.*]] = insertelement <1 x float> undef, float [[TMP12]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT15:%.*]] = shufflevector <1 x float> [[SPLAT_SPLATINSERT14]], <1 x float> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP13:%.*]] = fmul <1 x float> [[BLOCK13]], [[SPLAT_SPLAT15]] +; CHECK-NEXT: [[BLOCK16:%.*]] = shufflevector <2 x float> [[SPLIT1]], <2 x float> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP14:%.*]] = extractelement <2 x float> [[SPLIT3]], i64 1 +; CHECK-NEXT: [[SPLAT_SPLATINSERT17:%.*]] = insertelement <1 x float> undef, float [[TMP14]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT18:%.*]] = shufflevector <1 x float> [[SPLAT_SPLATINSERT17]], <1 x float> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP15:%.*]] = call <1 x float> @llvm.fmuladd.v1f32(<1 x float> [[BLOCK16]], <1 x float> [[SPLAT_SPLAT18]], <1 x float> [[TMP13]]) +; CHECK-NEXT: [[TMP16:%.*]] = shufflevector <1 x float> [[TMP15]], <1 x float> undef, <2 x i32> +; CHECK-NEXT: [[TMP17:%.*]] = shufflevector <2 x float> undef, <2 x float> [[TMP16]], <2 x i32> +; CHECK-NEXT: [[BLOCK19:%.*]] = shufflevector <2 x float> [[SPLIT]], <2 x float> undef, <1 x i32> +; CHECK-NEXT: [[TMP18:%.*]] = extractelement <2 x float> [[SPLIT3]], i64 0 +; CHECK-NEXT: [[SPLAT_SPLATINSERT20:%.*]] = insertelement <1 x float> undef, float [[TMP18]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT21:%.*]] = shufflevector <1 x float> [[SPLAT_SPLATINSERT20]], <1 x float> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP19:%.*]] = fmul <1 x float> [[BLOCK19]], [[SPLAT_SPLAT21]] +; CHECK-NEXT: [[BLOCK22:%.*]] = shufflevector <2 x float> [[SPLIT1]], <2 x float> undef, <1 x i32> +; CHECK-NEXT: [[TMP20:%.*]] = extractelement <2 x float> [[SPLIT3]], i64 1 +; CHECK-NEXT: [[SPLAT_SPLATINSERT23:%.*]] = insertelement <1 x float> undef, float [[TMP20]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT24:%.*]] = shufflevector <1 x float> [[SPLAT_SPLATINSERT23]], <1 x float> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP21:%.*]] = call <1 x float> @llvm.fmuladd.v1f32(<1 x float> [[BLOCK22]], <1 x float> [[SPLAT_SPLAT24]], <1 x float> [[TMP19]]) +; CHECK-NEXT: [[TMP22:%.*]] = shufflevector <1 x float> [[TMP21]], <1 x float> undef, <2 x i32> +; CHECK-NEXT: [[TMP23:%.*]] = shufflevector <2 x float> [[TMP17]], <2 x float> [[TMP22]], <2 x i32> +; CHECK-NEXT: [[TMP24:%.*]] = shufflevector <2 x float> [[TMP11]], <2 x float> [[TMP23]], <4 x i32> +; CHECK-NEXT: ret <4 x float> [[TMP24]] +; +entry: + %c = call <4 x float> @llvm.matrix.multiply.v4f32.v4f32.v4f32(<4 x float> %a, <4 x float> %b, i32 2, i32 2, i32 2) + ret <4 x float> %c +} + +declare <4 x float> @llvm.matrix.multiply.v4f32.v4f32.v4f32(<4 x float>, <4 x float>, i32, i32, i32) + +define <4 x float> @multiply_1x2(<2 x float> %a, <2 x float> %b) { +; CHECK-LABEL: @multiply_1x2( +; CHECK-NEXT: entry: +; CHECK-NEXT: [[SPLIT:%.*]] = shufflevector <2 x float> [[A:%.*]], <2 x float> undef, <2 x i32> +; CHECK-NEXT: [[SPLIT1:%.*]] = shufflevector <2 x float> [[B:%.*]], <2 x float> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[SPLIT2:%.*]] = shufflevector <2 x float> [[B]], <2 x float> undef, <1 x i32> +; CHECK-NEXT: [[BLOCK:%.*]] = shufflevector <2 x float> [[SPLIT]], <2 x float> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP0:%.*]] = extractelement <1 x float> [[SPLIT1]], i64 0 +; CHECK-NEXT: [[SPLAT_SPLATINSERT:%.*]] = insertelement <1 x float> undef, float [[TMP0]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT:%.*]] = shufflevector <1 x float> [[SPLAT_SPLATINSERT]], <1 x float> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP1:%.*]] = fmul <1 x float> [[BLOCK]], [[SPLAT_SPLAT]] +; CHECK-NEXT: [[TMP2:%.*]] = shufflevector <1 x float> [[TMP1]], <1 x float> undef, <2 x i32> +; CHECK-NEXT: [[TMP3:%.*]] = shufflevector <2 x float> undef, <2 x float> [[TMP2]], <2 x i32> +; CHECK-NEXT: [[BLOCK3:%.*]] = shufflevector <2 x float> [[SPLIT]], <2 x float> undef, <1 x i32> +; CHECK-NEXT: [[TMP4:%.*]] = extractelement <1 x float> [[SPLIT1]], i64 0 +; CHECK-NEXT: [[SPLAT_SPLATINSERT4:%.*]] = insertelement <1 x float> undef, float [[TMP4]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT5:%.*]] = shufflevector <1 x float> [[SPLAT_SPLATINSERT4]], <1 x float> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP5:%.*]] = fmul <1 x float> [[BLOCK3]], [[SPLAT_SPLAT5]] +; CHECK-NEXT: [[TMP6:%.*]] = shufflevector <1 x float> [[TMP5]], <1 x float> undef, <2 x i32> +; CHECK-NEXT: [[TMP7:%.*]] = shufflevector <2 x float> [[TMP3]], <2 x float> [[TMP6]], <2 x i32> +; CHECK-NEXT: [[BLOCK6:%.*]] = shufflevector <2 x float> [[SPLIT]], <2 x float> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP8:%.*]] = extractelement <1 x float> [[SPLIT2]], i64 0 +; CHECK-NEXT: [[SPLAT_SPLATINSERT7:%.*]] = insertelement <1 x float> undef, float [[TMP8]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT8:%.*]] = shufflevector <1 x float> [[SPLAT_SPLATINSERT7]], <1 x float> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP9:%.*]] = fmul <1 x float> [[BLOCK6]], [[SPLAT_SPLAT8]] +; CHECK-NEXT: [[TMP10:%.*]] = shufflevector <1 x float> [[TMP9]], <1 x float> undef, <2 x i32> +; CHECK-NEXT: [[TMP11:%.*]] = shufflevector <2 x float> undef, <2 x float> [[TMP10]], <2 x i32> +; CHECK-NEXT: [[BLOCK9:%.*]] = shufflevector <2 x float> [[SPLIT]], <2 x float> undef, <1 x i32> +; CHECK-NEXT: [[TMP12:%.*]] = extractelement <1 x float> [[SPLIT2]], i64 0 +; CHECK-NEXT: [[SPLAT_SPLATINSERT10:%.*]] = insertelement <1 x float> undef, float [[TMP12]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT11:%.*]] = shufflevector <1 x float> [[SPLAT_SPLATINSERT10]], <1 x float> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP13:%.*]] = fmul <1 x float> [[BLOCK9]], [[SPLAT_SPLAT11]] +; CHECK-NEXT: [[TMP14:%.*]] = shufflevector <1 x float> [[TMP13]], <1 x float> undef, <2 x i32> +; CHECK-NEXT: [[TMP15:%.*]] = shufflevector <2 x float> [[TMP11]], <2 x float> [[TMP14]], <2 x i32> +; CHECK-NEXT: [[TMP16:%.*]] = shufflevector <2 x float> [[TMP7]], <2 x float> [[TMP15]], <4 x i32> +; CHECK-NEXT: ret <4 x float> [[TMP16]] +; +entry: + %c = call <4 x float> @llvm.matrix.multiply.v4f32.v2f32.v2f32(<2 x float> %a, <2 x float> %b, i32 2, i32 1, i32 2) + ret <4 x float> %c +} + +declare <4 x float> @llvm.matrix.multiply.v4f32.v2f32.v2f32(<2 x float>, <2 x float>, i32, i32, i32) + +define <9 x float> @multiply_2x3(<6 x float> %a, <6 x float> %b) { +; CHECK-LABEL: @multiply_2x3( +; CHECK-NEXT: entry: +; CHECK-NEXT: [[SPLIT:%.*]] = shufflevector <6 x float> [[A:%.*]], <6 x float> undef, <3 x i32> +; CHECK-NEXT: [[SPLIT1:%.*]] = shufflevector <6 x float> [[A]], <6 x float> undef, <3 x i32> +; CHECK-NEXT: [[SPLIT2:%.*]] = shufflevector <6 x float> [[B:%.*]], <6 x float> undef, <2 x i32> +; CHECK-NEXT: [[SPLIT3:%.*]] = shufflevector <6 x float> [[B]], <6 x float> undef, <2 x i32> +; CHECK-NEXT: [[SPLIT4:%.*]] = shufflevector <6 x float> [[B]], <6 x float> undef, <2 x i32> +; CHECK-NEXT: [[BLOCK:%.*]] = shufflevector <3 x float> [[SPLIT]], <3 x float> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP0:%.*]] = extractelement <2 x float> [[SPLIT2]], i64 0 +; CHECK-NEXT: [[SPLAT_SPLATINSERT:%.*]] = insertelement <1 x float> undef, float [[TMP0]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT:%.*]] = shufflevector <1 x float> [[SPLAT_SPLATINSERT]], <1 x float> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP1:%.*]] = fmul <1 x float> [[BLOCK]], [[SPLAT_SPLAT]] +; CHECK-NEXT: [[BLOCK5:%.*]] = shufflevector <3 x float> [[SPLIT1]], <3 x float> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP2:%.*]] = extractelement <2 x float> [[SPLIT2]], i64 1 +; CHECK-NEXT: [[SPLAT_SPLATINSERT6:%.*]] = insertelement <1 x float> undef, float [[TMP2]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT7:%.*]] = shufflevector <1 x float> [[SPLAT_SPLATINSERT6]], <1 x float> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP3:%.*]] = call <1 x float> @llvm.fmuladd.v1f32(<1 x float> [[BLOCK5]], <1 x float> [[SPLAT_SPLAT7]], <1 x float> [[TMP1]]) +; CHECK-NEXT: [[TMP4:%.*]] = shufflevector <1 x float> [[TMP3]], <1 x float> undef, <3 x i32> +; CHECK-NEXT: [[TMP5:%.*]] = shufflevector <3 x float> undef, <3 x float> [[TMP4]], <3 x i32> +; CHECK-NEXT: [[BLOCK8:%.*]] = shufflevector <3 x float> [[SPLIT]], <3 x float> undef, <1 x i32> +; CHECK-NEXT: [[TMP6:%.*]] = extractelement <2 x float> [[SPLIT2]], i64 0 +; CHECK-NEXT: [[SPLAT_SPLATINSERT9:%.*]] = insertelement <1 x float> undef, float [[TMP6]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT10:%.*]] = shufflevector <1 x float> [[SPLAT_SPLATINSERT9]], <1 x float> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP7:%.*]] = fmul <1 x float> [[BLOCK8]], [[SPLAT_SPLAT10]] +; CHECK-NEXT: [[BLOCK11:%.*]] = shufflevector <3 x float> [[SPLIT1]], <3 x float> undef, <1 x i32> +; CHECK-NEXT: [[TMP8:%.*]] = extractelement <2 x float> [[SPLIT2]], i64 1 +; CHECK-NEXT: [[SPLAT_SPLATINSERT12:%.*]] = insertelement <1 x float> undef, float [[TMP8]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT13:%.*]] = shufflevector <1 x float> [[SPLAT_SPLATINSERT12]], <1 x float> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP9:%.*]] = call <1 x float> @llvm.fmuladd.v1f32(<1 x float> [[BLOCK11]], <1 x float> [[SPLAT_SPLAT13]], <1 x float> [[TMP7]]) +; CHECK-NEXT: [[TMP10:%.*]] = shufflevector <1 x float> [[TMP9]], <1 x float> undef, <3 x i32> +; CHECK-NEXT: [[TMP11:%.*]] = shufflevector <3 x float> [[TMP5]], <3 x float> [[TMP10]], <3 x i32> +; CHECK-NEXT: [[BLOCK14:%.*]] = shufflevector <3 x float> [[SPLIT]], <3 x float> undef, <1 x i32> +; CHECK-NEXT: [[TMP12:%.*]] = extractelement <2 x float> [[SPLIT2]], i64 0 +; CHECK-NEXT: [[SPLAT_SPLATINSERT15:%.*]] = insertelement <1 x float> undef, float [[TMP12]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT16:%.*]] = shufflevector <1 x float> [[SPLAT_SPLATINSERT15]], <1 x float> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP13:%.*]] = fmul <1 x float> [[BLOCK14]], [[SPLAT_SPLAT16]] +; CHECK-NEXT: [[BLOCK17:%.*]] = shufflevector <3 x float> [[SPLIT1]], <3 x float> undef, <1 x i32> +; CHECK-NEXT: [[TMP14:%.*]] = extractelement <2 x float> [[SPLIT2]], i64 1 +; CHECK-NEXT: [[SPLAT_SPLATINSERT18:%.*]] = insertelement <1 x float> undef, float [[TMP14]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT19:%.*]] = shufflevector <1 x float> [[SPLAT_SPLATINSERT18]], <1 x float> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP15:%.*]] = call <1 x float> @llvm.fmuladd.v1f32(<1 x float> [[BLOCK17]], <1 x float> [[SPLAT_SPLAT19]], <1 x float> [[TMP13]]) +; CHECK-NEXT: [[TMP16:%.*]] = shufflevector <1 x float> [[TMP15]], <1 x float> undef, <3 x i32> +; CHECK-NEXT: [[TMP17:%.*]] = shufflevector <3 x float> [[TMP11]], <3 x float> [[TMP16]], <3 x i32> +; CHECK-NEXT: [[BLOCK20:%.*]] = shufflevector <3 x float> [[SPLIT]], <3 x float> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP18:%.*]] = extractelement <2 x float> [[SPLIT3]], i64 0 +; CHECK-NEXT: [[SPLAT_SPLATINSERT21:%.*]] = insertelement <1 x float> undef, float [[TMP18]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT22:%.*]] = shufflevector <1 x float> [[SPLAT_SPLATINSERT21]], <1 x float> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP19:%.*]] = fmul <1 x float> [[BLOCK20]], [[SPLAT_SPLAT22]] +; CHECK-NEXT: [[BLOCK23:%.*]] = shufflevector <3 x float> [[SPLIT1]], <3 x float> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP20:%.*]] = extractelement <2 x float> [[SPLIT3]], i64 1 +; CHECK-NEXT: [[SPLAT_SPLATINSERT24:%.*]] = insertelement <1 x float> undef, float [[TMP20]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT25:%.*]] = shufflevector <1 x float> [[SPLAT_SPLATINSERT24]], <1 x float> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP21:%.*]] = call <1 x float> @llvm.fmuladd.v1f32(<1 x float> [[BLOCK23]], <1 x float> [[SPLAT_SPLAT25]], <1 x float> [[TMP19]]) +; CHECK-NEXT: [[TMP22:%.*]] = shufflevector <1 x float> [[TMP21]], <1 x float> undef, <3 x i32> +; CHECK-NEXT: [[TMP23:%.*]] = shufflevector <3 x float> undef, <3 x float> [[TMP22]], <3 x i32> +; CHECK-NEXT: [[BLOCK26:%.*]] = shufflevector <3 x float> [[SPLIT]], <3 x float> undef, <1 x i32> +; CHECK-NEXT: [[TMP24:%.*]] = extractelement <2 x float> [[SPLIT3]], i64 0 +; CHECK-NEXT: [[SPLAT_SPLATINSERT27:%.*]] = insertelement <1 x float> undef, float [[TMP24]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT28:%.*]] = shufflevector <1 x float> [[SPLAT_SPLATINSERT27]], <1 x float> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP25:%.*]] = fmul <1 x float> [[BLOCK26]], [[SPLAT_SPLAT28]] +; CHECK-NEXT: [[BLOCK29:%.*]] = shufflevector <3 x float> [[SPLIT1]], <3 x float> undef, <1 x i32> +; CHECK-NEXT: [[TMP26:%.*]] = extractelement <2 x float> [[SPLIT3]], i64 1 +; CHECK-NEXT: [[SPLAT_SPLATINSERT30:%.*]] = insertelement <1 x float> undef, float [[TMP26]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT31:%.*]] = shufflevector <1 x float> [[SPLAT_SPLATINSERT30]], <1 x float> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP27:%.*]] = call <1 x float> @llvm.fmuladd.v1f32(<1 x float> [[BLOCK29]], <1 x float> [[SPLAT_SPLAT31]], <1 x float> [[TMP25]]) +; CHECK-NEXT: [[TMP28:%.*]] = shufflevector <1 x float> [[TMP27]], <1 x float> undef, <3 x i32> +; CHECK-NEXT: [[TMP29:%.*]] = shufflevector <3 x float> [[TMP23]], <3 x float> [[TMP28]], <3 x i32> +; CHECK-NEXT: [[BLOCK32:%.*]] = shufflevector <3 x float> [[SPLIT]], <3 x float> undef, <1 x i32> +; CHECK-NEXT: [[TMP30:%.*]] = extractelement <2 x float> [[SPLIT3]], i64 0 +; CHECK-NEXT: [[SPLAT_SPLATINSERT33:%.*]] = insertelement <1 x float> undef, float [[TMP30]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT34:%.*]] = shufflevector <1 x float> [[SPLAT_SPLATINSERT33]], <1 x float> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP31:%.*]] = fmul <1 x float> [[BLOCK32]], [[SPLAT_SPLAT34]] +; CHECK-NEXT: [[BLOCK35:%.*]] = shufflevector <3 x float> [[SPLIT1]], <3 x float> undef, <1 x i32> +; CHECK-NEXT: [[TMP32:%.*]] = extractelement <2 x float> [[SPLIT3]], i64 1 +; CHECK-NEXT: [[SPLAT_SPLATINSERT36:%.*]] = insertelement <1 x float> undef, float [[TMP32]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT37:%.*]] = shufflevector <1 x float> [[SPLAT_SPLATINSERT36]], <1 x float> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP33:%.*]] = call <1 x float> @llvm.fmuladd.v1f32(<1 x float> [[BLOCK35]], <1 x float> [[SPLAT_SPLAT37]], <1 x float> [[TMP31]]) +; CHECK-NEXT: [[TMP34:%.*]] = shufflevector <1 x float> [[TMP33]], <1 x float> undef, <3 x i32> +; CHECK-NEXT: [[TMP35:%.*]] = shufflevector <3 x float> [[TMP29]], <3 x float> [[TMP34]], <3 x i32> +; CHECK-NEXT: [[BLOCK38:%.*]] = shufflevector <3 x float> [[SPLIT]], <3 x float> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP36:%.*]] = extractelement <2 x float> [[SPLIT4]], i64 0 +; CHECK-NEXT: [[SPLAT_SPLATINSERT39:%.*]] = insertelement <1 x float> undef, float [[TMP36]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT40:%.*]] = shufflevector <1 x float> [[SPLAT_SPLATINSERT39]], <1 x float> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP37:%.*]] = fmul <1 x float> [[BLOCK38]], [[SPLAT_SPLAT40]] +; CHECK-NEXT: [[BLOCK41:%.*]] = shufflevector <3 x float> [[SPLIT1]], <3 x float> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP38:%.*]] = extractelement <2 x float> [[SPLIT4]], i64 1 +; CHECK-NEXT: [[SPLAT_SPLATINSERT42:%.*]] = insertelement <1 x float> undef, float [[TMP38]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT43:%.*]] = shufflevector <1 x float> [[SPLAT_SPLATINSERT42]], <1 x float> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP39:%.*]] = call <1 x float> @llvm.fmuladd.v1f32(<1 x float> [[BLOCK41]], <1 x float> [[SPLAT_SPLAT43]], <1 x float> [[TMP37]]) +; CHECK-NEXT: [[TMP40:%.*]] = shufflevector <1 x float> [[TMP39]], <1 x float> undef, <3 x i32> +; CHECK-NEXT: [[TMP41:%.*]] = shufflevector <3 x float> undef, <3 x float> [[TMP40]], <3 x i32> +; CHECK-NEXT: [[BLOCK44:%.*]] = shufflevector <3 x float> [[SPLIT]], <3 x float> undef, <1 x i32> +; CHECK-NEXT: [[TMP42:%.*]] = extractelement <2 x float> [[SPLIT4]], i64 0 +; CHECK-NEXT: [[SPLAT_SPLATINSERT45:%.*]] = insertelement <1 x float> undef, float [[TMP42]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT46:%.*]] = shufflevector <1 x float> [[SPLAT_SPLATINSERT45]], <1 x float> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP43:%.*]] = fmul <1 x float> [[BLOCK44]], [[SPLAT_SPLAT46]] +; CHECK-NEXT: [[BLOCK47:%.*]] = shufflevector <3 x float> [[SPLIT1]], <3 x float> undef, <1 x i32> +; CHECK-NEXT: [[TMP44:%.*]] = extractelement <2 x float> [[SPLIT4]], i64 1 +; CHECK-NEXT: [[SPLAT_SPLATINSERT48:%.*]] = insertelement <1 x float> undef, float [[TMP44]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT49:%.*]] = shufflevector <1 x float> [[SPLAT_SPLATINSERT48]], <1 x float> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP45:%.*]] = call <1 x float> @llvm.fmuladd.v1f32(<1 x float> [[BLOCK47]], <1 x float> [[SPLAT_SPLAT49]], <1 x float> [[TMP43]]) +; CHECK-NEXT: [[TMP46:%.*]] = shufflevector <1 x float> [[TMP45]], <1 x float> undef, <3 x i32> +; CHECK-NEXT: [[TMP47:%.*]] = shufflevector <3 x float> [[TMP41]], <3 x float> [[TMP46]], <3 x i32> +; CHECK-NEXT: [[BLOCK50:%.*]] = shufflevector <3 x float> [[SPLIT]], <3 x float> undef, <1 x i32> +; CHECK-NEXT: [[TMP48:%.*]] = extractelement <2 x float> [[SPLIT4]], i64 0 +; CHECK-NEXT: [[SPLAT_SPLATINSERT51:%.*]] = insertelement <1 x float> undef, float [[TMP48]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT52:%.*]] = shufflevector <1 x float> [[SPLAT_SPLATINSERT51]], <1 x float> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP49:%.*]] = fmul <1 x float> [[BLOCK50]], [[SPLAT_SPLAT52]] +; CHECK-NEXT: [[BLOCK53:%.*]] = shufflevector <3 x float> [[SPLIT1]], <3 x float> undef, <1 x i32> +; CHECK-NEXT: [[TMP50:%.*]] = extractelement <2 x float> [[SPLIT4]], i64 1 +; CHECK-NEXT: [[SPLAT_SPLATINSERT54:%.*]] = insertelement <1 x float> undef, float [[TMP50]], i32 0 +; CHECK-NEXT: [[SPLAT_SPLAT55:%.*]] = shufflevector <1 x float> [[SPLAT_SPLATINSERT54]], <1 x float> undef, <1 x i32> zeroinitializer +; CHECK-NEXT: [[TMP51:%.*]] = call <1 x float> @llvm.fmuladd.v1f32(<1 x float> [[BLOCK53]], <1 x float> [[SPLAT_SPLAT55]], <1 x float> [[TMP49]]) +; CHECK-NEXT: [[TMP52:%.*]] = shufflevector <1 x float> [[TMP51]], <1 x float> undef, <3 x i32> +; CHECK-NEXT: [[TMP53:%.*]] = shufflevector <3 x float> [[TMP47]], <3 x float> [[TMP52]], <3 x i32> +; CHECK-NEXT: [[TMP54:%.*]] = shufflevector <3 x float> [[TMP17]], <3 x float> [[TMP35]], <6 x i32> +; CHECK-NEXT: [[TMP55:%.*]] = shufflevector <3 x float> [[TMP53]], <3 x float> undef, <6 x i32> +; CHECK-NEXT: [[TMP56:%.*]] = shufflevector <6 x float> [[TMP54]], <6 x float> [[TMP55]], <9 x i32> +; CHECK-NEXT: ret <9 x float> [[TMP56]] +; +entry: + %c = call <9 x float> @llvm.matrix.multiply.v6f32.v6f32.v6f32(<6 x float> %a, <6 x float> %b, i32 3, i32 2, i32 3) + ret <9 x float> %c +} + +declare <9 x float> @llvm.matrix.multiply.v6f32.v6f32.v6f32(<6 x float>, <6 x float>, i32, i32, i32)