| // RUN: mlir-opt %s -canonicalize="test-convergence" -split-input-file -allow-unregistered-dialect | FileCheck %s |
| |
| // CHECK-LABEL: create_vector_mask_to_constant_mask |
| func.func @create_vector_mask_to_constant_mask() -> (vector<4x3xi1>) { |
| %c2 = arith.constant 2 : index |
| %c3 = arith.constant 3 : index |
| // CHECK: vector.constant_mask [3, 2] : vector<4x3xi1> |
| %0 = vector.create_mask %c3, %c2 : vector<4x3xi1> |
| return %0 : vector<4x3xi1> |
| } |
| // ----- |
| |
| // CHECK-LABEL: create_scalable_vector_mask_to_constant_mask |
| func.func @create_scalable_vector_mask_to_constant_mask() -> (vector<[8]xi1>) { |
| %c-1 = arith.constant -1 : index |
| // CHECK: vector.constant_mask [0] : vector<[8]xi1> |
| %0 = vector.create_mask %c-1 : vector<[8]xi1> |
| return %0 : vector<[8]xi1> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: create_vector_mask_to_constant_mask_truncation |
| func.func @create_vector_mask_to_constant_mask_truncation() -> (vector<4x3xi1>) { |
| %c2 = arith.constant 2 : index |
| %c5 = arith.constant 5 : index |
| // CHECK: vector.constant_mask [4, 2] : vector<4x3xi1> |
| %0 = vector.create_mask %c5, %c2 : vector<4x3xi1> |
| return %0 : vector<4x3xi1> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: create_vector_mask_to_constant_mask_truncation_neg |
| func.func @create_vector_mask_to_constant_mask_truncation_neg() -> (vector<4x3xi1>) { |
| %cneg2 = arith.constant -2 : index |
| %c5 = arith.constant 5 : index |
| // CHECK: vector.constant_mask [0, 0] : vector<4x3xi1> |
| %0 = vector.create_mask %c5, %cneg2 : vector<4x3xi1> |
| return %0 : vector<4x3xi1> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: create_vector_mask_to_constant_mask_truncation_zero |
| func.func @create_vector_mask_to_constant_mask_truncation_zero() -> (vector<4x3xi1>) { |
| %c2 = arith.constant 2 : index |
| %c0 = arith.constant 0 : index |
| // CHECK: vector.constant_mask [0, 0] : vector<4x3xi1> |
| %0 = vector.create_mask %c0, %c2 : vector<4x3xi1> |
| return %0 : vector<4x3xi1> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: create_vector_mask_to_constant_mask_scalable_all_true |
| func.func @create_vector_mask_to_constant_mask_scalable_all_true() -> (vector<8x[16]xi1>) { |
| %c8 = arith.constant 8 : index |
| %c16 = arith.constant 16 : index |
| %0 = vector.vscale |
| %1 = arith.muli %0, %c16 : index |
| // CHECK: vector.constant_mask [8, 16] : vector<8x[16]xi1> |
| %10 = vector.create_mask %c8, %1 : vector<8x[16]xi1> |
| return %10 : vector<8x[16]xi1> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: create_mask_transpose_to_transposed_create_mask |
| // CHECK-SAME: %[[DIM0:.*]]: index, %[[DIM1:.*]]: index, %[[DIM2:.*]]: index |
| func.func @create_mask_transpose_to_transposed_create_mask( |
| %dim0: index, %dim1: index, %dim2: index) -> (vector<2x3x4xi1>, vector<4x2x3xi1>) { |
| // CHECK: vector.create_mask %[[DIM0]], %[[DIM1]], %[[DIM2]] : vector<2x3x4xi1> |
| // CHECK: vector.create_mask %[[DIM2]], %[[DIM0]], %[[DIM1]] : vector<4x2x3xi1> |
| // CHECK-NOT: vector.transpose |
| %0 = vector.create_mask %dim0, %dim1, %dim2 : vector<2x3x4xi1> |
| %1 = vector.transpose %0, [2, 0, 1] : vector<2x3x4xi1> to vector<4x2x3xi1> |
| return %0, %1 : vector<2x3x4xi1>, vector<4x2x3xi1> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: extract_from_create_mask |
| // CHECK-SAME: %[[DIM0:.*]]: index, %[[DIM1:.*]]: index |
| func.func @extract_from_create_mask(%dim0: index, %dim1: index) -> vector<[4]x[4]xi1> { |
| %c2 = arith.constant 2 : index |
| %mask = vector.create_mask %c2, %dim0, %dim1 : vector<4x[4]x[4]xi1> |
| // CHECK: vector.create_mask %[[DIM0]], %[[DIM1]] : vector<[4]x[4]xi1> |
| // CHECK-NOT: vector.extract |
| %extract = vector.extract %mask[1] : vector<[4]x[4]xi1> from vector<4x[4]x[4]xi1> |
| return %extract : vector<[4]x[4]xi1> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: extract_from_create_mask_all_false |
| func.func @extract_from_create_mask_all_false(%dim0: index, %dim1: index) -> vector<[4]x[4]xi1> { |
| %c2 = arith.constant 2 : index |
| %mask = vector.create_mask %c2, %dim0, %dim1 : vector<4x[4]x[4]xi1> |
| // CHECK: arith.constant dense<false> : vector<[4]x[4]xi1> |
| // CHECK-NOT: vector.extract |
| %extract = vector.extract %mask[2] : vector<[4]x[4]xi1> from vector<4x[4]x[4]xi1> |
| return %extract : vector<[4]x[4]xi1> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: extract_from_create_mask_leading_scalable |
| // CHECK-SAME: %[[DIM0:.*]]: index |
| func.func @extract_from_create_mask_leading_scalable(%dim0: index) -> vector<8xi1> { |
| %c3 = arith.constant 3 : index |
| %mask = vector.create_mask %c3, %dim0 : vector<[4]x8xi1> |
| // CHECK: vector.create_mask %[[DIM0]] : vector<8xi1> |
| // CHECK-NOT: vector.extract |
| %extract = vector.extract %mask[1] : vector<8xi1> from vector<[4]x8xi1> |
| return %extract : vector<8xi1> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: extract_from_create_mask_dynamic_position |
| // CHECK-SAME: %[[DIM0:.*]]: index, %[[INDEX:.*]]: index |
| func.func @extract_from_create_mask_dynamic_position(%dim0: index, %index: index) -> vector<6xi1> { |
| %c4 = arith.constant 4 : index |
| %c3 = arith.constant 3 : index |
| %mask = vector.create_mask %c3, %c4, %dim0 : vector<4x4x6xi1> |
| // CHECK: vector.create_mask %[[DIM0]] : vector<6xi1> |
| // CHECK-NOT: vector.extract |
| %extract = vector.extract %mask[2, %index] : vector<6xi1> from vector<4x4x6xi1> |
| return %extract : vector<6xi1> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: extract_from_create_mask_dynamic_position_all_false |
| // CHECK-SAME: %[[DIM0:.*]]: index, %[[INDEX:.*]]: index |
| func.func @extract_from_create_mask_dynamic_position_all_false(%dim0: index, %index: index) -> vector<6xi1> { |
| %c0 = arith.constant 0 : index |
| %c1 = arith.constant 1 : index |
| %mask = vector.create_mask %c1, %c0, %dim0 : vector<1x4x6xi1> |
| // CHECK: arith.constant dense<false> : vector<6xi1> |
| // CHECK-NOT: vector.extract |
| %extract = vector.extract %mask[0, %index] : vector<6xi1> from vector<1x4x6xi1> |
| return %extract : vector<6xi1> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: extract_from_create_mask_dynamic_position_unknown |
| // CHECK-SAME: %[[DIM0:.*]]: index, %[[INDEX:.*]]: index |
| func.func @extract_from_create_mask_dynamic_position_unknown(%dim0: index, %index: index) -> vector<6xi1> { |
| %c2 = arith.constant 2 : index |
| %mask = vector.create_mask %c2, %dim0 : vector<4x6xi1> |
| // CHECK: %[[C2:.*]] = arith.constant 2 : index |
| // CHECK-NEXT: %[[MASK:.*]] = vector.create_mask %[[C2]], %[[DIM0]] : vector<4x6xi1> |
| // CHECK-NEXT: vector.extract %[[MASK]][%[[INDEX]]] : vector<6xi1> from vector<4x6xi1> |
| %extract = vector.extract %mask[%index] : vector<6xi1> from vector<4x6xi1> |
| return %extract : vector<6xi1> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: extract_from_create_mask_mixed_position_unknown |
| // CHECK-SAME: %[[DIM0:.*]]: index, %[[INDEX:.*]]: index |
| func.func @extract_from_create_mask_mixed_position_unknown(%dim0: index, %index0: index) -> vector<4xi1> { |
| %c2 = arith.constant 2 : index |
| %mask = vector.create_mask %c2, %c2, %dim0 : vector<2x4x4xi1> |
| // CHECK: %[[C2:.*]] = arith.constant 2 : index |
| // CHECK-NEXT: %[[MASK:.*]] = vector.create_mask %[[C2]], %[[C2]], %[[DIM0]] : vector<2x4x4xi1> |
| // CHECK-NEXT: vector.extract %[[MASK]][1, %[[INDEX]]] : vector<4xi1> from vector<2x4x4xi1> |
| %extract = vector.extract %mask[1, %index0] : vector<4xi1> from vector<2x4x4xi1> |
| return %extract : vector<4xi1> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: extract_from_non_constant_create_mask |
| // CHECK-SAME: %[[DIM0:.*]]: index |
| func.func @extract_from_non_constant_create_mask(%dim0: index) -> vector<[2]xi1> { |
| %mask = vector.create_mask %dim0, %dim0 : vector<[2]x[2]xi1> |
| // CHECK: %[[MASK:.*]] = vector.create_mask %[[DIM0]], %[[DIM0]] : vector<[2]x[2]xi1> |
| // CHECK-NEXT: vector.extract %[[MASK]][0] : vector<[2]xi1> from vector<[2]x[2]xi1> |
| %extract = vector.extract %mask[0] : vector<[2]xi1> from vector<[2]x[2]xi1> |
| return %extract : vector<[2]xi1> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: constant_mask_transpose_to_transposed_constant_mask |
| func.func @constant_mask_transpose_to_transposed_constant_mask() -> (vector<2x3x4xi1>, vector<4x2x3xi1>) { |
| // CHECK: vector.constant_mask [1, 2, 3] : vector<2x3x4xi1> |
| // CHECK: vector.constant_mask [3, 1, 2] : vector<4x2x3xi1> |
| // CHECK-NOT: vector.transpose |
| %0 = vector.constant_mask [1, 2, 3] : vector<2x3x4xi1> |
| %1 = vector.transpose %0, [2, 0, 1] : vector<2x3x4xi1> to vector<4x2x3xi1> |
| return %0, %1 : vector<2x3x4xi1>, vector<4x2x3xi1> |
| } |
| |
| // ----- |
| |
| func.func @extract_strided_slice_of_constant_mask() -> (vector<2x2xi1>) { |
| %0 = vector.constant_mask [2, 2] : vector<4x3xi1> |
| %1 = vector.extract_strided_slice %0 |
| {offsets = [0, 0], sizes = [2, 2], strides = [1, 1]} |
| : vector<4x3xi1> to vector<2x2xi1> |
| // CHECK: vector.constant_mask [2, 2] : vector<2x2xi1> |
| return %1 : vector<2x2xi1> |
| } |
| |
| // ----- |
| |
| func.func @extract_strided_slice_of_constant_mask() -> (vector<2x2xi1>) { |
| %0 = vector.constant_mask [2, 2] : vector<4x3xi1> |
| %1 = vector.extract_strided_slice %0 |
| {offsets = [1, 0], sizes = [2, 2], strides = [1, 1]} |
| : vector<4x3xi1> to vector<2x2xi1> |
| // CHECK: vector.constant_mask [1, 2] : vector<2x2xi1> |
| return %1 : vector<2x2xi1> |
| } |
| |
| // ----- |
| |
| func.func @extract_strided_slice_of_constant_mask() -> (vector<2x2xi1>) { |
| %0 = vector.constant_mask [2, 2] : vector<4x3xi1> |
| %1 = vector.extract_strided_slice %0 |
| {offsets = [0, 1], sizes = [2, 2], strides = [1, 1]} |
| : vector<4x3xi1> to vector<2x2xi1> |
| // CHECK: vector.constant_mask [2, 1] : vector<2x2xi1> |
| return %1 : vector<2x2xi1> |
| } |
| |
| // ----- |
| |
| func.func @extract_strided_slice_of_constant_mask() -> (vector<2x2xi1>) { |
| %0 = vector.constant_mask [2, 2] : vector<4x3xi1> |
| %1 = vector.extract_strided_slice %0 |
| {offsets = [2, 0], sizes = [2, 2], strides = [1, 1]} |
| : vector<4x3xi1> to vector<2x2xi1> |
| // CHECK: vector.constant_mask [0, 0] : vector<2x2xi1> |
| return %1 : vector<2x2xi1> |
| } |
| |
| // ----- |
| |
| func.func @extract_strided_slice_of_constant_mask() -> (vector<2x1xi1>) { |
| %0 = vector.constant_mask [2, 2] : vector<4x3xi1> |
| %1 = vector.extract_strided_slice %0 |
| {offsets = [0, 2], sizes = [2, 1], strides = [1, 1]} |
| : vector<4x3xi1> to vector<2x1xi1> |
| // CHECK: vector.constant_mask [0, 0] : vector<2x1xi1> |
| return %1 : vector<2x1xi1> |
| } |
| |
| // ----- |
| |
| func.func @extract_strided_slice_of_constant_mask() -> (vector<2x1xi1>) { |
| %0 = vector.constant_mask [2, 2] : vector<4x3xi1> |
| %1 = vector.extract_strided_slice %0 |
| {offsets = [0, 1], sizes = [2, 1], strides = [1, 1]} |
| : vector<4x3xi1> to vector<2x1xi1> |
| // CHECK: vector.constant_mask [2, 1] : vector<2x1xi1> |
| return %1 : vector<2x1xi1> |
| } |
| |
| // ----- |
| |
| func.func @extract_strided_slice_of_constant_mask() -> (vector<2x1xi1>) { |
| %0 = vector.constant_mask [2, 2] : vector<4x3xi1> |
| %1 = vector.extract_strided_slice %0 |
| {offsets = [1, 1], sizes = [2, 1], strides = [1, 1]} |
| : vector<4x3xi1> to vector<2x1xi1> |
| // CHECK: vector.constant_mask [1, 1] : vector<2x1xi1> |
| return %1 : vector<2x1xi1> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: extract_strided_fold |
| // CHECK-SAME: (%[[ARG:.*]]: vector<4x3xi1>) |
| // CHECK-NEXT: return %[[ARG]] : vector<4x3xi1> |
| func.func @extract_strided_fold(%arg : vector<4x3xi1>) -> (vector<4x3xi1>) { |
| %0 = vector.extract_strided_slice %arg |
| {offsets = [0, 0], sizes = [4, 3], strides = [1, 1]} |
| : vector<4x3xi1> to vector<4x3xi1> |
| return %0 : vector<4x3xi1> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: extract_strided_fold_insert |
| // CHECK-SAME: (%[[ARG:.*]]: vector<4x4xf32> |
| // CHECK-NEXT: return %[[ARG]] : vector<4x4xf32> |
| func.func @extract_strided_fold_insert(%a: vector<4x4xf32>, %b: vector<8x16xf32>) |
| -> (vector<4x4xf32>) { |
| %0 = vector.insert_strided_slice %a, %b {offsets = [2, 2], strides = [1, 1]} |
| : vector<4x4xf32> into vector<8x16xf32> |
| %1 = vector.extract_strided_slice %0 |
| {offsets = [2, 2], sizes = [4, 4], strides = [1, 1]} |
| : vector<8x16xf32> to vector<4x4xf32> |
| return %1 : vector<4x4xf32> |
| } |
| |
| // ----- |
| |
| // Case where the vector inserted is a subset of the vector extracted. |
| // CHECK-LABEL: extract_strided_fold_insert |
| // CHECK-SAME: (%[[ARG0:.*]]: vector<6x4xf32> |
| // CHECK-NEXT: %[[EXT:.*]] = vector.extract_strided_slice %[[ARG0]] |
| // CHECK-SAME: {offsets = [0, 0], sizes = [4, 4], strides = [1, 1]} |
| // CHECK-SAME: : vector<6x4xf32> to vector<4x4xf32> |
| // CHECK-NEXT: return %[[EXT]] : vector<4x4xf32> |
| func.func @extract_strided_fold_insert(%a: vector<6x4xf32>, %b: vector<8x16xf32>) |
| -> (vector<4x4xf32>) { |
| %0 = vector.insert_strided_slice %a, %b {offsets = [2, 2], strides = [1, 1]} |
| : vector<6x4xf32> into vector<8x16xf32> |
| %1 = vector.extract_strided_slice %0 |
| {offsets = [2, 2], sizes = [4, 4], strides = [1, 1]} |
| : vector<8x16xf32> to vector<4x4xf32> |
| return %1 : vector<4x4xf32> |
| } |
| |
| // ----- |
| |
| // Negative test where the extract is not a subset of the element inserted. |
| // CHECK-LABEL: extract_strided_fold_negative |
| // CHECK-SAME: (%[[ARG0:.*]]: vector<4x4xf32>, %[[ARG1:.*]]: vector<8x16xf32> |
| // CHECK: %[[INS:.*]] = vector.insert_strided_slice %[[ARG0]], %[[ARG1]] |
| // CHECK-SAME: {offsets = [2, 2], strides = [1, 1]} |
| // CHECK-SAME: : vector<4x4xf32> into vector<8x16xf32> |
| // CHECK: %[[EXT:.*]] = vector.extract_strided_slice %[[INS]] |
| // CHECK-SAME: {offsets = [2, 2], sizes = [6, 4], strides = [1, 1]} |
| // CHECK-SAME: : vector<8x16xf32> to vector<6x4xf32> |
| // CHECK-NEXT: return %[[EXT]] : vector<6x4xf32> |
| func.func @extract_strided_fold_negative(%a: vector<4x4xf32>, %b: vector<8x16xf32>) |
| -> (vector<6x4xf32>) { |
| %0 = vector.insert_strided_slice %a, %b {offsets = [2, 2], strides = [1, 1]} |
| : vector<4x4xf32> into vector<8x16xf32> |
| %1 = vector.extract_strided_slice %0 |
| {offsets = [2, 2], sizes = [6, 4], strides = [1, 1]} |
| : vector<8x16xf32> to vector<6x4xf32> |
| return %1 : vector<6x4xf32> |
| } |
| |
| // ----- |
| |
| // Case where we need to go through 2 level of insert element. |
| // CHECK-LABEL: extract_strided_fold_insert |
| // CHECK-SAME: (%[[ARG0:.*]]: vector<2x8xf32>, %[[ARG1:.*]]: vector<1x4xf32>, |
| // CHECK-NEXT: %[[EXT:.*]] = vector.extract_strided_slice %[[ARG1]] |
| // CHECK-SAME: {offsets = [0, 0], sizes = [1, 1], strides = [1, 1]} |
| // CHECK-SAME: : vector<1x4xf32> to vector<1x1xf32> |
| // CHECK-NEXT: return %[[EXT]] : vector<1x1xf32> |
| func.func @extract_strided_fold_insert(%a: vector<2x8xf32>, %b: vector<1x4xf32>, |
| %c : vector<1x4xf32>) -> (vector<1x1xf32>) { |
| %0 = vector.insert_strided_slice %b, %a {offsets = [0, 1], strides = [1, 1]} |
| : vector<1x4xf32> into vector<2x8xf32> |
| %1 = vector.insert_strided_slice %c, %0 {offsets = [1, 0], strides = [1, 1]} |
| : vector<1x4xf32> into vector<2x8xf32> |
| %2 = vector.extract_strided_slice %1 |
| {offsets = [0, 1], sizes = [1, 1], strides = [1, 1]} |
| : vector<2x8xf32> to vector<1x1xf32> |
| return %2 : vector<1x1xf32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: transpose_1D_identity |
| // CHECK-SAME: ([[ARG:%.*]]: vector<4xf32>) |
| func.func @transpose_1D_identity(%arg : vector<4xf32>) -> vector<4xf32> { |
| // CHECK-NOT: transpose |
| %0 = vector.transpose %arg, [0] : vector<4xf32> to vector<4xf32> |
| // CHECK-NEXT: return [[ARG]] |
| return %0 : vector<4xf32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: transpose_2D_identity |
| // CHECK-SAME: ([[ARG:%.*]]: vector<4x3xf32>) |
| func.func @transpose_2D_identity(%arg : vector<4x3xf32>) -> vector<4x3xf32> { |
| // CHECK-NOT: transpose |
| %0 = vector.transpose %arg, [0, 1] : vector<4x3xf32> to vector<4x3xf32> |
| // CHECK-NEXT: return [[ARG]] |
| return %0 : vector<4x3xf32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: transpose_3D_identity |
| // CHECK-SAME: ([[ARG:%.*]]: vector<4x3x2xf32>) |
| func.func @transpose_3D_identity(%arg : vector<4x3x2xf32>) -> vector<4x3x2xf32> { |
| // CHECK-NOT: transpose |
| %0 = vector.transpose %arg, [0, 1, 2] : vector<4x3x2xf32> to vector<4x3x2xf32> |
| // CHECK-NEXT: return [[ARG]] |
| return %0 : vector<4x3x2xf32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: transpose_2D_sequence |
| // CHECK-SAME: ([[ARG:%.*]]: vector<4x3xf32>) |
| func.func @transpose_2D_sequence(%arg : vector<4x3xf32>) -> vector<4x3xf32> { |
| // CHECK-NOT: transpose |
| %0 = vector.transpose %arg, [1, 0] : vector<4x3xf32> to vector<3x4xf32> |
| %1 = vector.transpose %0, [0, 1] : vector<3x4xf32> to vector<3x4xf32> |
| %2 = vector.transpose %1, [1, 0] : vector<3x4xf32> to vector<4x3xf32> |
| %3 = vector.transpose %2, [0, 1] : vector<4x3xf32> to vector<4x3xf32> |
| // CHECK: [[ADD:%.*]] = arith.addf [[ARG]], [[ARG]] |
| %4 = arith.addf %2, %3 : vector<4x3xf32> |
| // CHECK-NEXT: return [[ADD]] |
| return %4 : vector<4x3xf32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: transpose_3D_sequence |
| // CHECK-SAME: ([[ARG:%.*]]: vector<4x3x2xf32>) |
| func.func @transpose_3D_sequence(%arg : vector<4x3x2xf32>) -> vector<4x3x2xf32> { |
| // CHECK: [[T0:%.*]] = vector.transpose [[ARG]], [2, 1, 0] |
| %0 = vector.transpose %arg, [1, 2, 0] : vector<4x3x2xf32> to vector<3x2x4xf32> |
| %1 = vector.transpose %0, [1, 0, 2] : vector<3x2x4xf32> to vector<2x3x4xf32> |
| // CHECK: [[T1:%.*]] = vector.transpose %arg0, [2, 1, 0] |
| %2 = vector.transpose %1, [2, 1, 0] : vector<2x3x4xf32> to vector<4x3x2xf32> |
| %3 = vector.transpose %2, [2, 1, 0] : vector<4x3x2xf32> to vector<2x3x4xf32> |
| // CHECK: [[MUL:%.*]] = arith.mulf [[T0]], [[T1]] |
| %4 = arith.mulf %1, %3 : vector<2x3x4xf32> |
| // CHECK: [[T5:%.*]] = vector.transpose [[MUL]], [2, 1, 0] |
| %5 = vector.transpose %4, [2, 1, 0] : vector<2x3x4xf32> to vector<4x3x2xf32> |
| // CHECK-NOT: transpose |
| %6 = vector.transpose %3, [2, 1, 0] : vector<2x3x4xf32> to vector<4x3x2xf32> |
| // CHECK: [[ADD:%.*]] = arith.addf [[T5]], [[ARG]] |
| %7 = arith.addf %5, %6 : vector<4x3x2xf32> |
| // CHECK-NEXT: return [[ADD]] |
| return %7 : vector<4x3x2xf32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: cast_transfers |
| func.func @cast_transfers(%A: memref<4x8xf32>) -> (vector<4x8xf32>) { |
| %c0 = arith.constant 0 : index |
| %f0 = arith.constant 0.0 : f32 |
| %0 = memref.cast %A : memref<4x8xf32> to memref<?x?xf32> |
| |
| // CHECK: vector.transfer_read %{{.*}} {in_bounds = [true, true]} : memref<4x8xf32>, vector<4x8xf32> |
| %1 = vector.transfer_read %0[%c0, %c0], %f0 : memref<?x?xf32>, vector<4x8xf32> |
| |
| // CHECK: vector.transfer_write %{{.*}} {in_bounds = [true, true]} : vector<4x8xf32>, memref<4x8xf32> |
| vector.transfer_write %1, %0[%c0, %c0] : vector<4x8xf32>, memref<?x?xf32> |
| return %1 : vector<4x8xf32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: cast_transfers |
| func.func @cast_transfers(%A: tensor<4x8xf32>) -> (vector<4x8xf32>) { |
| %c0 = arith.constant 0 : index |
| %f0 = arith.constant 0.0 : f32 |
| %0 = tensor.cast %A : tensor<4x8xf32> to tensor<?x?xf32> |
| |
| // CHECK: vector.transfer_read %{{.*}} {in_bounds = [true, true]} : tensor<4x8xf32>, vector<4x8xf32> |
| %1 = vector.transfer_read %0[%c0, %c0], %f0 : tensor<?x?xf32>, vector<4x8xf32> |
| |
| return %1 : vector<4x8xf32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func @insert_extract_transpose_2d( |
| // CHECK-SAME: %[[V:[a-zA-Z0-9]*]]: vector<2x3xf32>, |
| // CHECK-SAME: %[[F0:[a-zA-Z0-9]*]]: f32, |
| // CHECK-SAME: %[[F1:[a-zA-Z0-9]*]]: f32, |
| // CHECK-SAME: %[[F2:[a-zA-Z0-9]*]]: f32, |
| // CHECK-SAME: %[[F3:[a-zA-Z0-9]*]]: f32 |
| func.func @insert_extract_transpose_2d( |
| %v: vector<2x3xf32>, %f0: f32, %f1: f32, %f2: f32, %f3: f32) |
| -> (f32, f32, f32) |
| { |
| %0 = vector.insert %f0, %v[0, 0] : f32 into vector<2x3xf32> |
| %1 = vector.insert %f1, %0[0, 1] : f32 into vector<2x3xf32> |
| %2 = vector.insert %f2, %1[1, 0] : f32 into vector<2x3xf32> |
| %3 = vector.insert %f3, %2[1, 1] : f32 into vector<2x3xf32> |
| %4 = vector.transpose %3, [1, 0] : vector<2x3xf32> to vector<3x2xf32> |
| %5 = vector.insert %f3, %4[1, 0] : f32 into vector<3x2xf32> |
| %6 = vector.transpose %5, [1, 0] : vector<3x2xf32> to vector<2x3xf32> |
| |
| // Expected %f2 from %2 = vector.insert %f2, %1[1, 0]. |
| %r1 = vector.extract %3[1, 0] : f32 from vector<2x3xf32> |
| |
| // Expected %f1 from %1 = vector.insert %f1, %0[0, 1] followed by |
| // transpose [1, 0]. |
| %r2 = vector.extract %4[1, 0] : f32 from vector<3x2xf32> |
| |
| // Expected %f2 from %2 = vector.insert %f2, %1[1, 0] followed by double |
| // transpose [1, 0]. |
| %r3 = vector.extract %6[1, 0] : f32 from vector<2x3xf32> |
| |
| // CHECK-NEXT: return %[[F2]], %[[F1]], %[[F2]] : f32, f32, f32 |
| return %r1, %r2, %r3 : f32, f32, f32 |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: insert_extract_chain |
| // CHECK-SAME: %[[V234:[a-zA-Z0-9]*]]: vector<2x3x4xf32> |
| // CHECK-SAME: %[[V34:[a-zA-Z0-9]*]]: vector<3x4xf32> |
| // CHECK-SAME: %[[V4:[a-zA-Z0-9]*]]: vector<4xf32> |
| func.func @insert_extract_chain(%v234: vector<2x3x4xf32>, %v34: vector<3x4xf32>, %v4: vector<4xf32>) |
| -> (vector<4xf32>, vector<4xf32>, vector<3x4xf32>, vector<3x4xf32>) { |
| // CHECK-NEXT: %[[A34:.*]] = vector.insert |
| %A34 = vector.insert %v34, %v234[0]: vector<3x4xf32> into vector<2x3x4xf32> |
| // CHECK-NEXT: %[[B34:.*]] = vector.insert |
| %B34 = vector.insert %v34, %A34[1]: vector<3x4xf32> into vector<2x3x4xf32> |
| // CHECK-NEXT: %[[A4:.*]] = vector.insert |
| %A4 = vector.insert %v4, %B34[1, 0]: vector<4xf32> into vector<2x3x4xf32> |
| // CHECK-NEXT: %[[B4:.*]] = vector.insert |
| %B4 = vector.insert %v4, %A4[1, 1]: vector<4xf32> into vector<2x3x4xf32> |
| |
| // Case 2.a. [1, 1] == insertpos ([1, 1]) |
| // Match %A4 insertionpos and fold to its source(i.e. %V4). |
| %r0 = vector.extract %B4[1, 1]: vector<4xf32> from vector<2x3x4xf32> |
| |
| // Case 3.a. insertpos ([1]) is a prefix of [1, 0]. |
| // Traverse %B34 to its source(i.e. %V34@[*0*]). |
| // CHECK-NEXT: %[[R1:.*]] = vector.extract %[[V34]][0] |
| %r1 = vector.extract %B34[1, 0]: vector<4xf32> from vector<2x3x4xf32> |
| |
| // Case 4. [1] is a prefix of insertpos ([1, 1]). |
| // Cannot traverse %B4. |
| // CHECK-NEXT: %[[R2:.*]] = vector.extract %[[B4]][1] |
| %r2 = vector.extract %B4[1]: vector<3x4xf32> from vector<2x3x4xf32> |
| |
| // Case 5. [0] is disjoint from insertpos ([1, 1]). |
| // Traverse %B4 to its dest(i.e. %A4@[0]). |
| // Traverse %A4 to its dest(i.e. %B34@[0]). |
| // Traverse %B34 to its dest(i.e. %A34@[0]). |
| // Match %A34 insertionpos and fold to its source(i.e. %V34). |
| %r3 = vector.extract %B4[0]: vector<3x4xf32> from vector<2x3x4xf32> |
| |
| // CHECK: return %[[V4]], %[[R1]], %[[R2]], %[[V34]] |
| return %r0, %r1, %r2, %r3: |
| vector<4xf32>, vector<4xf32>, vector<3x4xf32>, vector<3x4xf32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func @insert_extract_transpose_3d( |
| // CHECK-SAME: %[[V234:[a-zA-Z0-9]*]]: vector<2x3x4xf32> |
| func.func @insert_extract_transpose_3d( |
| %v234: vector<2x3x4xf32>, %v43: vector<4x3xf32>, %f0: f32) |
| -> (vector<4xf32>, vector<4xf32>, vector<4xf32>, vector<3x4xf32>) { |
| |
| %a432 = vector.transpose %v234, [2, 1, 0] : vector<2x3x4xf32> to vector<4x3x2xf32> |
| %b432 = vector.insert %f0, %a432[0, 0, 1] : f32 into vector<4x3x2xf32> |
| %c234 = vector.transpose %b432, [2, 1, 0] : vector<4x3x2xf32> to vector<2x3x4xf32> |
| // Case 1. %c234 = transpose [2,1,0] posWithSentinels [1,2,-1] -> [-1,2,1] |
| // Case 5. %b432 = insert [0,0,1] (inter([.,2,1], [.,0,1]) == 0) prop to %v432 |
| // Case 1. %a432 = transpose [2,1,0] posWithSentinels [-1,2,1] -> [1,2,-1] |
| // can extract directly from %v234, the rest folds. |
| // CHECK: %[[R0:.*]] = vector.extract %[[V234]][1, 2] |
| %r0 = vector.extract %c234[1, 2] : vector<4xf32> from vector<2x3x4xf32> |
| |
| // CHECK-NEXT: vector.transpose |
| // CHECK-NEXT: vector.insert |
| // CHECK-NEXT: %[[F234:.*]] = vector.transpose |
| %d432 = vector.transpose %v234, [2, 1, 0] : vector<2x3x4xf32> to vector<4x3x2xf32> |
| %e432 = vector.insert %f0, %d432[0, 2, 1] : f32 into vector<4x3x2xf32> |
| %f234 = vector.transpose %e432, [2, 1, 0] : vector<4x3x2xf32> to vector<2x3x4xf32> |
| // Case 1. %c234 = transpose [2,1,0] posWithSentinels [1,2,-1] -> [-1,2,1] |
| // Case 4. %b432 = insert [0,0,1] (inter([.,2,1], [.,2,1]) != 0) |
| // Bail, cannot do better than the current. |
| // CHECK: %[[R1:.*]] = vector.extract %[[F234]] |
| %r1 = vector.extract %f234[1, 2] : vector<4xf32> from vector<2x3x4xf32> |
| |
| // CHECK-NEXT: vector.transpose |
| // CHECK-NEXT: vector.insert |
| // CHECK-NEXT: %[[H234:.*]] = vector.transpose |
| %g243 = vector.transpose %v234, [0, 2, 1] : vector<2x3x4xf32> to vector<2x4x3xf32> |
| %h243 = vector.insert %v43, %g243[0] : vector<4x3xf32> into vector<2x4x3xf32> |
| %i234 = vector.transpose %h243, [0, 2, 1] : vector<2x4x3xf32> to vector<2x3x4xf32> |
| // Case 1. %i234 = transpose [0,2,1] posWithSentinels [0,-1,-2] -> [0,-2,-1] |
| // Case 3.b. %b432 = insert [0] is prefix of [0,.,.] but internal transpose. |
| // Bail, cannot do better than the current. |
| // CHECK: %[[R2:.*]] = vector.extract %[[H234]][0, 1] |
| %r2 = vector.extract %i234[0, 1] : vector<4xf32> from vector<2x3x4xf32> |
| |
| // CHECK-NEXT: vector.transpose |
| // CHECK-NEXT: vector.insert |
| // CHECK-NEXT: %[[K234:.*]] = vector.transpose |
| %j243 = vector.transpose %v234, [0, 2, 1] : vector<2x3x4xf32> to vector<2x4x3xf32> |
| %k243 = vector.insert %v43, %j243[0] : vector<4x3xf32> into vector<2x4x3xf32> |
| %l234 = vector.transpose %k243, [0, 2, 1] : vector<2x4x3xf32> to vector<2x3x4xf32> |
| // Case 1. %i234 = transpose [0,2,1] posWithSentinels [0,-1,-2] -> [0,-2,-1] |
| // Case 2.b. %b432 = insert [0] == [0,.,.] but internal transpose. |
| // Bail, cannot do better than the current. |
| // CHECK: %[[R3:.*]] = vector.extract %[[K234]][0] |
| %r3 = vector.extract %l234[0] : vector<3x4xf32> from vector<2x3x4xf32> |
| |
| // CHECK-NEXT: return %[[R0]], %[[R1]], %[[R2]], %[[R3]] |
| return %r0, %r1, %r2, %r3: vector<4xf32>, vector<4xf32>, vector<4xf32>, vector<3x4xf32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: fold_extracts |
| // CHECK-SAME: %[[A:[a-zA-Z0-9]*]]: vector<3x4x5x6xf32> |
| func.func @fold_extracts(%a : vector<3x4x5x6xf32>) -> (f32, vector<4x5x6xf32>) { |
| %b = vector.extract %a[0] : vector<4x5x6xf32> from vector<3x4x5x6xf32> |
| %c = vector.extract %b[1, 2] : vector<6xf32> from vector<4x5x6xf32> |
| // CHECK-NEXT: vector.extract %[[A]][0, 1, 2, 3] : f32 from vector<3x4x5x6xf32> |
| %d = vector.extract %c[3] : f32 from vector<6xf32> |
| |
| // CHECK-NEXT: vector.extract %[[A]][0] : vector<4x5x6xf32> from vector<3x4x5x6xf32> |
| %e = vector.extract %a[0] : vector<4x5x6xf32> from vector<3x4x5x6xf32> |
| |
| // CHECK-NEXT: return |
| return %d, %e : f32, vector<4x5x6xf32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: fold_extract_transpose |
| // CHECK-SAME: %[[A:[a-zA-Z0-9]*]]: vector<3x4x5x6xf32> |
| // CHECK-SAME: %[[B:[a-zA-Z0-9]*]]: vector<3x6x5x6xf32> |
| func.func @fold_extract_transpose( |
| %a : vector<3x4x5x6xf32>, %b : vector<3x6x5x6xf32>) -> ( |
| vector<6xf32>, vector<6xf32>, vector<6xf32>) { |
| // [3] is a proper most minor identity map in transpose. |
| // Permutation is a self inverse and we have. |
| // [0, 2, 1] ^ -1 o [0, 1, 2] = [0, 2, 1] o [0, 1, 2] |
| // = [0, 2, 1] |
| // CHECK-NEXT: vector.extract %[[A]][0, 2, 1] : vector<6xf32> from vector<3x4x5x6xf32> |
| %0 = vector.transpose %a, [0, 2, 1, 3] : vector<3x4x5x6xf32> to vector<3x5x4x6xf32> |
| %1 = vector.extract %0[0, 1, 2] : vector<6xf32> from vector<3x5x4x6xf32> |
| |
| // [3] is a proper most minor identity map in transpose. |
| // Permutation is a not self inverse and we have. |
| // [1, 2, 0] ^ -1 o [0, 1, 2] = [2, 0, 1] o [0, 1, 2] |
| // = [2, 0, 1] |
| // CHECK-NEXT: vector.extract %[[A]][2, 0, 1] : vector<6xf32> from vector<3x4x5x6xf32> |
| %2 = vector.transpose %a, [1, 2, 0, 3] : vector<3x4x5x6xf32> to vector<4x5x3x6xf32> |
| %3 = vector.extract %2[0, 1, 2] : vector<6xf32> from vector<4x5x3x6xf32> |
| |
| // Not a minor identity map so intra-vector level has been permuted |
| // CHECK-NEXT: vector.transpose %[[B]], [0, 2, 3, 1] |
| // CHECK-NEXT: vector.extract %{{.*}}[0, 1, 2] |
| %4 = vector.transpose %b, [0, 2, 3, 1] : vector<3x6x5x6xf32> to vector<3x5x6x6xf32> |
| %5 = vector.extract %4[0, 1, 2] : vector<6xf32> from vector<3x5x6x6xf32> |
| |
| return %1, %3, %5 : vector<6xf32>, vector<6xf32>, vector<6xf32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: fold_extract_broadcast |
| // CHECK-SAME: %[[A:.*]]: f32 |
| // CHECK: return %[[A]] : f32 |
| func.func @fold_extract_broadcast(%a : f32) -> f32 { |
| %b = vector.broadcast %a : f32 to vector<1x2x4xf32> |
| %r = vector.extract %b[0, 1, 2] : f32 from vector<1x2x4xf32> |
| return %r : f32 |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: fold_extract_broadcast_0dvec |
| // CHECK-SAME: %[[A:.*]]: vector<f32> |
| // CHECK: %[[B:.+]] = vector.extractelement %[[A]][] : vector<f32> |
| // CHECK: return %[[B]] : f32 |
| func.func @fold_extract_broadcast_0dvec(%a : vector<f32>) -> f32 { |
| %b = vector.broadcast %a : vector<f32> to vector<1x2x4xf32> |
| %r = vector.extract %b[0, 1, 2] : f32 from vector<1x2x4xf32> |
| return %r : f32 |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: fold_extract_broadcast_negative |
| // CHECK: vector.broadcast %{{.*}} : vector<1x1xf32> to vector<1x1x4xf32> |
| // CHECK: vector.extract %{{.*}}[0, 0] : vector<4xf32> from vector<1x1x4xf32> |
| func.func @fold_extract_broadcast_negative(%a : vector<1x1xf32>) -> vector<4xf32> { |
| %b = vector.broadcast %a : vector<1x1xf32> to vector<1x1x4xf32> |
| %r = vector.extract %b[0, 0] : vector<4xf32> from vector<1x1x4xf32> |
| return %r : vector<4xf32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: fold_extract_splat |
| // CHECK-SAME: %[[A:.*]]: f32 |
| // CHECK: return %[[A]] : f32 |
| func.func @fold_extract_splat(%a : f32) -> f32 { |
| %b = vector.splat %a : vector<1x2x4xf32> |
| %r = vector.extract %b[0, 1, 2] : f32 from vector<1x2x4xf32> |
| return %r : f32 |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: fold_extract_broadcast_vector |
| // CHECK-SAME: %[[A:.*]]: vector<4xf32> |
| // CHECK: return %[[A]] : vector<4xf32> |
| func.func @fold_extract_broadcast_vector(%a : vector<4xf32>) -> vector<4xf32> { |
| %b = vector.broadcast %a : vector<4xf32> to vector<1x2x4xf32> |
| %r = vector.extract %b[0, 1] : vector<4xf32> from vector<1x2x4xf32> |
| return %r : vector<4xf32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: fold_extract_broadcast |
| // CHECK-SAME: %[[A:.*]]: vector<4xf32> |
| // CHECK: %[[R:.*]] = vector.extract %[[A]][2] : f32 from vector<4xf32> |
| // CHECK: return %[[R]] : f32 |
| func.func @fold_extract_broadcast(%a : vector<4xf32>) -> f32 { |
| %b = vector.broadcast %a : vector<4xf32> to vector<1x2x4xf32> |
| %r = vector.extract %b[0, 1, 2] : f32 from vector<1x2x4xf32> |
| return %r : f32 |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: fold_extract_broadcast |
| // CHECK: %[[B:.*]] = vector.broadcast %{{.*}} : f32 to vector<4xf32> |
| // CHECK: return %[[B]] : vector<4xf32> |
| func.func @fold_extract_broadcast(%a : f32) -> vector<4xf32> { |
| %b = vector.broadcast %a : f32 to vector<1x2x4xf32> |
| %r = vector.extract %b[0, 1] : vector<4xf32> from vector<1x2x4xf32> |
| return %r : vector<4xf32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: fold_extract_broadcast |
| // CHECK-SAME: %[[A:.*]]: vector<1xf32> |
| // CHECK: %[[R:.*]] = vector.broadcast %[[A]] : vector<1xf32> to vector<8xf32> |
| // CHECK: return %[[R]] : vector<8xf32> |
| func.func @fold_extract_broadcast(%a : vector<1xf32>) -> vector<8xf32> { |
| %b = vector.broadcast %a : vector<1xf32> to vector<1x8xf32> |
| %r = vector.extract %b[0] : vector<8xf32> from vector<1x8xf32> |
| return %r : vector<8xf32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func @fold_extract_shapecast |
| // CHECK-SAME: (%[[A0:.*]]: vector<5x1x3x2xf32>, %[[A1:.*]]: vector<8x4x2xf32> |
| // CHECK: %[[R0:.*]] = vector.extract %[[A0]][1, 0, 1, 1] : f32 from vector<5x1x3x2xf32> |
| // CHECK: %[[R1:.*]] = vector.extract %[[A0]][1, 0, 2] : vector<2xf32> from vector<5x1x3x2xf32> |
| // CHECK: %[[R2:.*]] = vector.extract %[[A1]][7] : vector<4x2xf32> from vector<8x4x2xf32> |
| // CHECK: return %[[R0]], %[[R1]], %[[R2]], %[[A1]] : f32, vector<2xf32>, vector<4x2xf32>, vector<8x4x2xf32> |
| func.func @fold_extract_shapecast(%arg0 : vector<5x1x3x2xf32>, |
| %arg1 : vector<8x4x2xf32>) |
| -> (f32, vector<2xf32>, vector<4x2xf32>, vector<8x4x2xf32>) { |
| %0 = vector.shape_cast %arg0 : vector<5x1x3x2xf32> to vector<15x2xf32> |
| %1 = vector.shape_cast %arg1 : vector<8x4x2xf32> to vector<4x2x4x2xf32> |
| %2 = vector.shape_cast %arg1 : vector<8x4x2xf32> to vector<1x8x4x2xf32> |
| %r1 = vector.extract %0[4, 1] : f32 from vector<15x2xf32> |
| %r2 = vector.extract %0[5] : vector<2xf32> from vector<15x2xf32> |
| %r3 = vector.extract %1[3, 1] : vector<4x2xf32> from vector<4x2x4x2xf32> |
| %r4 = vector.extract %2[0] : vector<8x4x2xf32> from vector<1x8x4x2xf32> |
| return %r1, %r2, %r3, %r4 : f32, vector<2xf32>, vector<4x2xf32>, vector<8x4x2xf32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: fold_extract_shapecast_negative |
| // CHECK: %[[V:.*]] = vector.shape_cast %{{.*}} : vector<16xf32> to vector<2x4x2xf32> |
| // CHECK: %[[R:.*]] = vector.extract %[[V]][1] : vector<4x2xf32> from vector<2x4x2xf32> |
| // CHECK: return %[[R]] : vector<4x2xf32> |
| func.func @fold_extract_shapecast_negative(%arg0 : vector<16xf32>) -> vector<4x2xf32> { |
| %0 = vector.shape_cast %arg0 : vector<16xf32> to vector<2x4x2xf32> |
| %r = vector.extract %0[1] : vector<4x2xf32> from vector<2x4x2xf32> |
| return %r : vector<4x2xf32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: dont_fold_0d_extract_shapecast |
| // CHECK: %[[V:.*]] = vector.shape_cast %{{.*}} : vector<f32> to vector<1xf32> |
| // CHECK: %[[R:.*]] = vector.extract %[[V]][0] : f32 from vector<1xf32> |
| // CHECK: return %[[R]] : f32 |
| func.func @dont_fold_0d_extract_shapecast(%arg0 : vector<f32>) -> f32 { |
| %0 = vector.shape_cast %arg0 : vector<f32> to vector<1xf32> |
| %r = vector.extract %0[0] : f32 from vector<1xf32> |
| return %r : f32 |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: fold_extract_shapecast_to_shapecast |
| // CHECK-SAME: (%[[ARG:.+]]: vector<3x4xf32>) |
| // CHECK: %[[R:.+]] = vector.shape_cast %[[ARG]] : vector<3x4xf32> to vector<12xf32> |
| // CHECK: return %[[R]] |
| func.func @fold_extract_shapecast_to_shapecast(%arg0 : vector<3x4xf32>) -> vector<12xf32> { |
| %0 = vector.shape_cast %arg0 : vector<3x4xf32> to vector<1x12xf32> |
| %r = vector.extract %0[0] : vector<12xf32> from vector<1x12xf32> |
| return %r : vector<12xf32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: dont_fold_expand_collapse |
| // CHECK: %[[A:.*]] = vector.shape_cast %{{.*}} : vector<1x1x64xf32> to vector<1x1x8x8xf32> |
| // CHECK: %[[B:.*]] = vector.shape_cast %{{.*}} : vector<1x1x8x8xf32> to vector<8x8xf32> |
| // CHECK: return %[[B]] : vector<8x8xf32> |
| func.func @dont_fold_expand_collapse(%arg0: vector<1x1x64xf32>) -> vector<8x8xf32> { |
| %0 = vector.shape_cast %arg0 : vector<1x1x64xf32> to vector<1x1x8x8xf32> |
| %1 = vector.shape_cast %0 : vector<1x1x8x8xf32> to vector<8x8xf32> |
| return %1 : vector<8x8xf32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func @fold_broadcast_shapecast |
| // CHECK-SAME: (%[[V:.+]]: vector<4xf32>) |
| // CHECK: return %[[V]] |
| func.func @fold_broadcast_shapecast(%arg0: vector<4xf32>) -> vector<4xf32> { |
| %0 = vector.broadcast %arg0 : vector<4xf32> to vector<1x1x4xf32> |
| %1 = vector.shape_cast %0 : vector<1x1x4xf32> to vector<4xf32> |
| return %1 : vector<4xf32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func @canonicalize_broadcast_shapecast_scalar |
| // CHECK: vector.broadcast |
| // CHECK-NOT: vector.shape_cast |
| func.func @canonicalize_broadcast_shapecast_scalar(%arg0: f32) -> vector<1xf32> { |
| %0 = vector.broadcast %arg0 : f32 to vector<1x1x1xf32> |
| %1 = vector.shape_cast %0 : vector<1x1x1xf32> to vector<1xf32> |
| return %1 : vector<1xf32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func @dont_fold_broadcast_shapecast_diff_shape |
| // CHECK: vector.broadcast |
| // CHECK: vector.shape_cast |
| func.func @dont_fold_broadcast_shapecast_diff_shape(%arg0: vector<4xf32>) -> vector<8xf32> { |
| %0 = vector.broadcast %arg0 : vector<4xf32> to vector<1x2x4xf32> |
| %1 = vector.shape_cast %0 : vector<1x2x4xf32> to vector<8xf32> |
| return %1 : vector<8xf32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func @canonicalize_broadcast_shapecast_to_broadcast |
| // CHECK: vector.broadcast |
| // CHECK-NOT: vector.shape_cast |
| func.func @canonicalize_broadcast_shapecast_to_broadcast(%arg0: vector<3xf32>) -> vector<8x3xf32> { |
| %0 = vector.broadcast %arg0 : vector<3xf32> to vector<2x4x3xf32> |
| %1 = vector.shape_cast %0 : vector<2x4x3xf32> to vector<8x3xf32> |
| return %1 : vector<8x3xf32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func @canonicalize_broadcast_shapecast_to_shapecast |
| // CHECK-NOT: vector.broadcast |
| // CHECK: vector.shape_cast {{.+}} : vector<3x4xf32> to vector<1x12xf32> |
| func.func @canonicalize_broadcast_shapecast_to_shapecast(%arg0: vector<3x4xf32>) -> vector<1x12xf32> { |
| %0 = vector.broadcast %arg0 : vector<3x4xf32> to vector<1x1x3x4xf32> |
| %1 = vector.shape_cast %0 : vector<1x1x3x4xf32> to vector<1x12xf32> |
| return %1 : vector<1x12xf32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: fold_vector_transfer_masks |
| func.func @fold_vector_transfer_masks(%A: memref<?x?xf32>) -> (vector<4x8xf32>) { |
| // CHECK: %[[C0:.+]] = arith.constant 0 : index |
| %c0 = arith.constant 0 : index |
| // CHECK: %[[F0:.+]] = arith.constant 0.000000e+00 : f32 |
| %f0 = arith.constant 0.0 : f32 |
| |
| %mask = vector.constant_mask [8, 4] : vector<8x4xi1> |
| |
| // CHECK: vector.transfer_read %{{.*}}, %[[F0]] {permutation_map |
| %1 = vector.transfer_read %A[%c0, %c0], %f0, %mask |
| {permutation_map = affine_map<(d0, d1) -> (d1, d0)>} : memref<?x?xf32>, vector<4x8xf32> |
| |
| // CHECK: vector.transfer_write {{.*}}[%[[C0]], %[[C0]]] {permutation_map |
| vector.transfer_write %1, %A[%c0, %c0], %mask |
| {permutation_map = affine_map<(d0, d1) -> (d1, d0)>} : vector<4x8xf32>, memref<?x?xf32> |
| |
| // CHECK: return |
| return %1 : vector<4x8xf32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: fold_vector_transfers |
| func.func @fold_vector_transfers(%A: memref<?x8xf32>) -> (vector<4x8xf32>, vector<4x9xf32>) { |
| %c0 = arith.constant 0 : index |
| %f0 = arith.constant 0.0 : f32 |
| |
| // CHECK: vector.transfer_read %{{.*}} {in_bounds = [false, true]} |
| %1 = vector.transfer_read %A[%c0, %c0], %f0 : memref<?x8xf32>, vector<4x8xf32> |
| |
| // CHECK: vector.transfer_write %{{.*}} {in_bounds = [false, true]} |
| vector.transfer_write %1, %A[%c0, %c0] : vector<4x8xf32>, memref<?x8xf32> |
| |
| // Both dims may be out-of-bounds, attribute is elided. |
| // CHECK: vector.transfer_read %{{.*}} |
| // CHECK-NOT: in_bounds |
| %2 = vector.transfer_read %A[%c0, %c0], %f0 : memref<?x8xf32>, vector<4x9xf32> |
| |
| // Both dims may be out-of-bounds, attribute is elided. |
| // CHECK: vector.transfer_write %{{.*}} |
| // CHECK-NOT: in_bounds |
| vector.transfer_write %2, %A[%c0, %c0] : vector<4x9xf32>, memref<?x8xf32> |
| |
| // CHECK: return |
| return %1, %2 : vector<4x8xf32>, vector<4x9xf32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: bitcast_folding |
| // CHECK-SAME: %[[A:.*]]: vector<4x8xf32> |
| // CHECK-SAME: %[[B:.*]]: vector<2xi32> |
| // CHECK: return %[[A]], %[[B]] : vector<4x8xf32>, vector<2xi32> |
| func.func @bitcast_folding(%I1: vector<4x8xf32>, %I2: vector<2xi32>) -> (vector<4x8xf32>, vector<2xi32>) { |
| %0 = vector.bitcast %I1 : vector<4x8xf32> to vector<4x8xf32> |
| %1 = vector.bitcast %I2 : vector<2xi32> to vector<4xi16> |
| %2 = vector.bitcast %1 : vector<4xi16> to vector<2xi32> |
| return %0, %2 : vector<4x8xf32>, vector<2xi32> |
| } |
| |
| // CHECK-LABEL: func @bitcast_f16_to_f32 |
| // bit pattern: 0x40004000 |
| // CHECK-DAG: %[[CST1:.+]] = arith.constant dense<2.00390625> : vector<4xf32> |
| // bit pattern: 0x00000000 |
| // CHECK-DAG: %[[CST0:.+]] = arith.constant dense<0.000000e+00> : vector<4xf32> |
| // CHECK: return %[[CST0]], %[[CST1]] |
| func.func @bitcast_f16_to_f32() -> (vector<4xf32>, vector<4xf32>) { |
| %cst0 = arith.constant dense<0.0> : vector<8xf16> // bit pattern: 0x0000 |
| %cst1 = arith.constant dense<2.0> : vector<8xf16> // bit pattern: 0x4000 |
| %cast0 = vector.bitcast %cst0: vector<8xf16> to vector<4xf32> |
| %cast1 = vector.bitcast %cst1: vector<8xf16> to vector<4xf32> |
| return %cast0, %cast1: vector<4xf32>, vector<4xf32> |
| } |
| |
| // CHECK-LABEL: func @bitcast_i8_to_i32 |
| // bit pattern: 0xA0A0A0A0 |
| // CHECK-DAG: %[[CST1:.+]] = arith.constant dense<-1600085856> : vector<4xi32> |
| // bit pattern: 0x00000000 |
| // CHECK-DAG: %[[CST0:.+]] = arith.constant dense<0> : vector<4xi32> |
| // CHECK: return %[[CST0]], %[[CST1]] |
| func.func @bitcast_i8_to_i32() -> (vector<4xi32>, vector<4xi32>) { |
| %cst0 = arith.constant dense<0> : vector<16xi8> // bit pattern: 0x00 |
| %cst1 = arith.constant dense<160> : vector<16xi8> // bit pattern: 0xA0 |
| %cast0 = vector.bitcast %cst0: vector<16xi8> to vector<4xi32> |
| %cast1 = vector.bitcast %cst1: vector<16xi8> to vector<4xi32> |
| return %cast0, %cast1: vector<4xi32>, vector<4xi32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: broadcast_folding1 |
| // CHECK: %[[CST:.*]] = arith.constant dense<42> : vector<4xi32> |
| // CHECK-NOT: vector.broadcast |
| // CHECK: return %[[CST]] |
| func.func @broadcast_folding1() -> vector<4xi32> { |
| %0 = arith.constant 42 : i32 |
| %1 = vector.broadcast %0 : i32 to vector<4xi32> |
| return %1 : vector<4xi32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: @broadcast_folding2 |
| // CHECK: %[[CST:.*]] = arith.constant dense<42> : vector<4x16xi32> |
| // CHECK-NOT: vector.broadcast |
| // CHECK: return %[[CST]] |
| func.func @broadcast_folding2() -> vector<4x16xi32> { |
| %0 = arith.constant 42 : i32 |
| %1 = vector.broadcast %0 : i32 to vector<16xi32> |
| %2 = vector.broadcast %1 : vector<16xi32> to vector<4x16xi32> |
| return %2 : vector<4x16xi32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: @fold_consecutive_broadcasts( |
| // CHECK-SAME: %[[ARG0:.*]]: i32 |
| // CHECK: %[[RESULT:.*]] = vector.broadcast %[[ARG0]] : i32 to vector<4x16xi32> |
| // CHECK: return %[[RESULT]] |
| func.func @fold_consecutive_broadcasts(%a : i32) -> vector<4x16xi32> { |
| %1 = vector.broadcast %a : i32 to vector<16xi32> |
| %2 = vector.broadcast %1 : vector<16xi32> to vector<4x16xi32> |
| return %2 : vector<4x16xi32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: shape_cast_constant |
| // CHECK-DAG: %[[CST1:.*]] = arith.constant dense<1> : vector<3x4x2xi32> |
| // CHECK-DAG: %[[CST0:.*]] = arith.constant dense<2.000000e+00> : vector<20x2xf32> |
| // CHECK: return %[[CST0]], %[[CST1]] : vector<20x2xf32>, vector<3x4x2xi32> |
| func.func @shape_cast_constant() -> (vector<20x2xf32>, vector<3x4x2xi32>) { |
| %cst = arith.constant dense<2.000000e+00> : vector<5x4x2xf32> |
| %cst_1 = arith.constant dense<1> : vector<12x2xi32> |
| %0 = vector.shape_cast %cst : vector<5x4x2xf32> to vector<20x2xf32> |
| %1 = vector.shape_cast %cst_1 : vector<12x2xi32> to vector<3x4x2xi32> |
| return %0, %1 : vector<20x2xf32>, vector<3x4x2xi32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: extract_strided_constant |
| // CHECK-DAG: %[[CST1:.*]] = arith.constant dense<1> : vector<2x13x3xi32> |
| // CHECK-DAG: %[[CST0:.*]] = arith.constant dense<2.000000e+00> : vector<12x2xf32> |
| // CHECK: return %[[CST0]], %[[CST1]] : vector<12x2xf32>, vector<2x13x3xi32> |
| func.func @extract_strided_constant() -> (vector<12x2xf32>, vector<2x13x3xi32>) { |
| %cst = arith.constant dense<2.000000e+00> : vector<29x7xf32> |
| %cst_1 = arith.constant dense<1> : vector<4x37x9xi32> |
| %0 = vector.extract_strided_slice %cst |
| {offsets = [2, 3], sizes = [12, 2], strides = [1, 1]} |
| : vector<29x7xf32> to vector<12x2xf32> |
| %1 = vector.extract_strided_slice %cst_1 |
| {offsets = [1, 2, 5], sizes = [2, 13, 3], strides = [1, 1, 1]} |
| : vector<4x37x9xi32> to vector<2x13x3xi32> |
| return %0, %1 : vector<12x2xf32>, vector<2x13x3xi32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: extract_strided_broadcast |
| // CHECK: %[[B:.*]] = vector.broadcast %{{.*}} : vector<4xf16> to vector<2x4xf16> |
| // CHECK-NEXT: return %[[B]] : vector<2x4xf16> |
| func.func @extract_strided_broadcast(%arg0: vector<4xf16>) -> vector<2x4xf16> { |
| %0 = vector.broadcast %arg0 : vector<4xf16> to vector<16x4xf16> |
| %1 = vector.extract_strided_slice %0 |
| {offsets = [0, 0], sizes = [2, 4], strides = [1, 1]} : |
| vector<16x4xf16> to vector<2x4xf16> |
| return %1 : vector<2x4xf16> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: extract_strided_broadcast2 |
| // CHECK: %[[E:.*]] = vector.extract_strided_slice %{{.*}} {offsets = [0], sizes = [2], strides = [1]} : vector<4xf16> to vector<2xf16> |
| // CHECK-NEXT: %[[B:.*]] = vector.broadcast %[[E]] : vector<2xf16> to vector<2x2xf16> |
| // CHECK-NEXT: return %[[B]] : vector<2x2xf16> |
| func.func @extract_strided_broadcast2(%arg0: vector<4xf16>) -> vector<2x2xf16> { |
| %0 = vector.broadcast %arg0 : vector<4xf16> to vector<16x4xf16> |
| %1 = vector.extract_strided_slice %0 |
| {offsets = [0, 0], sizes = [2, 2], strides = [1, 1]} : |
| vector<16x4xf16> to vector<2x2xf16> |
| return %1 : vector<2x2xf16> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func @extract_strided_broadcast3 |
| // CHECK-SAME: (%[[ARG:.+]]: vector<1xf32>) |
| // CHECK: %[[V:.+]] = vector.broadcast %[[ARG]] : vector<1xf32> to vector<1x4xf32> |
| // CHECK: return %[[V]] |
| func.func @extract_strided_broadcast3(%arg0: vector<1xf32>) -> vector<1x4xf32> { |
| %0 = vector.broadcast %arg0 : vector<1xf32> to vector<1x8xf32> |
| %1 = vector.extract_strided_slice %0 |
| {offsets = [0, 4], sizes = [1, 4], strides = [1, 1]} |
| : vector<1x8xf32> to vector<1x4xf32> |
| return %1 : vector<1x4xf32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func @extract_strided_broadcast4 |
| // CHECK-SAME: (%[[ARG:.+]]: f32) |
| // CHECK: %[[V:.+]] = vector.broadcast %[[ARG]] : f32 to vector<1x4xf32> |
| // CHECK: return %[[V]] |
| func.func @extract_strided_broadcast4(%arg0: f32) -> vector<1x4xf32> { |
| %0 = vector.broadcast %arg0 : f32 to vector<1x8xf32> |
| %1 = vector.extract_strided_slice %0 |
| {offsets = [0, 4], sizes = [1, 4], strides = [1, 1]} |
| : vector<1x8xf32> to vector<1x4xf32> |
| return %1 : vector<1x4xf32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: consecutive_shape_cast |
| // CHECK: %[[C:.*]] = vector.shape_cast %{{.*}} : vector<16xf16> to vector<4x4xf16> |
| // CHECK-NEXT: return %[[C]] : vector<4x4xf16> |
| func.func @consecutive_shape_cast(%arg0: vector<16xf16>) -> vector<4x4xf16> { |
| %0 = vector.shape_cast %arg0 : vector<16xf16> to vector<2x8xf16> |
| %1 = vector.shape_cast %0 : vector<2x8xf16> to vector<4x4xf16> |
| return %1 : vector<4x4xf16> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func @dead_transfer_op |
| // CHECK-NOT: vector.transfer_read |
| // CHECK-NOT: vector.transfer_write |
| // CHECK: return |
| func.func @dead_transfer_op(%arg0 : tensor<4x4xf32>, %arg1 : memref<4x4xf32>, |
| %v0 : vector<1x4xf32>) { |
| %c0 = arith.constant 0 : index |
| %cf0 = arith.constant 0.0 : f32 |
| %r = vector.transfer_read %arg1[%c0, %c0], %cf0 : |
| memref<4x4xf32>, vector<1x4xf32> |
| %w = vector.transfer_write %v0, %arg0[%c0, %c0] : |
| vector<1x4xf32>, tensor<4x4xf32> |
| return |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func @dead_load |
| // CHECK-NOT: vector.maskedload |
| // CHECK-NOT: vector.gather |
| // CHECK-NOT: vector.expandload |
| // CHECK: return |
| func.func @dead_load(%base: memref<?xf32>, %indices: vector<16xi32>, |
| %mask: vector<16xi1>, %passthru: vector<16xf32>) { |
| %c0 = arith.constant 0 : index |
| %0 = vector.maskedload %base[%c0], %mask, %passthru : |
| memref<?xf32>, vector<16xi1>, vector<16xf32> into vector<16xf32> |
| %1 = vector.gather %base[%c0][%indices], %mask, %passthru : |
| memref<?xf32>, vector<16xi32>, vector<16xi1>, vector<16xf32> into vector<16xf32> |
| %2 = vector.expandload %base[%c0], %mask, %passthru : |
| memref<?xf32>, vector<16xi1>, vector<16xf32> into vector<16xf32> |
| return |
| } |
| |
| // ----- |
| |
| #contraction_accesses0 = [ |
| affine_map<(i, j, k) -> (i, k)>, |
| affine_map<(i, j, k) -> (k, j)>, |
| affine_map<(i, j, k) -> (i, j)> |
| ] |
| #contraction_trait0 = { |
| indexing_maps = #contraction_accesses0, |
| iterator_types = ["parallel", "parallel", "reduction"] |
| } |
| |
| // CHECK-LABEL: func @contractions |
| // CHECK-SAME: %[[A:[0-9a-zA-Z]+]]: vector<2x3xf32> |
| // CHECK-SAME: %[[B:[0-9a-zA-Z]+]]: vector<3x4xf32> |
| // CHECK-SAME: %[[C:[0-9a-zA-Z]+]]: vector<2x4xf32> |
| // CHECK-SAME: %[[A_I8:[0-9a-zA-Z]+]]: vector<2x3xi8> |
| // CHECK-SAME: %[[B_I8:[0-9a-zA-Z]+]]: vector<3x4xi8> |
| // CHECK-SAME: %[[C_I8:[0-9a-zA-Z]+]]: vector<2x4xi8> |
| func.func @contractions(%a: vector<2x3xf32>, %b: vector<3x4xf32>, %c: vector<2x4xf32>, |
| %a_i8: vector<2x3xi8>, %b_i8: vector<3x4xi8>, %c_i8: vector<2x4xi8>) |
| -> (vector<2x4xf32>, vector<2x4xi8>) |
| { |
| // CHECK-NOT: arith.constant |
| %vf_0 = arith.constant dense <0.0>: vector<2x4xf32> |
| // CHECK-NOT: arith.addf |
| // CHECK: %[[D:.*]] = vector.contract {{.*}} %[[A]], %[[B]], %[[C]] |
| %0 = vector.contract #contraction_trait0 %a, %b, %vf_0: |
| vector<2x3xf32>, vector<3x4xf32> into vector<2x4xf32> |
| // CHECK-NOT: arith.addf |
| %1 = arith.addf %0, %c: vector<2x4xf32> |
| |
| // CHECK-NOT: arith.constant |
| %vi8_0 = arith.constant dense <0>: vector<2x4xi8> |
| // CHECK-NOT: arith.addi |
| // CHECK: %[[D_I8:.*]] = vector.contract {{.*}} %[[A_I8]], %[[B_I8]], %[[C_I8]] |
| %i8_0 = vector.contract #contraction_trait0 %a_i8, %b_i8, %vi8_0: |
| vector<2x3xi8>, vector<3x4xi8> into vector<2x4xi8> |
| // CHECK-NOT: arith.addi |
| %i8_1 = arith.addi %i8_0, %c_i8: vector<2x4xi8> |
| |
| // CHECK: return %[[D]], %[[D_I8]] |
| return %1, %i8_1: vector<2x4xf32>, vector<2x4xi8> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func @transfer_folding_1 |
| // CHECK-SAME: %[[T0:[0-9a-zA-Z]+]]: tensor<2x3x4xf32> |
| // CHECK-SAME: %[[T1:[0-9a-zA-Z]+]]: tensor<2x3x4xf32> |
| func.func @transfer_folding_1(%t0: tensor<2x3x4xf32>, %t1: tensor<2x3x4xf32>) |
| -> (tensor<2x3x4xf32>, tensor<2x3x4xf32>, tensor<2x3x4xf32>) |
| { |
| %c0 = arith.constant 0 : index |
| %pad = arith.constant 0.0 : f32 |
| %v = vector.transfer_read %t0[%c0, %c0, %c0], %pad {in_bounds = [true, true, true]} : |
| tensor<2x3x4xf32>, vector<2x3x4xf32> |
| |
| %r0 = vector.transfer_write %v, %t1[%c0, %c0, %c0] {in_bounds = [true, true, true]} : |
| vector<2x3x4xf32>, tensor<2x3x4xf32> |
| |
| %t2 = "test.constant"() { value = dense<6.0> : tensor<2x3x4xf32>} : () -> (tensor<2x3x4xf32>) |
| %r1 = vector.transfer_write %v, %t2[%c0, %c0, %c0] {in_bounds = [true, true, true]} : |
| vector<2x3x4xf32>, tensor<2x3x4xf32> |
| |
| |
| // CHECK-NEXT: some_op_that_may_have_side_effects |
| %t3 = "some_op_that_may_have_side_effects"() : () -> (tensor<2x3x4xf32>) |
| %r2 = vector.transfer_write %v, %t0[%c0, %c0, %c0] {in_bounds = [true, true, true]} : |
| vector<2x3x4xf32>, tensor<2x3x4xf32> |
| |
| // CHECK-NEXT: return %[[T0]], %[[T0]], %[[T0]] |
| return %r0, %r1, %r2: tensor<2x3x4xf32>, tensor<2x3x4xf32>, tensor<2x3x4xf32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func @store_after_load_tensor |
| // CHECK-SAME: (%[[ARG:.*]]: tensor<4x4xf32>) |
| // CHECK-NOT: vector.transfer_read |
| // CHECK-NOT: vector.transfer_write |
| // CHECK: return %[[ARG]] : tensor<4x4xf32> |
| func.func @store_after_load_tensor(%arg0 : tensor<4x4xf32>) -> tensor<4x4xf32> { |
| %c1 = arith.constant 1 : index |
| %c0 = arith.constant 0 : index |
| %cf0 = arith.constant 0.0 : f32 |
| %0 = vector.transfer_read %arg0[%c1, %c0], %cf0 : |
| tensor<4x4xf32>, vector<1x4xf32> |
| %w0 = vector.transfer_write %0, %arg0[%c1, %c0] : |
| vector<1x4xf32>, tensor<4x4xf32> |
| return %w0 : tensor<4x4xf32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func @store_after_load_tensor_negative |
| // CHECK: vector.transfer_read |
| // CHECK: vector.transfer_write |
| // CHECK: return |
| func.func @store_after_load_tensor_negative(%arg0 : tensor<4x4xf32>) -> tensor<4x4xf32> { |
| %c1 = arith.constant 1 : index |
| %c0 = arith.constant 0 : index |
| %cf0 = arith.constant 0.0 : f32 |
| %0 = vector.transfer_read %arg0[%c1, %c0], %cf0 : |
| tensor<4x4xf32>, vector<1x4xf32> |
| %w0 = vector.transfer_write %0, %arg0[%c0, %c0] : |
| vector<1x4xf32>, tensor<4x4xf32> |
| return %w0 : tensor<4x4xf32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func @store_to_load_tensor |
| // CHECK-SAME: (%[[ARG:.*]]: tensor<4x4xf32>, %[[V0:.*]]: vector<1x4xf32>, %[[V1:.*]]: vector<1x4xf32>) |
| // CHECK-NOT: vector.transfer_write |
| // CHECK-NOT: vector.transfer_read |
| // CHECK: return %[[V0]] : vector<1x4xf32> |
| func.func @store_to_load_tensor(%arg0 : tensor<4x4xf32>, |
| %v0 : vector<1x4xf32>, %v1 : vector<1x4xf32>) -> vector<1x4xf32> { |
| %c1 = arith.constant 1 : index |
| %c2 = arith.constant 2 : index |
| %c0 = arith.constant 0 : index |
| %cf0 = arith.constant 0.0 : f32 |
| %w0 = vector.transfer_write %v0, %arg0[%c1, %c0] {in_bounds = [true, true]} : |
| vector<1x4xf32>, tensor<4x4xf32> |
| %w1 = vector.transfer_write %v1, %w0[%c2, %c0] {in_bounds = [true, true]} : |
| vector<1x4xf32>, tensor<4x4xf32> |
| %0 = vector.transfer_read %w1[%c1, %c0], %cf0 {in_bounds = [true, true]} : |
| tensor<4x4xf32>, vector<1x4xf32> |
| return %0 : vector<1x4xf32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func @store_to_load_negative_tensor |
| // CHECK: vector.transfer_write |
| // CHECK: vector.transfer_write |
| // CHECK: %[[V:.*]] = vector.transfer_read |
| // CHECK: return %[[V]] : vector<1x4xf32> |
| func.func @store_to_load_negative_tensor(%arg0 : tensor<4x4xf32>, |
| %v0 : vector<1x4xf32>, %v1 : vector<1x4xf32>, %i : index) -> vector<1x4xf32> { |
| %c1 = arith.constant 1 : index |
| %c2 = arith.constant 2 : index |
| %c0 = arith.constant 0 : index |
| %cf0 = arith.constant 0.0 : f32 |
| %w0 = vector.transfer_write %v0, %arg0[%c1, %c0] {in_bounds = [true, true]} : |
| vector<1x4xf32>, tensor<4x4xf32> |
| %w1 = vector.transfer_write %v0, %w0[%i, %i] {in_bounds = [true, true]} : |
| vector<1x4xf32>, tensor<4x4xf32> |
| %0 = vector.transfer_read %w1[%c1, %c0], %cf0 {in_bounds = [true, true]} : |
| tensor<4x4xf32>, vector<1x4xf32> |
| return %0 : vector<1x4xf32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func @store_to_load_tensor_broadcast |
| // CHECK-SAME: (%[[ARG:.*]]: tensor<4x4xf32>, %[[V0:.*]]: vector<4x2xf32>) |
| // CHECK: %[[B:.*]] = vector.broadcast %[[V0]] : vector<4x2xf32> to vector<6x4x2xf32> |
| // CHECK: %[[T:.*]] = vector.transpose %[[B]], [1, 2, 0] : vector<6x4x2xf32> to vector<4x2x6xf32> |
| // CHECK: return %[[T]] : vector<4x2x6xf32> |
| func.func @store_to_load_tensor_broadcast(%arg0 : tensor<4x4xf32>, |
| %v0 : vector<4x2xf32>) -> vector<4x2x6xf32> { |
| %c0 = arith.constant 0 : index |
| %cf0 = arith.constant 0.0 : f32 |
| %w0 = vector.transfer_write %v0, %arg0[%c0, %c0] {in_bounds = [true, true]} : |
| vector<4x2xf32>, tensor<4x4xf32> |
| %0 = vector.transfer_read %w0[%c0, %c0], %cf0 {in_bounds = [true, true, true], |
| permutation_map = affine_map<(d0, d1) -> (d0, d1, 0)>} : |
| tensor<4x4xf32>, vector<4x2x6xf32> |
| return %0 : vector<4x2x6xf32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func @store_to_load_tensor_broadcast_scalable |
| // CHECK-SAME: (%[[ARG:.*]]: tensor<?xf32>, %[[V0:.*]]: vector<[4]xf32>) |
| // CHECK: %[[B:.*]] = vector.broadcast %[[V0]] : vector<[4]xf32> to vector<6x[4]xf32> |
| // CHECK: return %[[B]] : vector<6x[4]xf32> |
| func.func @store_to_load_tensor_broadcast_scalable(%arg0 : tensor<?xf32>, |
| %v0 : vector<[4]xf32>) -> vector<6x[4]xf32> { |
| %c0 = arith.constant 0 : index |
| %cf0 = arith.constant 0.0 : f32 |
| %w0 = vector.transfer_write %v0, %arg0[%c0] {in_bounds = [true]} : |
| vector<[4]xf32>, tensor<?xf32> |
| %0 = vector.transfer_read %w0[%c0], %cf0 {in_bounds = [true, true], |
| permutation_map = affine_map<(d0) -> (0, d0)>} : |
| tensor<?xf32>, vector<6x[4]xf32> |
| return %0 : vector<6x[4]xf32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func @store_to_load_tensor_perm_broadcast |
| // CHECK-SAME: (%[[ARG:.*]]: tensor<4x4x4xf32>, %[[V0:.*]]: vector<4x1xf32>) |
| // CHECK: %[[B:.*]] = vector.broadcast %[[V0]] : vector<4x1xf32> to vector<100x5x4x1xf32> |
| // CHECK: %[[T:.*]] = vector.transpose %[[B]], [3, 0, 2, 1] : vector<100x5x4x1xf32> to vector<1x100x4x5xf32> |
| // CHECK: return %[[T]] : vector<1x100x4x5xf32> |
| func.func @store_to_load_tensor_perm_broadcast(%arg0 : tensor<4x4x4xf32>, |
| %v0 : vector<4x1xf32>) -> vector<1x100x4x5xf32> { |
| %c0 = arith.constant 0 : index |
| %cf0 = arith.constant 0.0 : f32 |
| %w0 = vector.transfer_write %v0, %arg0[%c0, %c0, %c0] {in_bounds = [true, true], |
| permutation_map = affine_map<(d0, d1, d2) -> (d2, d1)>} : |
| vector<4x1xf32>, tensor<4x4x4xf32> |
| %0 = vector.transfer_read %w0[%c0, %c0, %c0], %cf0 {in_bounds = [true, true, true, true], |
| permutation_map = affine_map<(d0, d1, d2) -> (d1, 0, d2, 0)>} : |
| tensor<4x4x4xf32>, vector<1x100x4x5xf32> |
| return %0 : vector<1x100x4x5xf32> |
| } |
| |
| // ----- |
| |
| |
| // CHECK-LABEL: func @dead_store_tensor |
| // CHECK-DAG: %[[C0:.*]] = arith.constant 0 : index |
| // CHECK-DAG: %[[C1:.*]] = arith.constant 1 : index |
| // CHECK-DAG: %[[C2:.*]] = arith.constant 2 : index |
| // CHECK-NOT: vector.transfer_write {{.*}}, {{.*}}[%[[C1]], %[[C0]] |
| // CHECK: vector.transfer_write {{.*}}, {{.*}}[%[[C2]], %[[C0]] |
| // CHECK: %[[VTW:.*]] = vector.transfer_write {{.*}}, {{.*}}[%[[C1]], %[[C0]] |
| // CHECK: return %[[VTW]] : tensor<4x4xf32> |
| func.func @dead_store_tensor(%arg0 : tensor<4x4xf32>, |
| %v0 : vector<1x4xf32>, %v1 : vector<1x4xf32>, %i : index) -> tensor<4x4xf32> { |
| %c1 = arith.constant 1 : index |
| %c2 = arith.constant 2 : index |
| %c0 = arith.constant 0 : index |
| %cf0 = arith.constant 0.0 : f32 |
| %w0 = vector.transfer_write %v0, %arg0[%c1, %c0] {in_bounds = [true, true]} : |
| vector<1x4xf32>, tensor<4x4xf32> |
| %w1 = vector.transfer_write %v0, %w0[%c2, %c0] {in_bounds = [true, true]} : |
| vector<1x4xf32>, tensor<4x4xf32> |
| %w2 = vector.transfer_write %v1, %w1[%c1, %c0] {in_bounds = [true, true]} : |
| vector<1x4xf32>, tensor<4x4xf32> |
| return %w2 : tensor<4x4xf32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func @dead_store_tensor_negative |
| // CHECK-DAG: %[[C0:.*]] = arith.constant 0 : index |
| // CHECK-DAG: %[[C1:.*]] = arith.constant 1 : index |
| // CHECK: vector.transfer_write |
| // CHECK: vector.transfer_write |
| // CHECK: vector.transfer_read |
| // CHECK: %[[VTW:.*]] = vector.transfer_write {{.*}}, {{.*}}[%[[C1]], %[[C0]]] |
| // CHECK: return %[[VTW]] : tensor<4x4xf32> |
| func.func @dead_store_tensor_negative(%arg0 : tensor<4x4xf32>, |
| %v0 : vector<1x4xf32>, %v1 : vector<1x4xf32>, %i : index) -> tensor<4x4xf32> { |
| %c1 = arith.constant 1 : index |
| %c2 = arith.constant 2 : index |
| %c0 = arith.constant 0 : index |
| %cf0 = arith.constant 0.0 : f32 |
| %w0 = vector.transfer_write %v0, %arg0[%c1, %c0] {in_bounds = [true, true]} : |
| vector<1x4xf32>, tensor<4x4xf32> |
| %w1 = vector.transfer_write %v0, %w0[%c2, %c0] {in_bounds = [true, true]} : |
| vector<1x4xf32>, tensor<4x4xf32> |
| %0 = vector.transfer_read %w1[%i, %i], %cf0 {in_bounds = [true, true]} : |
| tensor<4x4xf32>, vector<1x4xf32> |
| %x = arith.addf %0, %0 : vector<1x4xf32> |
| %w2 = vector.transfer_write %x, %w0[%c1, %c0] {in_bounds = [true, true]} : |
| vector<1x4xf32>, tensor<4x4xf32> |
| return %w2 : tensor<4x4xf32> |
| } |
| |
| // ----- |
| |
| // CHECK: #[[$MAP:[0-9a-z]+]] = affine_map<(d0, d1) -> (d1, d0)> |
| |
| // CHECK-LABEL: func @swap_extract_slice_transfer_write |
| // CHECK-SAME: %[[VEC:.*]]: vector<8x4xf32> |
| // CHECK-SAME: %[[INIT_TENSOR:.*]]: tensor<4x8xf32>, |
| // CHECK-SAME: %[[ITER_ARG:.*]]: tensor<64x64xf32>, |
| // CHECK-SAME: %[[IV:.*]]: index, %[[SZ:.*]]: index) |
| func.func @swap_extract_slice_transfer_write(%arg0 : vector<8x4xf32>, |
| %arg1 : tensor<4x8xf32>, |
| %arg2 : tensor<64x64xf32>, |
| %iv : index, %sz : index) -> tensor<64x64xf32> { |
| // CHECK: %[[C0:.*]] = arith.constant 0 : index |
| %c0 = arith.constant 0 : index |
| |
| // CHECK: %[[T0:.*]] = tensor.extract_slice %[[ITER_ARG]] |
| // CHECK-SAME: [%[[IV]], 16] [%[[SZ]], 8] |
| // CHECK: %[[T1:.*]] = vector.transfer_write %[[VEC]] |
| // CHECK-SAME: %[[T0]][%[[C0]], %[[C0]]] |
| // CHECK-SAME: in_bounds = [true, false] |
| // CHECK-SAME: permutation_map = #[[$MAP]] |
| // CHECK: %[[T2:.*]] = tensor.insert_slice %[[T1]] into %[[ITER_ARG]] |
| // CHECK-SAME: [%[[IV]], 16] [%[[SZ]], 8] |
| %0 = vector.transfer_write %arg0, %arg1[%c0, %c0] {in_bounds = [true, true], permutation_map = affine_map<(d0, d1) -> (d1, d0)>} : vector<8x4xf32>, tensor<4x8xf32> |
| %1 = tensor.extract_slice %0[0, 0] [%sz, 8] [1, 1] : tensor<4x8xf32> to tensor<?x8xf32> |
| %2 = tensor.insert_slice %1 into %arg2[%iv, 16] [%sz, 8] [1, 1] : tensor<?x8xf32> into tensor<64x64xf32> |
| |
| // CHECK: return %[[T2]] |
| func.return %2 : tensor<64x64xf32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func @do_not_swap_extract_slice_transfer_write |
| // CHECK-SAME: %[[VEC:.*]]: vector<8xf32>, |
| // CHECK-SAME: %[[VEC_SMALL:.*]]: vector<4xf32>, |
| // CHECK-SAME: %[[INIT_TENSOR:.*]]: tensor<8xf32>, |
| // CHECK-SAME: %[[ITER_ARG:.*]]: tensor<64xf32>, |
| // CHECK-SAME: %[[IV:.*]]: index, %[[SZ:.*]]: index) |
| func.func @do_not_swap_extract_slice_transfer_write(%arg0 : vector<8xf32>, |
| %arg1 : vector<4xf32>, |
| %arg2 : tensor<8xf32>, |
| %arg3 : tensor<64xf32>, |
| %iv : index, %sz : index) -> (tensor<64xf32>, tensor<64xf32>, tensor<64xf32>) { |
| // CHECK: %[[C0:.*]] = arith.constant 0 : index |
| %c0 = arith.constant 0 : index |
| |
| // Don't swap if the extracted and inserted slices do not match. |
| // CHECK: %[[T0:.*]] = vector.transfer_write %[[VEC]] |
| // CHECK: %[[T1:.*]] = tensor.extract_slice %[[T0]] |
| // CHECK: %[[T2:.*]] = tensor.insert_slice %[[T1]] |
| %0 = vector.transfer_write %arg0, %arg2[%c0] {in_bounds = [true]} : vector<8xf32>, tensor<8xf32> |
| %1 = tensor.extract_slice %0[0] [%iv] [1] : tensor<8xf32> to tensor<?xf32> |
| %2 = tensor.insert_slice %1 into %arg3[%iv] [%sz] [1] : tensor<?xf32> into tensor<64xf32> |
| |
| // Don't swap if the TransferWriteOp takes a small vector. |
| // CHECK: %[[T3:.*]] = vector.transfer_write %[[VEC_SMALL]] |
| // CHECK: %[[T4:.*]] = tensor.extract_slice %[[T3]] |
| // CHECK: %[[T5:.*]] = tensor.insert_slice %[[T4]] |
| %3 = vector.transfer_write %arg1, %arg2[%c0] {in_bounds = [true]} : vector<4xf32>, tensor<8xf32> |
| %4 = tensor.extract_slice %3[0] [%sz] [1] : tensor<8xf32> to tensor<?xf32> |
| %5 = tensor.insert_slice %4 into %arg3[%iv] [%sz] [1] : tensor<?xf32> into tensor<64xf32> |
| |
| // Don't swap if the one of the operations is rank-reducing. |
| // CHECK: %[[T6:.*]] = vector.transfer_write %[[VEC]] |
| // CHECK: %[[T7:.*]] = tensor.extract_slice %[[T6]] |
| // CHECK: %[[T8:.*]] = tensor.insert_slice %[[T7]] |
| %6 = vector.transfer_write %arg0, %arg2[%c0] {in_bounds = [true]} : vector<8xf32>, tensor<8xf32> |
| %7 = tensor.extract_slice %6[0] [1] [1] : tensor<8xf32> to tensor<f32> |
| %8 = tensor.insert_slice %7 into %arg3[%iv] [1] [1] : tensor<f32> into tensor<64xf32> |
| |
| // CHECK: return %[[T2]], %[[T5]], %[[T8]] |
| func.return %2, %5, %8 : tensor<64xf32>, tensor<64xf32>, tensor<64xf32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func @vector_multi_reduction_single_parallel( |
| // CHECK-SAME: %[[v:.*]]: vector<2xf32>, |
| func.func @vector_multi_reduction_single_parallel(%arg0: vector<2xf32>, %acc: vector<2xf32>) -> vector<2xf32> { |
| %0 = vector.multi_reduction <mul>, %arg0, %acc [] : vector<2xf32> to vector<2xf32> |
| |
| // CHECK: return %[[v]] : vector<2xf32> |
| return %0 : vector<2xf32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func @masked_vector_multi_reduction_single_parallel( |
| // CHECK-SAME: %[[VAL_0:.*]]: vector<2xf32>, %{{.*}}: vector<2xf32>, |
| func.func @masked_vector_multi_reduction_single_parallel(%arg0: vector<2xf32>, %acc: vector<2xf32>, %mask: vector<2xi1>) -> vector<2xf32> { |
| %0 = vector.mask %mask { vector.multi_reduction <mul>, %arg0, %acc [] : vector<2xf32> to vector<2xf32> } : vector<2xi1> -> vector<2xf32> |
| // CHECK: return %[[VAL_0]] : vector<2xf32> |
| return %0 : vector<2xf32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func @vector_multi_reduction_unit_dimensions( |
| // CHECK-SAME: %[[SOURCE:.+]]: vector<5x1x4x1x20xf32>, %[[ACC:.+]]: vector<5x4x20xf32> |
| func.func @vector_multi_reduction_unit_dimensions(%source: vector<5x1x4x1x20xf32>, %acc: vector<5x4x20xf32>) -> vector<5x4x20xf32> { |
| // CHECK: %[[CAST:.+]] = vector.shape_cast %[[SOURCE]] : vector<5x1x4x1x20xf32> to vector<5x4x20xf32> |
| // CHECK: %[[RESULT:.+]] = arith.mulf %[[ACC]], %[[CAST]] : vector<5x4x20xf32> |
| %0 = vector.multi_reduction <mul>, %source, %acc [1, 3] : vector<5x1x4x1x20xf32> to vector<5x4x20xf32> |
| |
| // CHECK: return %[[RESULT]] : vector<5x4x20xf32> |
| return %0 : vector<5x4x20xf32> |
| } |
| |
| // ----- |
| // CHECK-LABEL: func.func @vector_multi_reduction_scalable( |
| // CHECK-SAME: %[[VAL_0:.*]]: vector<1x[4]x1xf32>, |
| // CHECK-SAME: %[[VAL_1:.*]]: vector<1x[4]xf32>, |
| // CHECK-SAME: %[[VAL_2:.*]]: vector<1x[4]x1xi1>) |
| func.func @vector_multi_reduction_scalable(%source: vector<1x[4]x1xf32>, |
| %acc: vector<1x[4]xf32>, |
| %mask: vector<1x[4]x1xi1>) -> vector<1x[4]xf32> { |
| // CHECK: %[[VAL_3:.*]] = vector.shape_cast %[[VAL_2]] : vector<1x[4]x1xi1> to vector<1x[4]xi1> |
| // CHECK: %[[VAL_4:.*]] = vector.shape_cast %[[VAL_0]] : vector<1x[4]x1xf32> to vector<1x[4]xf32> |
| // CHECK: %[[VAL_5:.*]] = arith.addf %[[VAL_1]], %[[VAL_4]] : vector<1x[4]xf32> |
| // CHECK: %[[VAL_6:.*]] = arith.select %[[VAL_3]], %[[VAL_5]], %[[VAL_4]] : vector<1x[4]xi1>, vector<1x[4]xf32> |
| %0 = vector.mask %mask { vector.multi_reduction <add>, %source, %acc [2] : vector<1x[4]x1xf32> to vector<1x[4]xf32> } : |
| vector<1x[4]x1xi1> -> vector<1x[4]xf32> |
| |
| return %0 : vector<1x[4]xf32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func @masked_vector_multi_reduction_unit_dimensions |
| // CHECK-SAME: %[[VAL_0:.*]]: vector<5x1x4x1x20xf32>, %[[VAL_1:.*]]: vector<5x4x20xf32>, |
| // CHECK-SAME: %[[VAL_2:.*]]: vector<5x1x4x1x20xi1>) |
| func.func @masked_vector_multi_reduction_unit_dimensions(%source: vector<5x1x4x1x20xf32>, |
| %acc: vector<5x4x20xf32>, |
| %mask: vector<5x1x4x1x20xi1>) -> vector<5x4x20xf32> { |
| // CHECK: %[[VAL_3:.*]] = vector.shape_cast %[[VAL_2]] : vector<5x1x4x1x20xi1> to vector<5x4x20xi1> |
| // CHECK: %[[VAL_4:.*]] = vector.shape_cast %[[VAL_0]] : vector<5x1x4x1x20xf32> to vector<5x4x20xf32> |
| // CHECK: %[[VAL_5:.*]] = arith.mulf %[[VAL_1]], %[[VAL_4]] : vector<5x4x20xf32> |
| // CHECK: %[[VAL_6:.*]] = arith.select %[[VAL_3]], %[[VAL_5]], %[[VAL_4]] : vector<5x4x20xi1>, vector<5x4x20xf32> |
| %0 = vector.mask %mask { vector.multi_reduction <mul>, %source, %acc [1, 3] : vector<5x1x4x1x20xf32> to vector<5x4x20xf32> } : |
| vector<5x1x4x1x20xi1> -> vector<5x4x20xf32> |
| return %0 : vector<5x4x20xf32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func @vector_multi_reduction_unit_dimensions_fail( |
| // CHECK-SAME: %[[SRC:.+]]: vector<5x1x4x1x20xf32>, %[[ACCUM:.+]]: vector<5x1x20xf32> |
| func.func @vector_multi_reduction_unit_dimensions_fail(%source: vector<5x1x4x1x20xf32>, %acc: vector<5x1x20xf32>) -> vector<5x1x20xf32> { |
| // CHECK: %[[RES:.+]] = vector.multi_reduction <mul>, %[[SRC]], %[[ACCUM]] [1, 2] : vector<5x1x4x1x20xf32> to vector<5x1x20xf32> |
| %0 = vector.multi_reduction <mul>, %source, %acc [1, 2] : vector<5x1x4x1x20xf32> to vector<5x1x20xf32> |
| |
| // CHECK: return %[[RES]] : vector<5x1x20xf32> |
| return %0 : vector<5x1x20xf32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func @vector_multi_reduction_unit_dimensions_single_elem( |
| // CHECK-SAME: %[[SOURCE:.+]]: vector<1x1x1xf32>, %[[ACC:.+]]: f32 |
| func.func @vector_multi_reduction_unit_dimensions_single_elem(%source: vector<1x1x1xf32>, %acc: f32) -> f32 { |
| // CHECK: %[[CAST:.+]] = vector.extract %[[SOURCE]][0, 0, 0] : f32 from vector<1x1x1xf32> |
| // CHECK: %[[RESULT:.+]] = arith.mulf %[[ACC]], %[[CAST]] : f32 |
| %0 = vector.multi_reduction <mul>, %source, %acc [0,1,2] : vector<1x1x1xf32> to f32 |
| |
| // CHECK: return %[[RESULT]] : f32 |
| return %0 : f32 |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func @masked_vector_multi_reduction_unit_dimensions_single_elem( |
| // CHECK-SAME: %[[VAL_0:.*]]: vector<1x1x1xf32>, %[[VAL_1:.*]]: f32, |
| // CHECK-SAME: %[[VAL_2:.*]]: vector<1x1x1xi1>) |
| func.func @masked_vector_multi_reduction_unit_dimensions_single_elem(%source: vector<1x1x1xf32>, %acc: f32, %mask: vector<1x1x1xi1>) -> f32 { |
| // CHECK: %[[VAL_3:.*]] = vector.extract %[[VAL_2]][0, 0, 0] : i1 from vector<1x1x1xi1> |
| // CHECK: %[[VAL_4:.*]] = vector.extract %[[VAL_0]][0, 0, 0] : f32 from vector<1x1x1xf32> |
| // CHECK: %[[VAL_5:.*]] = arith.mulf %[[VAL_1]], %[[VAL_4]] : f32 |
| // CHECK: %[[VAL_6:.*]] = arith.select %[[VAL_3]], %[[VAL_5]], %[[VAL_4]] : f32 |
| %0 = vector.mask %mask { vector.multi_reduction <mul>, %source, %acc [0,1,2] : vector<1x1x1xf32> to f32 } : vector<1x1x1xi1> -> f32 |
| return %0 : f32 |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func @insert_strided_slice_full_range |
| // CHECK-SAME: %[[SOURCE:.+]]: vector<16x16xf16>, %{{.+}}: vector<16x16xf16> |
| func.func @insert_strided_slice_full_range(%source: vector<16x16xf16>, %dest: vector<16x16xf16>) -> vector<16x16xf16> { |
| %0 = vector.insert_strided_slice %source, %dest {offsets = [0, 0], strides = [1, 1]} : vector<16x16xf16> into vector<16x16xf16> |
| // CHECK: return %[[SOURCE]] |
| return %0: vector<16x16xf16> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: extract_strided_splat |
| // CHECK: %[[B:.*]] = vector.splat %{{.*}} : vector<2x4xf16> |
| // CHECK-NEXT: return %[[B]] : vector<2x4xf16> |
| func.func @extract_strided_splat(%arg0: f16) -> vector<2x4xf16> { |
| %0 = vector.splat %arg0 : vector<16x4xf16> |
| %1 = vector.extract_strided_slice %0 |
| {offsets = [1, 0], sizes = [2, 4], strides = [1, 1]} : |
| vector<16x4xf16> to vector<2x4xf16> |
| return %1 : vector<2x4xf16> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func @insert_extract_to_broadcast |
| // CHECK-SAME: (%[[ARG0:.*]]: vector<1x1x4xf32>, %[[ARG1:.*]]: vector<4xf32>) |
| // CHECK: %[[V0:.*]] = vector.extract %[[ARG0]][0, 0] : vector<4xf32> from vector<1x1x4xf32> |
| // CHECK: %[[V1:.*]] = vector.broadcast %[[ARG1]] : vector<4xf32> to vector<1x1x4xf32> |
| // CHECK: return %[[V0]], %[[V1]] : vector<4xf32>, vector<1x1x4xf32> |
| func.func @insert_extract_to_broadcast(%arg0 : vector<1x1x4xf32>, |
| %arg1 : vector<4xf32>) -> (vector<4xf32>, vector<1x1x4xf32>) { |
| %0 = vector.extract %arg0[0, 0] : vector<4xf32> from vector<1x1x4xf32> |
| %1 = vector.insert %arg1, %arg0 [0, 0] : vector<4xf32> into vector<1x1x4xf32> |
| return %0, %1 : vector<4xf32>, vector<1x1x4xf32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func.func @extract_splat_constant |
| // CHECK-DAG: %[[CST1:.*]] = arith.constant 1 : i32 |
| // CHECK-DAG: %[[CST0:.*]] = arith.constant dense<2.000000e+00> : vector<7xf32> |
| // CHECK-NEXT: return %[[CST0]], %[[CST1]] : vector<7xf32>, i32 |
| func.func @extract_splat_constant() -> (vector<7xf32>, i32) { |
| %cst = arith.constant dense<2.000000e+00> : vector<29x7xf32> |
| %cst_1 = arith.constant dense<1> : vector<4x37x9xi32> |
| %0 = vector.extract %cst[2] : vector<7xf32> from vector<29x7xf32> |
| %1 = vector.extract %cst_1[1, 4, 5] : i32 from vector<4x37x9xi32> |
| return %0, %1 : vector<7xf32>, i32 |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func.func @extract_1d_constant |
| // CHECK-DAG: %[[I32CST:.*]] = arith.constant 3 : i32 |
| // CHECK-DAG: %[[IDXCST:.*]] = arith.constant 1 : index |
| // CHECK-DAG: %[[F32CST:.*]] = arith.constant 2.000000e+00 : f32 |
| // CHECK-NEXT: return %[[I32CST]], %[[IDXCST]], %[[F32CST]] : i32, index, f32 |
| func.func @extract_1d_constant() -> (i32, index, f32) { |
| %icst = arith.constant dense<[1, 2, 3, 4]> : vector<4xi32> |
| %e = vector.extract %icst[2] : i32 from vector<4xi32> |
| %idx_cst = arith.constant dense<[0, 1, 2]> : vector<3xindex> |
| %f = vector.extract %idx_cst[1] : index from vector<3xindex> |
| %fcst = arith.constant dense<[2.000000e+00, 3.000000e+00, 4.000000e+00]> : vector<3xf32> |
| %g = vector.extract %fcst[0] : f32 from vector<3xf32> |
| return %e, %f, %g : i32, index, f32 |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func.func @extract_2d_constant |
| // CHECK-DAG: %[[ACST:.*]] = arith.constant 0 : i32 |
| // CHECK-DAG: %[[BCST:.*]] = arith.constant 2 : i32 |
| // CHECK-DAG: %[[CCST:.*]] = arith.constant 3 : i32 |
| // CHECK-DAG: %[[DCST:.*]] = arith.constant 5 : i32 |
| // CHECK-NEXT: return %[[ACST]], %[[BCST]], %[[CCST]], %[[DCST]] : i32, i32, i32, i32 |
| func.func @extract_2d_constant() -> (i32, i32, i32, i32) { |
| %cst = arith.constant dense<[[0, 1, 2], [3, 4, 5]]> : vector<2x3xi32> |
| %a = vector.extract %cst[0, 0] : i32 from vector<2x3xi32> |
| %b = vector.extract %cst[0, 2] : i32 from vector<2x3xi32> |
| %c = vector.extract %cst[1, 0] : i32 from vector<2x3xi32> |
| %d = vector.extract %cst[1, 2] : i32 from vector<2x3xi32> |
| return %a, %b, %c, %d : i32, i32, i32, i32 |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func.func @extract_vector_2d_constant |
| // CHECK-DAG: %[[ACST:.*]] = arith.constant dense<[0, 1, 2]> : vector<3xi32> |
| // CHECK-DAG: %[[BCST:.*]] = arith.constant dense<[3, 4, 5]> : vector<3xi32> |
| // CHECK-NEXT: return %[[ACST]], %[[BCST]] : vector<3xi32>, vector<3xi32> |
| func.func @extract_vector_2d_constant() -> (vector<3xi32>, vector<3xi32>) { |
| %cst = arith.constant dense<[[0, 1, 2], [3, 4, 5]]> : vector<2x3xi32> |
| %a = vector.extract %cst[0] : vector<3xi32> from vector<2x3xi32> |
| %b = vector.extract %cst[1] : vector<3xi32> from vector<2x3xi32> |
| return %a, %b : vector<3xi32>, vector<3xi32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func.func @extract_3d_constant |
| // CHECK-DAG: %[[ACST:.*]] = arith.constant 0 : i32 |
| // CHECK-DAG: %[[BCST:.*]] = arith.constant 1 : i32 |
| // CHECK-DAG: %[[CCST:.*]] = arith.constant 9 : i32 |
| // CHECK-DAG: %[[DCST:.*]] = arith.constant 10 : i32 |
| // CHECK-NEXT: return %[[ACST]], %[[BCST]], %[[CCST]], %[[DCST]] : i32, i32, i32, i32 |
| func.func @extract_3d_constant() -> (i32, i32, i32, i32) { |
| %cst = arith.constant dense<[[[0, 1], [2, 3], [4, 5]], [[6, 7], [8, 9], [10, 11]]]> : vector<2x3x2xi32> |
| %a = vector.extract %cst[0, 0, 0] : i32 from vector<2x3x2xi32> |
| %b = vector.extract %cst[0, 0, 1] : i32 from vector<2x3x2xi32> |
| %c = vector.extract %cst[1, 1, 1] : i32 from vector<2x3x2xi32> |
| %d = vector.extract %cst[1, 2, 0] : i32 from vector<2x3x2xi32> |
| return %a, %b, %c, %d : i32, i32, i32, i32 |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func.func @extract_vector_3d_constant |
| // CHECK-DAG: %[[ACST:.*]] = arith.constant dense<{{\[\[0, 1\], \[2, 3\], \[4, 5\]\]}}> : vector<3x2xi32> |
| // CHECK-DAG: %[[BCST:.*]] = arith.constant dense<{{\[\[6, 7\], \[8, 9\], \[10, 11\]\]}}> : vector<3x2xi32> |
| // CHECK-DAG: %[[CCST:.*]] = arith.constant dense<[8, 9]> : vector<2xi32> |
| // CHECK-DAG: %[[DCST:.*]] = arith.constant dense<[10, 11]> : vector<2xi32> |
| // CHECK-NEXT: return %[[ACST]], %[[BCST]], %[[CCST]], %[[DCST]] : vector<3x2xi32>, vector<3x2xi32>, vector<2xi32>, vector<2xi32> |
| func.func @extract_vector_3d_constant() -> (vector<3x2xi32>, vector<3x2xi32>, vector<2xi32>, vector<2xi32>) { |
| %cst = arith.constant dense<[[[0, 1], [2, 3], [4, 5]], [[6, 7], [8, 9], [10, 11]]]> : vector<2x3x2xi32> |
| %a = vector.extract %cst[0] : vector<3x2xi32> from vector<2x3x2xi32> |
| %b = vector.extract %cst[1] : vector<3x2xi32> from vector<2x3x2xi32> |
| %c = vector.extract %cst[1, 1] : vector<2xi32> from vector<2x3x2xi32> |
| %d = vector.extract %cst[1, 2] : vector<2xi32> from vector<2x3x2xi32> |
| return %a, %b, %c, %d : vector<3x2xi32>, vector<3x2xi32>, vector<2xi32>, vector<2xi32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func.func @extract_splat_vector_3d_constant |
| // CHECK-DAG: %[[ACST:.*]] = arith.constant dense<0> : vector<2xi32> |
| // CHECK-DAG: %[[BCST:.*]] = arith.constant dense<4> : vector<2xi32> |
| // CHECK-DAG: %[[CCST:.*]] = arith.constant dense<5> : vector<2xi32> |
| // CHECK-NEXT: return %[[ACST]], %[[BCST]], %[[CCST]] : vector<2xi32>, vector<2xi32>, vector<2xi32> |
| func.func @extract_splat_vector_3d_constant() -> (vector<2xi32>, vector<2xi32>, vector<2xi32>) { |
| %cst = arith.constant dense<[[[0, 0], [1, 1], [2, 2]], [[3, 3], [4, 4], [5, 5]]]> : vector<2x3x2xi32> |
| %a = vector.extract %cst[0, 0] : vector<2xi32> from vector<2x3x2xi32> |
| %b = vector.extract %cst[1, 1] : vector<2xi32> from vector<2x3x2xi32> |
| %c = vector.extract %cst[1, 2] : vector<2xi32> from vector<2x3x2xi32> |
| return %a, %b, %c : vector<2xi32>, vector<2xi32>, vector<2xi32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func.func @extract_strided_slice_1d_constant |
| // CHECK-DAG: %[[ACST:.*]] = arith.constant dense<[0, 1, 2]> : vector<3xi32> |
| // CHECK-DAG: %[[BCST:.*]] = arith.constant dense<[1, 2]> : vector<2xi32> |
| // CHECK-DAG: %[[CCST:.*]] = arith.constant dense<2> : vector<1xi32> |
| // CHECK-NEXT: return %[[ACST]], %[[BCST]], %[[CCST]] : vector<3xi32>, vector<2xi32>, vector<1xi32> |
| func.func @extract_strided_slice_1d_constant() -> (vector<3xi32>, vector<2xi32>, vector<1xi32>) { |
| %cst = arith.constant dense<[0, 1, 2]> : vector<3xi32> |
| %a = vector.extract_strided_slice %cst |
| {offsets = [0], sizes = [3], strides = [1]} : vector<3xi32> to vector<3xi32> |
| %b = vector.extract_strided_slice %cst |
| {offsets = [1], sizes = [2], strides = [1]} : vector<3xi32> to vector<2xi32> |
| %c = vector.extract_strided_slice %cst |
| {offsets = [2], sizes = [1], strides = [1]} : vector<3xi32> to vector<1xi32> |
| return %a, %b, %c : vector<3xi32>, vector<2xi32>, vector<1xi32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func.func @extract_strided_slice_2d_constant |
| // CHECK-DAG: %[[ACST:.*]] = arith.constant dense<0> : vector<1x1xi32> |
| // CHECK-DAG: %[[BCST:.*]] = arith.constant dense<{{\[\[4, 5\]\]}}> : vector<1x2xi32> |
| // CHECK-DAG: %[[CCST:.*]] = arith.constant dense<{{\[\[1, 2\], \[4, 5\]\]}}> : vector<2x2xi32> |
| // CHECK-NEXT: return %[[ACST]], %[[BCST]], %[[CCST]] : vector<1x1xi32>, vector<1x2xi32>, vector<2x2xi32> |
| func.func @extract_strided_slice_2d_constant() -> (vector<1x1xi32>, vector<1x2xi32>, vector<2x2xi32>) { |
| %cst = arith.constant dense<[[0, 1, 2], [3, 4, 5]]> : vector<2x3xi32> |
| %a = vector.extract_strided_slice %cst |
| {offsets = [0, 0], sizes = [1, 1], strides = [1, 1]} : vector<2x3xi32> to vector<1x1xi32> |
| %b = vector.extract_strided_slice %cst |
| {offsets = [1, 1], sizes = [1, 2], strides = [1, 1]} : vector<2x3xi32> to vector<1x2xi32> |
| %c = vector.extract_strided_slice %cst |
| {offsets = [0, 1], sizes = [2, 2], strides = [1, 1]} : vector<2x3xi32> to vector<2x2xi32> |
| return %a, %b, %c : vector<1x1xi32>, vector<1x2xi32>, vector<2x2xi32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func.func @extract_strided_slice_3d_constant |
| // CHECK-DAG: %[[ACST:.*]] = arith.constant dense<{{\[\[\[8, 9\], \[10, 11\]\]\]}}> : vector<1x2x2xi32> |
| // CHECK-DAG: %[[BCST:.*]] = arith.constant dense<{{\[\[\[2, 3\]\]\]}}> : vector<1x1x2xi32> |
| // CHECK-DAG: %[[CCST:.*]] = arith.constant dense<{{\[\[\[6, 7\]\], \[\[10, 11\]\]\]}}> : vector<2x1x2xi32> |
| // CHECK-DAG: %[[DCST:.*]] = arith.constant dense<11> : vector<1x1x1xi32> |
| // CHECK-NEXT: return %[[ACST]], %[[BCST]], %[[CCST]], %[[DCST]] |
| func.func @extract_strided_slice_3d_constant() -> (vector<1x2x2xi32>, vector<1x1x2xi32>, vector<2x1x2xi32>, vector<1x1x1xi32>) { |
| %cst = arith.constant dense<[[[0, 1], [2, 3]], [[4, 5], [6, 7]], [[8, 9], [10, 11]]]> : vector<3x2x2xi32> |
| %a = vector.extract_strided_slice %cst |
| {offsets = [2], sizes = [1], strides = [1]} : vector<3x2x2xi32> to vector<1x2x2xi32> |
| %b = vector.extract_strided_slice %cst |
| {offsets = [0, 1], sizes = [1, 1], strides = [1, 1]} : vector<3x2x2xi32> to vector<1x1x2xi32> |
| %c = vector.extract_strided_slice %cst |
| {offsets = [1, 1, 0], sizes = [2, 1, 2], strides = [1, 1, 1]} : vector<3x2x2xi32> to vector<2x1x2xi32> |
| %d = vector.extract_strided_slice %cst |
| {offsets = [2, 1, 1], sizes = [1, 1, 1], strides = [1, 1, 1]} : vector<3x2x2xi32> to vector<1x1x1xi32> |
| return %a, %b, %c, %d : vector<1x2x2xi32>, vector<1x1x2xi32>, vector<2x1x2xi32>, vector<1x1x1xi32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: extract_extract_strided |
| // CHECK-SAME: %[[A:.*]]: vector<32x16x4xf16> |
| // CHECK: %[[V:.*]] = vector.extract %[[A]][9, 7] : vector<4xf16> from vector<32x16x4xf16> |
| // CHECK: return %[[V]] : vector<4xf16> |
| func.func @extract_extract_strided(%arg0: vector<32x16x4xf16>) -> vector<4xf16> { |
| %1 = vector.extract_strided_slice %arg0 |
| {offsets = [7, 3], sizes = [10, 8], strides = [1, 1]} : |
| vector<32x16x4xf16> to vector<10x8x4xf16> |
| %2 = vector.extract %1[2, 4] : vector<4xf16> from vector<10x8x4xf16> |
| return %2 : vector<4xf16> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: extract_insert_strided |
| // CHECK-SAME: %[[A:.*]]: vector<6x4xf32> |
| // CHECK: %[[V:.*]] = vector.extract %[[A]][0, 2] : f32 from vector<6x4xf32> |
| // CHECK: return %[[V]] : f32 |
| func.func @extract_insert_strided(%a: vector<6x4xf32>, %b: vector<8x16xf32>) |
| -> f32 { |
| %0 = vector.insert_strided_slice %a, %b {offsets = [2, 2], strides = [1, 1]} |
| : vector<6x4xf32> into vector<8x16xf32> |
| %2 = vector.extract %0[2, 4] : f32 from vector<8x16xf32> |
| return %2 : f32 |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: extract_insert_rank_reduce |
| // CHECK-SAME: %[[A:.*]]: vector<4xf32> |
| // CHECK: %[[V:.*]] = vector.extract %[[A]][2] : f32 from vector<4xf32> |
| // CHECK: return %[[V]] : f32 |
| func.func @extract_insert_rank_reduce(%a: vector<4xf32>, %b: vector<8x16xf32>) |
| -> f32 { |
| %0 = vector.insert_strided_slice %a, %b {offsets = [2, 2], strides = [1]} |
| : vector<4xf32> into vector<8x16xf32> |
| %2 = vector.extract %0[2, 4] : f32 from vector<8x16xf32> |
| return %2 : f32 |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: extract_insert_negative |
| // CHECK: vector.insert_strided_slice |
| // CHECK: vector.extract |
| func.func @extract_insert_negative(%a: vector<2x15xf32>, %b: vector<12x8x16xf32>) |
| -> vector<16xf32> { |
| %0 = vector.insert_strided_slice %a, %b {offsets = [4, 2, 0], strides = [1, 1]} |
| : vector<2x15xf32> into vector<12x8x16xf32> |
| %2 = vector.extract %0[4, 2] : vector<16xf32> from vector<12x8x16xf32> |
| return %2 : vector<16xf32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: extract_insert_chain |
| // CHECK-SAME: (%[[A:.*]]: vector<2x16xf32>, %[[B:.*]]: vector<12x8x16xf32>, %[[C:.*]]: vector<2x16xf32>) |
| // CHECK: %[[V:.*]] = vector.extract %[[C]][0] : vector<16xf32> from vector<2x16xf32> |
| // CHECK: return %[[V]] : vector<16xf32> |
| func.func @extract_insert_chain(%a: vector<2x16xf32>, %b: vector<12x8x16xf32>, %c: vector<2x16xf32>) |
| -> vector<16xf32> { |
| %0 = vector.insert_strided_slice %c, %b {offsets = [4, 2, 0], strides = [1, 1]} |
| : vector<2x16xf32> into vector<12x8x16xf32> |
| %1 = vector.insert_strided_slice %a, %0 {offsets = [0, 2, 0], strides = [1, 1]} |
| : vector<2x16xf32> into vector<12x8x16xf32> |
| %2 = vector.extract %1[4, 2] : vector<16xf32> from vector<12x8x16xf32> |
| return %2 : vector<16xf32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: extract_from_extract_chain_should_not_fold_dynamic_extracts |
| // CHECK-SAME: (%[[VEC:.*]]: vector<2x4xf32>, %[[IDX:.*]]: index) |
| // CHECK: %[[A:.*]] = vector.extract %[[VEC]][%[[IDX]]] : vector<4xf32> from vector<2x4xf32> |
| // CHECK: %[[B:.*]] = vector.extract %[[A]][1] : f32 from vector<4xf32> |
| func.func @extract_from_extract_chain_should_not_fold_dynamic_extracts(%v: vector<2x4xf32>, %index: index) -> f32 { |
| %0 = vector.extract %v[%index] : vector<4xf32> from vector<2x4xf32> |
| %1 = vector.extract %0[1] : f32 from vector<4xf32> |
| return %1 : f32 |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: extract_extract_strided2 |
| // CHECK-SAME: %[[A:.*]]: vector<2x4xf32> |
| // CHECK: %[[V:.*]] = vector.extract %[[A]][1] : vector<4xf32> from vector<2x4xf32> |
| // CHECK: return %[[V]] : vector<4xf32> |
| func.func @extract_extract_strided2(%A: vector<2x4xf32>) |
| -> (vector<4xf32>) { |
| %0 = vector.extract_strided_slice %A {offsets = [1, 0], sizes = [1, 4], strides = [1, 1]} : vector<2x4xf32> to vector<1x4xf32> |
| %1 = vector.extract %0[0] : vector<4xf32> from vector<1x4xf32> |
| return %1 : vector<4xf32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func @splat_fold |
| func.func @splat_fold() -> vector<4xf32> { |
| %c = arith.constant 1.0 : f32 |
| %v = vector.splat %c : vector<4xf32> |
| return %v : vector<4xf32> |
| |
| // CHECK-NEXT: [[V:%.*]] = arith.constant dense<1.000000e+00> : vector<4xf32> |
| // CHECK-NEXT: return [[V]] : vector<4xf32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func @shuffle_1d |
| // CHECK: %[[V:.+]] = arith.constant dense<[3, 2, 5, 1]> : vector<4xi32> |
| // CHECK: return %[[V]] |
| func.func @shuffle_1d() -> vector<4xi32> { |
| %v0 = arith.constant dense<[0, 1, 2]> : vector<3xi32> |
| %v1 = arith.constant dense<[3, 4, 5]> : vector<3xi32> |
| %shuffle = vector.shuffle %v0, %v1 [3, 2, 5, 1] : vector<3xi32>, vector<3xi32> |
| return %shuffle : vector<4xi32> |
| } |
| |
| // CHECK-LABEL: func @shuffle_canonicalize_0d |
| func.func @shuffle_canonicalize_0d(%v0 : vector<i32>, %v1 : vector<i32>) -> vector<1xi32> { |
| // CHECK: vector.broadcast %{{.*}} : vector<i32> to vector<1xi32> |
| %shuffle = vector.shuffle %v0, %v1 [0] : vector<i32>, vector<i32> |
| return %shuffle : vector<1xi32> |
| } |
| |
| // CHECK-LABEL: func @shuffle_fold1 |
| // CHECK: %arg0 : vector<4xi32> |
| func.func @shuffle_fold1(%v0 : vector<4xi32>, %v1 : vector<2xi32>) -> vector<4xi32> { |
| %shuffle = vector.shuffle %v0, %v1 [0, 1, 2, 3] : vector<4xi32>, vector<2xi32> |
| return %shuffle : vector<4xi32> |
| } |
| |
| // CHECK-LABEL: func @shuffle_fold2 |
| // CHECK: %arg1 : vector<2xi32> |
| func.func @shuffle_fold2(%v0 : vector<4xi32>, %v1 : vector<2xi32>) -> vector<2xi32> { |
| %shuffle = vector.shuffle %v0, %v1 [4, 5] : vector<4xi32>, vector<2xi32> |
| return %shuffle : vector<2xi32> |
| } |
| |
| // CHECK-LABEL: func @shuffle_fold3 |
| // CHECK: return %arg0 : vector<4x5x6xi32> |
| func.func @shuffle_fold3(%v0 : vector<4x5x6xi32>, %v1 : vector<2x5x6xi32>) -> vector<4x5x6xi32> { |
| %shuffle = vector.shuffle %v0, %v1 [0, 1, 2, 3] : vector<4x5x6xi32>, vector<2x5x6xi32> |
| return %shuffle : vector<4x5x6xi32> |
| } |
| |
| // CHECK-LABEL: func @shuffle_fold4 |
| // CHECK: return %arg1 : vector<2x5x6xi32> |
| func.func @shuffle_fold4(%v0 : vector<4x5x6xi32>, %v1 : vector<2x5x6xi32>) -> vector<2x5x6xi32> { |
| %shuffle = vector.shuffle %v0, %v1 [4, 5] : vector<4x5x6xi32>, vector<2x5x6xi32> |
| return %shuffle : vector<2x5x6xi32> |
| } |
| |
| // CHECK-LABEL: func @shuffle_nofold1 |
| // CHECK: %[[V:.+]] = vector.shuffle %arg0, %arg1 [0, 1, 2, 3, 4] : vector<4xi32>, vector<2xi32> |
| // CHECK: return %[[V]] |
| func.func @shuffle_nofold1(%v0 : vector<4xi32>, %v1 : vector<2xi32>) -> vector<5xi32> { |
| %shuffle = vector.shuffle %v0, %v1 [0, 1, 2, 3, 4] : vector<4xi32>, vector<2xi32> |
| return %shuffle : vector<5xi32> |
| } |
| |
| // CHECK-LABEL: func @shuffle_nofold2 |
| // CHECK: %[[V:.+]] = vector.shuffle %arg0, %arg1 [0, 1, 2, 3] : vector<[4]xi32>, vector<[2]xi32> |
| // CHECK: return %[[V]] |
| func.func @shuffle_nofold2(%v0 : vector<[4]xi32>, %v1 : vector<[2]xi32>) -> vector<4xi32> { |
| %shuffle = vector.shuffle %v0, %v1 [0, 1, 2, 3] : vector<[4]xi32>, vector<[2]xi32> |
| return %shuffle : vector<4xi32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func @transpose_scalar_broadcast1 |
| // CHECK-SAME: (%[[ARG:.+]]: vector<1xf32>) |
| // CHECK: %[[V:.+]] = vector.broadcast %[[ARG]] : vector<1xf32> to vector<1x8xf32> |
| // CHECK: return %[[V]] : vector<1x8xf32> |
| func.func @transpose_scalar_broadcast1(%value: vector<1xf32>) -> vector<1x8xf32> { |
| %bcast = vector.broadcast %value : vector<1xf32> to vector<8x1xf32> |
| %t = vector.transpose %bcast, [1, 0] : vector<8x1xf32> to vector<1x8xf32> |
| return %t : vector<1x8xf32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func @transpose_scalar_broadcast2 |
| // CHECK-SAME: (%[[ARG:.+]]: f32) |
| // CHECK: %[[V:.+]] = vector.broadcast %[[ARG]] : f32 to vector<1x8xf32> |
| // CHECK: return %[[V]] : vector<1x8xf32> |
| func.func @transpose_scalar_broadcast2(%value: f32) -> vector<1x8xf32> { |
| %bcast = vector.broadcast %value : f32 to vector<8x1xf32> |
| %t = vector.transpose %bcast, [1, 0] : vector<8x1xf32> to vector<1x8xf32> |
| return %t : vector<1x8xf32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func @transpose_splat_constant |
| // CHECK: %[[CST:.+]] = arith.constant dense<5.000000e+00> : vector<8x4xf32> |
| // CHECK: return %[[CST]] |
| func.func @transpose_splat_constant() -> vector<8x4xf32> { |
| %cst = arith.constant dense<5.0> : vector<4x8xf32> |
| %0 = vector.transpose %cst, [1, 0] : vector<4x8xf32> to vector<8x4xf32> |
| return %0 : vector<8x4xf32> |
| } |
| |
| // CHECK-LABEL: func @transpose_splat2( |
| // CHECK-SAME: %[[VAL_0:.*]]: f32) -> vector<3x4xf32> { |
| // CHECK: %[[VAL_1:.*]] = vector.splat %[[VAL_0]] : vector<3x4xf32> |
| // CHECK: return %[[VAL_1]] : vector<3x4xf32> |
| // CHECK: } |
| func.func @transpose_splat2(%arg : f32) -> vector<3x4xf32> { |
| %splat = vector.splat %arg : vector<4x3xf32> |
| %0 = vector.transpose %splat, [1, 0] : vector<4x3xf32> to vector<3x4xf32> |
| return %0 : vector<3x4xf32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func.func @insert_1d_constant |
| // CHECK-DAG: %[[ACST:.*]] = arith.constant dense<[9, 1, 2]> : vector<3xi32> |
| // CHECK-DAG: %[[BCST:.*]] = arith.constant dense<[0, 9, 2]> : vector<3xi32> |
| // CHECK-DAG: %[[CCST:.*]] = arith.constant dense<[0, 1, 9]> : vector<3xi32> |
| // CHECK-NEXT: return %[[ACST]], %[[BCST]], %[[CCST]] : vector<3xi32>, vector<3xi32>, vector<3xi32> |
| func.func @insert_1d_constant() -> (vector<3xi32>, vector<3xi32>, vector<3xi32>) { |
| %vcst = arith.constant dense<[0, 1, 2]> : vector<3xi32> |
| %icst = arith.constant 9 : i32 |
| %a = vector.insert %icst, %vcst[0] : i32 into vector<3xi32> |
| %b = vector.insert %icst, %vcst[1] : i32 into vector<3xi32> |
| %c = vector.insert %icst, %vcst[2] : i32 into vector<3xi32> |
| return %a, %b, %c : vector<3xi32>, vector<3xi32>, vector<3xi32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func.func @insert_2d_constant |
| // CHECK-DAG: %[[ACST:.*]] = arith.constant dense<{{\[\[99, 1, 2\], \[3, 4, 5\]\]}}> : vector<2x3xi32> |
| // CHECK-DAG: %[[BCST:.*]] = arith.constant dense<{{\[\[0, 1, 2\], \[3, 4, 99\]\]}}> : vector<2x3xi32> |
| // CHECK-DAG: %[[CCST:.*]] = arith.constant dense<{{\[\[90, 91, 92\], \[3, 4, 5\]\]}}> : vector<2x3xi32> |
| // CHECK-DAG: %[[DCST:.*]] = arith.constant dense<{{\[\[0, 1, 2\], \[90, 91, 92\]\]}}> : vector<2x3xi32> |
| // CHECK-NEXT: return %[[ACST]], %[[BCST]], %[[CCST]], %[[DCST]] |
| func.func @insert_2d_constant() -> (vector<2x3xi32>, vector<2x3xi32>, vector<2x3xi32>, vector<2x3xi32>) { |
| %vcst = arith.constant dense<[[0, 1, 2], [3, 4, 5]]> : vector<2x3xi32> |
| %cst_scalar = arith.constant 99 : i32 |
| %cst_1d = arith.constant dense<[90, 91, 92]> : vector<3xi32> |
| %a = vector.insert %cst_scalar, %vcst[0, 0] : i32 into vector<2x3xi32> |
| %b = vector.insert %cst_scalar, %vcst[1, 2] : i32 into vector<2x3xi32> |
| %c = vector.insert %cst_1d, %vcst[0] : vector<3xi32> into vector<2x3xi32> |
| %d = vector.insert %cst_1d, %vcst[1] : vector<3xi32> into vector<2x3xi32> |
| return %a, %b, %c, %d : vector<2x3xi32>, vector<2x3xi32>, vector<2x3xi32>, vector<2x3xi32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func.func @insert_2d_splat_constant |
| // CHECK-DAG: %[[ACST:.*]] = arith.constant dense<0> : vector<2x3xi32> |
| // CHECK-DAG: %[[BCST:.*]] = arith.constant dense<{{\[\[99, 0, 0\], \[0, 0, 0\]\]}}> : vector<2x3xi32> |
| // CHECK-DAG: %[[CCST:.*]] = arith.constant dense<{{\[\[0, 0, 0\], \[0, 99, 0\]\]}}> : vector<2x3xi32> |
| // CHECK-DAG: %[[DCST:.*]] = arith.constant dense<{{\[\[33, 33, 33\], \[0, 0, 0\]\]}}> : vector<2x3xi32> |
| // CHECK-DAG: %[[ECST:.*]] = arith.constant dense<{{\[\[0, 0, 0\], \[33, 33, 33\]\]}}> : vector<2x3xi32> |
| // CHECK-NEXT: return %[[ACST]], %[[BCST]], %[[CCST]], %[[DCST]], %[[ECST]] |
| func.func @insert_2d_splat_constant() |
| -> (vector<2x3xi32>, vector<2x3xi32>, vector<2x3xi32>, vector<2x3xi32>, vector<2x3xi32>) { |
| %vcst = arith.constant dense<0> : vector<2x3xi32> |
| %cst_zero = arith.constant 0 : i32 |
| %cst_scalar = arith.constant 99 : i32 |
| %cst_1d = arith.constant dense<33> : vector<3xi32> |
| %a = vector.insert %cst_zero, %vcst[0, 0] : i32 into vector<2x3xi32> |
| %b = vector.insert %cst_scalar, %vcst[0, 0] : i32 into vector<2x3xi32> |
| %c = vector.insert %cst_scalar, %vcst[1, 1] : i32 into vector<2x3xi32> |
| %d = vector.insert %cst_1d, %vcst[0] : vector<3xi32> into vector<2x3xi32> |
| %e = vector.insert %cst_1d, %vcst[1] : vector<3xi32> into vector<2x3xi32> |
| return %a, %b, %c, %d, %e : vector<2x3xi32>, vector<2x3xi32>, vector<2x3xi32>, vector<2x3xi32>, vector<2x3xi32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func @insert_element_fold |
| // CHECK: %[[V:.+]] = arith.constant dense<[0, 1, 7, 3]> : vector<4xi32> |
| // CHECK: return %[[V]] |
| func.func @insert_element_fold() -> vector<4xi32> { |
| %v = arith.constant dense<[0, 1, 2, 3]> : vector<4xi32> |
| %s = arith.constant 7 : i32 |
| %i = arith.constant 2 : i32 |
| %1 = vector.insertelement %s, %v[%i : i32] : vector<4xi32> |
| return %1 : vector<4xi32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func @insert_element_invalid_fold |
| func.func @insert_element_invalid_fold() -> vector<1xf32> { |
| // Out-of-bound index here. |
| %c26 = arith.constant 26 : index |
| %cst_2 = arith.constant 1.60215309E+9 : f32 |
| %cst_20 = arith.constant dense<1.60215309E+9> : vector<1xf32> |
| // CHECK: vector.insertelement |
| %46 = vector.insertelement %cst_2, %cst_20[%c26 : index] : vector<1xf32> |
| return %46 : vector<1xf32> |
| } |
| |
| |
| // ----- |
| |
| // Do not crash on poison |
| // CHECK-LABEL: func @insert_poison_fold1 |
| // CHECK: vector.insertelement |
| func.func @insert_poison_fold1() -> vector<4xi32> { |
| %v = ub.poison : vector<4xi32> |
| %s = arith.constant 7 : i32 |
| %i = arith.constant 2 : i32 |
| %1 = vector.insertelement %s, %v[%i : i32] : vector<4xi32> |
| return %1 : vector<4xi32> |
| } |
| |
| // ----- |
| |
| // Do not crash on poison |
| // CHECK-LABEL: func @insert_poison_fold2 |
| // CHECK: vector.insertelement |
| func.func @insert_poison_fold2() -> vector<4xi32> { |
| %v = arith.constant dense<[0, 1, 2, 3]> : vector<4xi32> |
| %s = ub.poison : i32 |
| %i = arith.constant 2 : i32 |
| %1 = vector.insertelement %s, %v[%i : i32] : vector<4xi32> |
| return %1 : vector<4xi32> |
| } |
| |
| // ----- |
| |
| // Do not crash on poison |
| // CHECK-LABEL: func @insert_poison_fold3 |
| // CHECK: vector.insertelement |
| func.func @insert_poison_fold3() -> vector<4xi32> { |
| %v = arith.constant dense<[0, 1, 2, 3]> : vector<4xi32> |
| %s = arith.constant 7 : i32 |
| %i = ub.poison : i32 |
| %1 = vector.insertelement %s, %v[%i : i32] : vector<4xi32> |
| return %1 : vector<4xi32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func @extract_element_fold |
| // CHECK: %[[C:.+]] = arith.constant 5 : i32 |
| // CHECK: return %[[C]] |
| func.func @extract_element_fold() -> i32 { |
| %v = arith.constant dense<[1, 3, 5, 7]> : vector<4xi32> |
| %i = arith.constant 2 : i32 |
| %1 = vector.extractelement %v[%i : i32] : vector<4xi32> |
| return %1 : i32 |
| } |
| |
| // CHECK-LABEL: func @extract_element_splat_fold |
| // CHECK-SAME: (%[[ARG:.+]]: i32) |
| // CHECK: return %[[ARG]] |
| func.func @extract_element_splat_fold(%a : i32) -> i32 { |
| %v = vector.splat %a : vector<4xi32> |
| %i = arith.constant 2 : i32 |
| %1 = vector.extractelement %v[%i : i32] : vector<4xi32> |
| return %1 : i32 |
| } |
| |
| // ----- |
| |
| // Do not crash on poison |
| // CHECK-LABEL: func @extract_element_poison_fold1 |
| // CHECK: vector.extractelement |
| func.func @extract_element_poison_fold1() -> i32 { |
| %v = ub.poison : vector<4xi32> |
| %i = arith.constant 2 : i32 |
| %1 = vector.extractelement %v[%i : i32] : vector<4xi32> |
| return %1 : i32 |
| } |
| |
| // ----- |
| |
| // Do not crash on poison |
| // CHECK-LABEL: func @extract_element_poison_fold2 |
| // CHECK: vector.extractelement |
| func.func @extract_element_poison_fold2() -> i32 { |
| %v = arith.constant dense<[1, 3, 5, 7]> : vector<4xi32> |
| %i = ub.poison : i32 |
| %1 = vector.extractelement %v[%i : i32] : vector<4xi32> |
| return %1 : i32 |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func @reduce_one_element_vector_extract |
| // CHECK-SAME: (%[[V:.+]]: vector<1xf32>) |
| // CHECK: %[[S:.+]] = vector.extract %[[V]][0] : f32 from vector<1xf32> |
| // CHECK: return %[[S]] : f32 |
| func.func @reduce_one_element_vector_extract(%a : vector<1xf32>) -> f32 { |
| %s = vector.reduction <add>, %a : vector<1xf32> into f32 |
| return %s : f32 |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func @masked_reduce_one_element_vector_extract |
| // CHECK-SAME: %[[VAL_0:.*]]: vector<1xf32>, %[[VAL_1:.*]]: vector<1xi1>) |
| func.func @masked_reduce_one_element_vector_extract(%a : vector<1xf32>, %mask : vector<1xi1>) -> f32 { |
| // CHECK: %[[VAL_2:.*]] = vector.extract %[[VAL_0]][0] : f32 from vector<1xf32> |
| %s = vector.mask %mask { vector.reduction <add>, %a : vector<1xf32> into f32 } |
| : vector<1xi1> -> f32 |
| return %s : f32 |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func @reduce_one_element_vector_addf |
| // CHECK-SAME: (%[[V:.+]]: vector<1xf32>, %[[B:.+]]: f32) |
| // CHECK: %[[A:.+]] = vector.extract %[[V]][0] : f32 from vector<1xf32> |
| // CHECK: %[[S:.+]] = arith.addf %[[A]], %arg1 : f32 |
| // CHECK: return %[[S]] |
| func.func @reduce_one_element_vector_addf(%a : vector<1xf32>, %b: f32) -> f32 { |
| %s = vector.reduction <add>, %a, %b : vector<1xf32> into f32 |
| return %s : f32 |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func @reduce_one_element_vector_addf_fastmath |
| // CHECK-SAME: (%[[V:.+]]: vector<1xf32>, %[[B:.+]]: f32) |
| // CHECK: %[[A:.+]] = vector.extract %[[V]][0] : f32 from vector<1xf32> |
| // CHECK: %[[S:.+]] = arith.addf %[[A]], %arg1 fastmath<nnan,ninf> : f32 |
| // CHECK: return %[[S]] |
| func.func @reduce_one_element_vector_addf_fastmath(%a : vector<1xf32>, %b: f32) -> f32 { |
| %s = vector.reduction <add>, %a, %b fastmath<nnan,ninf> : vector<1xf32> into f32 |
| return %s : f32 |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func @masked_reduce_one_element_vector_addf |
| // CHECK-SAME: %[[VAL_0:.*]]: vector<1xf32>, %[[VAL_1:.*]]: f32, |
| // CHECK-SAME: %[[VAL_2:.*]]: vector<1xi1>) |
| func.func @masked_reduce_one_element_vector_addf(%a: vector<1xf32>, |
| %b: f32, |
| %mask: vector<1xi1>) -> f32 { |
| // CHECK: %[[VAL_3:.*]] = vector.extract %[[VAL_2]][0] : i1 from vector<1xi1> |
| // CHECK: %[[VAL_4:.*]] = vector.extract %[[VAL_0]][0] : f32 from vector<1xf32> |
| // CHECK: %[[VAL_5:.*]] = arith.addf %[[VAL_4]], %[[VAL_1]] : f32 |
| // CHECK: %[[VAL_6:.*]] = arith.select %[[VAL_3]], %[[VAL_5]], %[[VAL_1]] : f32 |
| %s = vector.mask %mask { vector.reduction <add>, %a, %b : vector<1xf32> into f32 } |
| : vector<1xi1> -> f32 |
| return %s : f32 |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func @reduce_one_element_vector_mulf |
| // CHECK-SAME: (%[[V:.+]]: vector<1xf32>, %[[B:.+]]: f32) |
| // CHECK: %[[A:.+]] = vector.extract %[[V]][0] : f32 from vector<1xf32> |
| // CHECK: %[[S:.+]] = arith.mulf %[[A]], %arg1 : f32 |
| // CHECK: return %[[S]] |
| func.func @reduce_one_element_vector_mulf(%a : vector<1xf32>, %b: f32) -> f32 { |
| %s = vector.reduction <mul>, %a, %b : vector<1xf32> into f32 |
| return %s : f32 |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func @dont_reduce_one_element_vector |
| // CHECK: vector.reduction |
| func.func @dont_reduce_one_element_vector(%a : vector<4xf32>) -> f32 { |
| %s = vector.reduction <add>, %a : vector<4xf32> into f32 |
| return %s : f32 |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func @reduce_one_element_vector_maximumf |
| // CHECK-SAME: (%[[V:.+]]: vector<1xf32>, %[[B:.+]]: f32) |
| // CHECK: %[[A:.+]] = vector.extract %[[V]][0] : f32 from vector<1xf32> |
| // CHECK: %[[S:.+]] = arith.maximumf %[[A]], %[[B]] : f32 |
| // CHECK: return %[[S]] |
| func.func @reduce_one_element_vector_maximumf(%a : vector<1xf32>, %b: f32) -> f32 { |
| %s = vector.reduction <maximumf>, %a, %b : vector<1xf32> into f32 |
| return %s : f32 |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func @bitcast( |
| // CHECK-SAME: %[[ARG:.*]]: vector<4x8xf32>) -> vector<4x16xi16> { |
| // CHECK: vector.bitcast %[[ARG:.*]] : vector<4x8xf32> to vector<4x16xi16> |
| func.func @bitcast(%a: vector<4x8xf32>) -> vector<4x16xi16> { |
| %0 = vector.bitcast %a : vector<4x8xf32> to vector<4x8xi32> |
| %1 = vector.bitcast %0 : vector<4x8xi32> to vector<4x16xi16> |
| return %1 : vector<4x16xi16> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: @insert_strided_slice_splat |
| // CHECK-SAME: (%[[ARG:.*]]: f32) |
| // CHECK-NEXT: %[[SPLAT:.*]] = vector.splat %[[ARG]] : vector<8x16xf32> |
| // CHECK-NEXT: return %[[SPLAT]] : vector<8x16xf32> |
| func.func @insert_strided_slice_splat(%x: f32) -> (vector<8x16xf32>) { |
| %splat0 = vector.splat %x : vector<4x4xf32> |
| %splat1 = vector.splat %x : vector<8x16xf32> |
| %0 = vector.insert_strided_slice %splat0, %splat1 {offsets = [2, 2], strides = [1, 1]} |
| : vector<4x4xf32> into vector<8x16xf32> |
| return %0 : vector<8x16xf32> |
| } |
| |
| |
| // ----- |
| |
| // CHECK-LABEL: @insert_extract_strided_slice |
| // CHECK-SAME: (%[[ARG:.*]]: vector<8x16xf32>) |
| // CHECK-NEXT: return %[[ARG]] : vector<8x16xf32> |
| func.func @insert_extract_strided_slice(%x: vector<8x16xf32>) -> (vector<8x16xf32>) { |
| %0 = vector.extract_strided_slice %x {offsets = [0, 8], sizes = [2, 4], strides = [1, 1]} |
| : vector<8x16xf32> to vector<2x4xf32> |
| %1 = vector.insert_strided_slice %0, %x {offsets = [0, 8], strides = [1, 1]} |
| : vector<2x4xf32> into vector<8x16xf32> |
| return %1 : vector<8x16xf32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func.func @insert_strided_1d_constant |
| // CHECK-DAG: %[[ACST:.*]] = arith.constant dense<[4, 1, 2]> : vector<3xi32> |
| // CHECK-DAG: %[[BCST:.*]] = arith.constant dense<[0, 1, 4]> : vector<3xi32> |
| // CHECK-DAG: %[[CCST:.*]] = arith.constant dense<[5, 6, 2]> : vector<3xi32> |
| // CHECK-DAG: %[[DCST:.*]] = arith.constant dense<[0, 5, 6]> : vector<3xi32> |
| // CHECK-DAG: %[[ECST:.*]] = arith.constant dense<[7, 8, 9]> : vector<3xi32> |
| // CHECK-NEXT: return %[[ACST]], %[[BCST]], %[[CCST]], %[[DCST]], %[[ECST]] |
| func.func @insert_strided_1d_constant() -> |
| (vector<3xi32>, vector<3xi32>, vector<3xi32>, vector<3xi32>, vector<3xi32>) { |
| %vcst = arith.constant dense<[0, 1, 2]> : vector<3xi32> |
| %cst_1 = arith.constant dense<4> : vector<1xi32> |
| %cst_2 = arith.constant dense<[5, 6]> : vector<2xi32> |
| %cst_3 = arith.constant dense<[7, 8, 9]> : vector<3xi32> |
| %a = vector.insert_strided_slice %cst_1, %vcst {offsets = [0], strides = [1]} : vector<1xi32> into vector<3xi32> |
| %b = vector.insert_strided_slice %cst_1, %vcst {offsets = [2], strides = [1]} : vector<1xi32> into vector<3xi32> |
| %c = vector.insert_strided_slice %cst_2, %vcst {offsets = [0], strides = [1]} : vector<2xi32> into vector<3xi32> |
| %d = vector.insert_strided_slice %cst_2, %vcst {offsets = [1], strides = [1]} : vector<2xi32> into vector<3xi32> |
| %e = vector.insert_strided_slice %cst_3, %vcst {offsets = [0], strides = [1]} : vector<3xi32> into vector<3xi32> |
| return %a, %b, %c, %d, %e : vector<3xi32>, vector<3xi32>, vector<3xi32>, vector<3xi32>, vector<3xi32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func.func @insert_strided_2d_constant |
| // CHECK-DAG: %[[ACST:.*]] = arith.constant dense<{{\[\[0, 1\], \[9, 3\], \[4, 5\]\]}}> : vector<3x2xi32> |
| // CHECK-DAG: %[[BCST:.*]] = arith.constant dense<{{\[\[0, 1\], \[2, 3\], \[4, 9\]\]}}> : vector<3x2xi32> |
| // CHECK-DAG: %[[CCST:.*]] = arith.constant dense<{{\[\[18, 19\], \[2, 3\], \[4, 5\]\]}}> : vector<3x2xi32> |
| // CHECK-DAG: %[[DCST:.*]] = arith.constant dense<{{\[\[0, 1\], \[18, 19\], \[4, 5\]\]}}> : vector<3x2xi32> |
| // CHECK-DAG: %[[ECST:.*]] = arith.constant dense<{{\[\[0, 1\], \[2, 3\], \[18, 19\]\]}}> : vector<3x2xi32> |
| // CHECK-DAG: %[[FCST:.*]] = arith.constant dense<{{\[\[28, 29\], \[38, 39\], \[4, 5\]\]}}> : vector<3x2xi32> |
| // CHECK-DAG: %[[GCST:.*]] = arith.constant dense<{{\[\[0, 1\], \[28, 29\], \[38, 39\]\]}}> : vector<3x2xi32> |
| // CHECK-NEXT: return %[[ACST]], %[[BCST]], %[[CCST]], %[[DCST]], %[[ECST]], %[[FCST]], %[[GCST]] |
| func.func @insert_strided_2d_constant() -> |
| (vector<3x2xi32>, vector<3x2xi32>, vector<3x2xi32>, vector<3x2xi32>, vector<3x2xi32>, vector<3x2xi32>, vector<3x2xi32>) { |
| %vcst = arith.constant dense<[[0, 1], [2, 3], [4, 5]]> : vector<3x2xi32> |
| %cst_1 = arith.constant dense<9> : vector<1xi32> |
| %cst_2 = arith.constant dense<[18, 19]> : vector<2xi32> |
| %cst_3 = arith.constant dense<[[28, 29], [38, 39]]> : vector<2x2xi32> |
| %a = vector.insert_strided_slice %cst_1, %vcst {offsets = [1, 0], strides = [1]} : vector<1xi32> into vector<3x2xi32> |
| %b = vector.insert_strided_slice %cst_1, %vcst {offsets = [2, 1], strides = [1]} : vector<1xi32> into vector<3x2xi32> |
| %c = vector.insert_strided_slice %cst_2, %vcst {offsets = [0, 0], strides = [1]} : vector<2xi32> into vector<3x2xi32> |
| %d = vector.insert_strided_slice %cst_2, %vcst {offsets = [1, 0], strides = [1]} : vector<2xi32> into vector<3x2xi32> |
| %e = vector.insert_strided_slice %cst_2, %vcst {offsets = [2, 0], strides = [1]} : vector<2xi32> into vector<3x2xi32> |
| %f = vector.insert_strided_slice %cst_3, %vcst {offsets = [0, 0], strides = [1, 1]} : vector<2x2xi32> into vector<3x2xi32> |
| %g = vector.insert_strided_slice %cst_3, %vcst {offsets = [1, 0], strides = [1, 1]} : vector<2x2xi32> into vector<3x2xi32> |
| return %a, %b, %c, %d, %e, %f, %g : |
| vector<3x2xi32>, vector<3x2xi32>, vector<3x2xi32>, vector<3x2xi32>, vector<3x2xi32>, vector<3x2xi32>, vector<3x2xi32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func @shuffle_splat |
| // CHECK-SAME: (%[[ARG:.*]]: i32) |
| // CHECK-NEXT: %[[SPLAT:.*]] = vector.splat %[[ARG]] : vector<4xi32> |
| // CHECK-NEXT: return %[[SPLAT]] : vector<4xi32> |
| func.func @shuffle_splat(%x : i32) -> vector<4xi32> { |
| %v0 = vector.splat %x : vector<4xi32> |
| %v1 = vector.splat %x : vector<2xi32> |
| %shuffle = vector.shuffle %v0, %v1 [2, 3, 4, 5] : vector<4xi32>, vector<2xi32> |
| return %shuffle : vector<4xi32> |
| } |
| |
| |
| // ----- |
| |
| // CHECK-LABEL: func @insert_splat |
| // CHECK-SAME: (%[[ARG:.*]]: i32) |
| // CHECK-NEXT: %[[SPLAT:.*]] = vector.splat %[[ARG]] : vector<2x4x3xi32> |
| // CHECK-NEXT: return %[[SPLAT]] : vector<2x4x3xi32> |
| func.func @insert_splat(%x : i32) -> vector<2x4x3xi32> { |
| %v0 = vector.splat %x : vector<4x3xi32> |
| %v1 = vector.splat %x : vector<2x4x3xi32> |
| %insert = vector.insert %v0, %v1[0] : vector<4x3xi32> into vector<2x4x3xi32> |
| return %insert : vector<2x4x3xi32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func.func @transfer_read_from_rank_reducing_extract_slice |
| // CHECK: tensor.extract_slice |
| // CHECK: vector.transfer_read |
| func.func @transfer_read_from_rank_reducing_extract_slice(%src: tensor<1x8x8x8xf32>, %i1: index, %i2: index, %i3: index, %i4: index) -> vector<4xf32> { |
| %c0 = arith.constant 0 : index |
| %f0 = arith.constant 0.000000e+00 : f32 |
| %0 = tensor.extract_slice %src[0, %i1, %i2, %i3] [1, 4, 1, 4] [1, 1, 1, 1] : tensor<1x8x8x8xf32> to tensor<1x4x4xf32> |
| %1 = vector.transfer_read %0[%c0, %i4, %c0], %f0 {in_bounds = [true]} : tensor<1x4x4xf32>, vector<4xf32> |
| return %1 : vector<4xf32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func.func @extract_from_broadcast |
| func.func @extract_from_broadcast(%src: vector<1x1x1xf32>) -> vector<1xf32> { |
| %0 = vector.broadcast %src : vector<1x1x1xf32> to vector<1x1x32x1xf32> |
| |
| // CHECK-NEXT: %0 = vector.extract {{.*}}[0, 0] : vector<1xf32> from vector<1x1x1xf32> |
| // CHECK-NEXT: return %0 : vector<1xf32> |
| %1 = vector.extract %0[0, 0, 31] : vector<1xf32> from vector<1x1x32x1xf32> |
| return %1: vector<1xf32> |
| } |
| |
| // CHECK-LABEL: func.func @extract_from_stretch_broadcast |
| func.func @extract_from_stretch_broadcast(%src: vector<3x1x2xf32>) -> f32 { |
| // CHECK-NEXT: %0 = vector.extract {{.*}}[0, 0, 0] : f32 from vector<3x1x2xf32> |
| // CHECK-NEXT: return %0 : f32 |
| %0 = vector.broadcast %src : vector<3x1x2xf32> to vector<3x4x2xf32> |
| %1 = vector.extract %0[0, 2, 0] : f32 from vector<3x4x2xf32> |
| return %1: f32 |
| } |
| |
| // ----- |
| // CHECK-LABEL: func.func @extract_strided_slice_of_constant_mask |
| func.func @extract_strided_slice_of_constant_mask() -> vector<5x7xi1>{ |
| // CHECK-NEXT: %[[RES:.*]] = vector.constant_mask [5, 4] : vector<5x7xi1> |
| // CHECK-NEXT: return %[[RES]] : vector<5x7xi1> |
| %c4 = arith.constant 4 : index |
| %c10 = arith.constant 10 : index |
| %mask = vector.create_mask %c10, %c4 : vector<12x7xi1> |
| %res = vector.extract_strided_slice %mask {offsets = [3], sizes = [5], strides = [1]} : vector<12x7xi1> to vector<5x7xi1> |
| return %res : vector<5x7xi1> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func.func @fold_extractelement_of_broadcast( |
| // CHECK-SAME: %[[f:.*]]: f32 |
| // CHECK: return %[[f]] |
| func.func @fold_extractelement_of_broadcast(%f: f32) -> f32 { |
| %0 = vector.broadcast %f : f32 to vector<15xf32> |
| %c5 = arith.constant 5 : index |
| %1 = vector.extractelement %0 [%c5 : index] : vector<15xf32> |
| return %1 : f32 |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func.func @fold_0d_vector_reduction |
| func.func @fold_0d_vector_reduction(%arg0: vector<f32>) -> f32 { |
| // CHECK-NEXT: %[[RES:.*]] = vector.extractelement %arg{{.*}}[] : vector<f32> |
| // CHECK-NEXT: return %[[RES]] : f32 |
| %0 = vector.reduction <add>, %arg0 : vector<f32> into f32 |
| return %0 : f32 |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func @empty_vector_mask |
| func.func @empty_vector_mask(%mask : vector<8xi1>) { |
| // CHECK-NOT: vector.mask |
| vector.mask %mask { } : vector<8xi1> |
| return |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func @empty_vector_mask_with_return |
| // CHECK-SAME: %[[IN:.*]]: vector<8xf32> |
| func.func @empty_vector_mask_with_return(%a : vector<8xf32>, %mask : vector<8xi1>) -> vector<8xf32> { |
| // CHECK-NOT: vector.mask |
| // CHECK: return %[[IN]] : vector<8xf32> |
| %0 = vector.mask %mask { vector.yield %a : vector<8xf32> } : vector<8xi1> -> vector<8xf32> |
| return %0 : vector<8xf32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func @all_true_vector_mask |
| // CHECK-SAME: %[[IN:.*]]: vector<3x4xf32> |
| func.func @all_true_vector_mask(%a : vector<3x4xf32>) -> vector<3x4xf32> { |
| // CHECK-NOT: vector.mask |
| // CHECK: %[[ADD:.*]] = arith.addf %[[IN]], %[[IN]] : vector<3x4xf32> |
| // CHECK: return %[[ADD]] : vector<3x4xf32> |
| %all_true = vector.constant_mask [3, 4] : vector<3x4xi1> |
| %0 = vector.mask %all_true { arith.addf %a, %a : vector<3x4xf32> } : vector<3x4xi1> -> vector<3x4xf32> |
| return %0 : vector<3x4xf32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func @all_true_vector_mask_no_result( |
| func.func @all_true_vector_mask_no_result(%a : vector<3x4xf32>, %m : memref<3x4xf32>) { |
| // CHECK-NOT: vector.mask |
| // CHECK: vector.transfer_write |
| %c0 = arith.constant 0 : index |
| %all_true = vector.constant_mask [3, 4] : vector<3x4xi1> |
| vector.mask %all_true { vector.transfer_write %a, %m[%c0, %c0] : vector<3x4xf32>, memref<3x4xf32> } : vector<3x4xi1> |
| return |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func.func @fold_shape_cast_with_mask( |
| // CHECK-SAME: %[[VAL_0:.*]]: tensor<1x?xf32>) -> vector<1x4xi1> { |
| func.func @fold_shape_cast_with_mask(%arg0: tensor<1x?xf32>) -> vector<1x4xi1> { |
| // CHECK-NOT: vector.shape_cast |
| // CHECK: %[[VAL_1:.*]] = arith.constant 1 : index |
| // CHECK: %[[VAL_2:.*]] = tensor.dim %[[VAL_0]], %[[VAL_1]] : tensor<1x?xf32> |
| // CHECK: %[[VAL_3:.*]] = vector.create_mask %[[VAL_1]], %[[VAL_2]] : vector<1x4xi1> |
| // CHECK: return %[[VAL_3]] : vector<1x4xi1> |
| %c1 = arith.constant 1 : index |
| %dim = tensor.dim %arg0, %c1 : tensor<1x?xf32> |
| %1 = vector.create_mask %c1, %dim, %c1, %c1 : vector<1x4x1x1xi1> |
| %2 = vector.shape_cast %1 : vector<1x4x1x1xi1> to vector<1x4xi1> |
| return %2 : vector<1x4xi1> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func.func @fold_shape_cast_with_mask_scalable( |
| // CHECK-SAME: %[[VAL_0:.*]]: tensor<1x?xf32>) -> vector<1x[4]xi1> { |
| func.func @fold_shape_cast_with_mask_scalable(%arg0: tensor<1x?xf32>) -> vector<1x[4]xi1> { |
| // CHECK-NOT: vector.shape_cast |
| // CHECK: %[[VAL_1:.*]] = arith.constant 1 : index |
| // CHECK: %[[VAL_2:.*]] = tensor.dim %[[VAL_0]], %[[VAL_1]] : tensor<1x?xf32> |
| // CHECK: %[[VAL_3:.*]] = vector.create_mask %[[VAL_1]], %[[VAL_2]] : vector<1x[4]xi1> |
| // CHECK: return %[[VAL_3]] : vector<1x[4]xi1> |
| %c1 = arith.constant 1 : index |
| %dim = tensor.dim %arg0, %c1 : tensor<1x?xf32> |
| %1 = vector.create_mask %c1, %dim, %c1, %c1 : vector<1x[4]x1x1xi1> |
| %2 = vector.shape_cast %1 : vector<1x[4]x1x1xi1> to vector<1x[4]xi1> |
| return %2 : vector<1x[4]xi1> |
| } |
| |
| // ----- |
| |
| // Check that scalable "1" (i.e. [1]) is not folded |
| // CHECK-LABEL: func.func @fold_shape_cast_with_mask_scalable_one( |
| // CHECK-SAME: %[[VAL_0:.*]]: tensor<1x?xf32>) -> vector<1x[1]xi1> { |
| func.func @fold_shape_cast_with_mask_scalable_one(%arg0: tensor<1x?xf32>) -> vector<1x[1]xi1>{ |
| // CHECK: %[[VAL_1:.*]] = arith.constant 1 : index |
| // CHECK: %[[VAL_2:.*]] = tensor.dim %[[VAL_0]], %[[VAL_1]] : tensor<1x?xf32> |
| // CHECK: %[[VAL_3:.*]] = vector.create_mask %[[VAL_1]], %[[VAL_2]] : vector<1x[1]xi1> |
| // CHECK: return %[[VAL_3]] : vector<1x[1]xi1> |
| %c1 = arith.constant 1 : index |
| %dim = tensor.dim %arg0, %c1 : tensor<1x?xf32> |
| %1 = vector.create_mask %c1, %dim, %c1 : vector<1x[1]x1xi1> |
| %2 = vector.shape_cast %1 : vector<1x[1]x1xi1> to vector<1x[1]xi1> |
| return %2 : vector<1x[1]xi1> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func.func @fold_shape_cast_with_constant_mask() -> vector<4xi1> { |
| func.func @fold_shape_cast_with_constant_mask() -> vector<4xi1>{ |
| // CHECK-NOT: vector.shape_cast |
| // CHECK: %[[VAL_0:.*]] = vector.constant_mask [1] : vector<4xi1> |
| // CHECK: return %[[VAL_0]] : vector<4xi1> |
| %1 = vector.constant_mask [1, 1, 1] : vector<4x1x1xi1> |
| %2 = vector.shape_cast %1 : vector<4x1x1xi1> to vector<4xi1> |
| return %2 : vector<4xi1> |
| } |
| |
| // ----- |
| |
| // TODO: This IR could be canonicalized but the canonicalization pattern is not |
| // smart enough. For now, just make sure that we do not crash. |
| |
| // CHECK-LABEL: func.func @load_store_forwarding_rank_mismatch( |
| // CHECK: vector.transfer_write |
| // CHECK: vector.transfer_read |
| func.func @load_store_forwarding_rank_mismatch(%v0: vector<4x1x1xf32>, %arg0: tensor<4x4x4xf32>) -> (vector<1x100x4x5xf32>) { |
| %c0 = arith.constant 0 : index |
| %cf0 = arith.constant 0.0 : f32 |
| // d0 is explicitly written. |
| %w0 = vector.transfer_write %v0, %arg0[%c0, %c0, %c0] |
| {in_bounds = [true, true, true], |
| permutation_map = affine_map<(d0, d1, d2) -> (d2, d1, d0)>} : |
| vector<4x1x1xf32>, tensor<4x4x4xf32> |
| // d0 is implicitly read (rank-reduction of unit dim). |
| %r = vector.transfer_read %w0[%c0, %c0, %c0], %cf0 |
| {in_bounds = [true, true, true, true], |
| permutation_map = affine_map<(d0, d1, d2) -> (d1, 0, d2, 0)>} : |
| tensor<4x4x4xf32>, vector<1x100x4x5xf32> |
| return %r : vector<1x100x4x5xf32> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func.func @rank_0_shuffle_to_interleave( |
| // CHECK-SAME: %[[LHS:.*]]: vector<f64>, %[[RHS:.*]]: vector<f64>) |
| func.func @rank_0_shuffle_to_interleave(%arg0: vector<f64>, %arg1: vector<f64>) -> vector<2xf64> |
| { |
| // CHECK: %[[ZIP:.*]] = vector.interleave %[[LHS]], %[[RHS]] : vector<f64> |
| // CHECK: return %[[ZIP]] |
| %0 = vector.shuffle %arg0, %arg1 [0, 1] : vector<f64>, vector<f64> |
| return %0 : vector<2xf64> |
| } |
| |
| // ----- |
| |
| // CHECK-LABEL: func.func @rank_1_shuffle_to_interleave( |
| // CHECK-SAME: %[[LHS:.*]]: vector<6xi32>, %[[RHS:.*]]: vector<6xi32>) |
| func.func @rank_1_shuffle_to_interleave(%arg0: vector<6xi32>, %arg1: vector<6xi32>) -> vector<12xi32> { |
| // CHECK: %[[ZIP:.*]] = vector.interleave %[[LHS]], %[[RHS]] : vector<6xi32> |
| // CHECK: return %[[ZIP]] |
| %0 = vector.shuffle %arg0, %arg1 [0, 6, 1, 7, 2, 8, 3, 9, 4, 10, 5, 11] : vector<6xi32>, vector<6xi32> |
| return %0 : vector<12xi32> |
| } |