//===----------------------------------------------------------------------===//
// Linalg matmul ops.
//===----------------------------------------------------------------------===//

func.func @matmul_384x384x512() -> tensor<384x512xf32> {
    %lhs = util.unfoldable_constant dense<1.0> : tensor<384x384xf32>
    %rhs = util.unfoldable_constant dense<1.0> : tensor<384x512xf32>
    %dst = util.unfoldable_constant dense<1.0> : tensor<384x512xf32>
    %0 = linalg.matmul ins(%lhs, %rhs : tensor<384x384xf32>, tensor<384x512xf32>) outs(%dst : tensor<384x512xf32>) -> tensor<384x512xf32>
    return %0 : tensor<384x512xf32>
}

func.func @mmt4d_384x384x512_4x1x4() -> tensor<96x128x4x4xf32> {
    %lhs = util.unfoldable_constant dense<1.0> : tensor<96x384x4x1xf32>
    %rhs = util.unfoldable_constant dense<1.0> : tensor<128x384x4x1xf32>
    %dst = util.unfoldable_constant dense<1.0> : tensor<96x128x4x4xf32>
    %0 = linalg.mmt4d ins(%lhs, %rhs : tensor<96x384x4x1xf32>, tensor<128x384x4x1xf32>) outs(%dst : tensor<96x128x4x4xf32>) -> tensor<96x128x4x4xf32>
    return %0 : tensor<96x128x4x4xf32>
}

func.func @mmt4d_384x384x512_8x1x8() -> tensor<48x64x8x8xf32> {
    %lhs = util.unfoldable_constant dense<1.0> : tensor<48x384x8x1xf32>
    %rhs = util.unfoldable_constant dense<1.0> : tensor<64x384x8x1xf32>
    %dst = util.unfoldable_constant dense<1.0> : tensor<48x64x8x8xf32>
    %0 = linalg.mmt4d ins(%lhs, %rhs : tensor<48x384x8x1xf32>, tensor<64x384x8x1xf32>) outs(%dst : tensor<48x64x8x8xf32>) -> tensor<48x64x8x8xf32>
    return %0 : tensor<48x64x8x8xf32>
}