| // RUN: iree-dialects-opt --linalg-transform-interp %s | FileCheck %s |
| |
| func.func @fun_to_benchmark(%arg0: tensor<128x128xf32>, %arg1: tensor<128x128xf32>, %arg2: tensor<128x128xf32>) -> |
| tensor<128x128xf32> attributes {passthrough = ["noinline", ["target-cpu", "skylake-avx512"], ["prefer-vector-width", "512"]]} { |
| // With scalarization we expect vectorization to still work albeit with a leading |
| // `1` dimension. |
| // CHECK: vector.contract {{.*}} : vector<1x32xf32>, vector<32x16xf32> into vector<1x16xf32> |
| %0 = linalg.matmul ins(%arg0, %arg1 : tensor<128x128xf32>, tensor<128x128xf32>) |
| outs(%arg2 : tensor<128x128xf32>) -> tensor<128x128xf32> |
| return %0 : tensor<128x128xf32> |
| } |
| |
| transform.with_pdl_patterns { |
| ^bb0(%arg0: !pdl.operation): |
| pdl.pattern @isa_linalg.matmul : benefit(1) { |
| %0 = operands |
| %1 = types |
| %2 = operation "linalg.matmul"(%0 : !pdl.range<value>) -> (%1 : !pdl.range<type>) |
| rewrite %2 with "transform.dialect" |
| } |
| |
| transform.structured.canonicalized_sequence %arg0 { |
| ^bb1(%arg1: !pdl.operation): |
| %0 = pdl_match @isa_linalg.matmul in %arg1 |
| %tiled_linalg_op, %loops:3 = transform.structured.tile %0 {interchange = [1, 0, 2], sizes = [6, 16, 32]} |
| %1 = transform.loop.peel %loops#0 |
| // This test checks the proper handling of the scalarize dims attribute. |
| // The first dimension does not divide but we can always scalarize a `?` into `1` |
| // and enable vectorization of a lower-rank op this way. |
| %tiled_linalg_op_0 = transform.structured.scalarize %tiled_linalg_op |
| transform.structured2.vectorize {vectorize_padding = false} |
| } |
| } |
