derek.holloway
7502bf64a4
CodeQL / Analyze (actions) (push) Canceled after 0s
CodeQL / Analyze (python) (push) Canceled after 0s
pre-commit / pre-commit (push) Canceled after 0s
Multi-Arch CI / setup (push) Canceled after 0s
Publish build_manylinux_x86_64 images / publish_build_manylinux_x86_64 (push) Canceled after 0s
Setup (Multi-Arch) / setup (push) Canceled after 0s
Multi-Arch CI / Linux::${{ fromJSON(needs.setup.outputs.linux_build_config || '{}').build_variant_label || 'skip' }} (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish no_rocm_image_ubi10 image / publish_no_rocm_image_ubi10 (push) Canceled after 0s
Multi-Arch CI / Windows::${{ fromJSON(needs.setup.outputs.windows_build_config || '{}').build_variant_label || 'skip' }} (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Multi-Arch CI / Manifest Diff (push) Canceled after 0s
TheRock Manifest Diff Report / Generate Manifest Diff Report (push) Canceled after 0s
Multi-Arch CI / CI Summary (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04 (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04_ocl_rt images / publish_no_rocm_image_ubuntu24_04_ocl_rt (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_rocgdb (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04_rocgdb images / publish_no_rocm_image_ubuntu24_04_rocgdb (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_ocl_rt (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Unit Tests / Unit Tests :: ubuntu-24.04 (push) Canceled after 0s
Unit Tests / Unit Tests :: windows-2022 (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_media (push) Canceled after 0s
Unit Tests / Unit Tests Summary (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_openmpi (push) Canceled after 0s
240 lines
13 KiB
Plaintext
240 lines
13 KiB
Plaintext
// Test hlfir.sum operation parse, verify (no errors), and unparse
|
|
|
|
// RUN: fir-opt %s | fir-opt | FileCheck %s
|
|
|
|
// array is an expression of known shape
|
|
func.func @sum0(%arg0: !hlfir.expr<42xi32>) {
|
|
%mask = fir.alloca !fir.logical<4>
|
|
%c_1 = arith.constant 1 : index
|
|
%true = arith.constant true
|
|
%true_logical = fir.convert %true : (i1) -> !fir.logical<4>
|
|
fir.store %true_logical to %mask : !fir.ref<!fir.logical<4>>
|
|
%mask_box = fir.embox %mask : (!fir.ref<!fir.logical<4>>) -> !fir.box<!fir.logical<4>>
|
|
%0 = hlfir.sum %arg0 dim %c_1 mask %mask_box : (!hlfir.expr<42xi32>, index, !fir.box<!fir.logical<4>>) -> i32
|
|
return
|
|
}
|
|
// CHECK: func.func @sum0(%[[ARRAY:.*]]: !hlfir.expr<42xi32>) {
|
|
// CHECK-NEXT: %[[MASK:.*]] = fir.alloca !fir.logical<4>
|
|
// CHECK-NEXT: %[[C1:.*]] = arith.constant 1 : index
|
|
// CHECK-NEXT: %[[TRUE:.*]] = arith.constant true
|
|
// CHECK-NEXT: %[[LOGICAL:.*]] = fir.convert %[[TRUE]] : (i1) -> !fir.logical<4>
|
|
// CHECK-NEXT: fir.store %[[LOGICAL]] to %[[MASK]] : !fir.ref<!fir.logical<4>>
|
|
// CHECK-NEXT: %[[BOX:.*]] = fir.embox %0 : (!fir.ref<!fir.logical<4>>) -> !fir.box<!fir.logical<4>>
|
|
// CHECK-NEXT: hlfir.sum %[[ARRAY]] dim %[[C1]] mask %[[BOX]] : (!hlfir.expr<42xi32>, index, !fir.box<!fir.logical<4>>) -> i32
|
|
// CHECK-NEXT: return
|
|
// CHECK-NEXT: }
|
|
|
|
// array is an expression of assumed shape
|
|
func.func @sum1(%arg0: !hlfir.expr<?xi32>) {
|
|
%mask = fir.alloca !fir.logical<4>
|
|
%c_1 = arith.constant 1 : index
|
|
%true = arith.constant true
|
|
%true_logical = fir.convert %true : (i1) -> !fir.logical<4>
|
|
fir.store %true_logical to %mask : !fir.ref<!fir.logical<4>>
|
|
%mask_box = fir.embox %mask : (!fir.ref<!fir.logical<4>>) -> !fir.box<!fir.logical<4>>
|
|
%0 = hlfir.sum %arg0 dim %c_1 mask %mask_box : (!hlfir.expr<?xi32>, index, !fir.box<!fir.logical<4>>) -> i32
|
|
return
|
|
}
|
|
// CHECK: func.func @sum1(%[[ARRAY:.*]]: !hlfir.expr<?xi32>) {
|
|
// CHECK-NEXT: %[[MASK:.*]] = fir.alloca !fir.logical<4>
|
|
// CHECK-NEXT: %[[C1:.*]] = arith.constant 1 : index
|
|
// CHECK-NEXT: %[[TRUE:.*]] = arith.constant true
|
|
// CHECK-NEXT: %[[LOGICAL:.*]] = fir.convert %[[TRUE]] : (i1) -> !fir.logical<4>
|
|
// CHECK-NEXT: fir.store %[[LOGICAL:.*]] to %[[MASK:.*]] : !fir.ref<!fir.logical<4>>
|
|
// CHECK-NEXT: %[[BOX:.*]] = fir.embox %[[MASK:.*]] : (!fir.ref<!fir.logical<4>>) -> !fir.box<!fir.logical<4>>
|
|
// CHECK-NEXT: hlfir.sum %[[ARRAY:.*]] dim %[[C1]] mask %[[BOX]] : (!hlfir.expr<?xi32>, index, !fir.box<!fir.logical<4>>) -> i32
|
|
// CHECK-NEXT: return
|
|
// CHECK-NEXT: }
|
|
|
|
// boxed array
|
|
func.func @sum2(%arg0: !fir.box<!fir.array<42xi32>>) {
|
|
%mask = fir.alloca !fir.logical<4>
|
|
%c_1 = arith.constant 1 : index
|
|
%true = arith.constant true
|
|
%true_logical = fir.convert %true : (i1) -> !fir.logical<4>
|
|
fir.store %true_logical to %mask : !fir.ref<!fir.logical<4>>
|
|
%mask_box = fir.embox %mask : (!fir.ref<!fir.logical<4>>) -> !fir.box<!fir.logical<4>>
|
|
%0 = hlfir.sum %arg0 dim %c_1 mask %mask_box : (!fir.box<!fir.array<42xi32>>, index, !fir.box<!fir.logical<4>>) -> i32
|
|
return
|
|
}
|
|
// CHECK: func.func @sum2(%[[ARRAY:.*]]: !fir.box<!fir.array<42xi32>>) {
|
|
// CHECK-NEXT: %[[MASK:.*]] = fir.alloca !fir.logical<4>
|
|
// CHECK-NEXT: %[[C1:.*]] = arith.constant 1 : index
|
|
// CHECK-NEXT: %[[TRUE:.*]] = arith.constant true
|
|
// CHECK-NEXT: %[[LOGICAL:.*]] = fir.convert %[[TRUE]] : (i1) -> !fir.logical<4>
|
|
// CHECK-NEXT: fir.store %[[LOGICAL:.*]] to %[[MASK:.*]] : !fir.ref<!fir.logical<4>>
|
|
// CHECK-NEXT: %[[BOX:.*]] = fir.embox %[[MASK:.*]] : (!fir.ref<!fir.logical<4>>) -> !fir.box<!fir.logical<4>>
|
|
// CHECK-NEXT: hlfir.sum %[[ARRAY:.*]] dim %[[C1]] mask %[[BOX]] : (!fir.box<!fir.array<42xi32>>, index, !fir.box<!fir.logical<4>>) -> i32
|
|
// CHECK-NEXT: return
|
|
// CHECK-NEXT: }
|
|
|
|
// assumed shape boxed array
|
|
func.func @sum3(%arg0: !fir.box<!fir.array<?xi32>>) {
|
|
%mask = fir.alloca !fir.logical<4>
|
|
%c_1 = arith.constant 1 : index
|
|
%true = arith.constant true
|
|
%true_logical = fir.convert %true : (i1) -> !fir.logical<4>
|
|
fir.store %true_logical to %mask : !fir.ref<!fir.logical<4>>
|
|
%mask_box = fir.embox %mask : (!fir.ref<!fir.logical<4>>) -> !fir.box<!fir.logical<4>>
|
|
%0 = hlfir.sum %arg0 dim %c_1 mask %mask_box : (!fir.box<!fir.array<?xi32>>, index, !fir.box<!fir.logical<4>>) -> i32
|
|
return
|
|
}
|
|
// CHECK: func.func @sum3(%[[ARRAY:.*]]: !fir.box<!fir.array<?xi32>>) {
|
|
// CHECK-NEXT: %[[MASK:.*]] = fir.alloca !fir.logical<4>
|
|
// CHECK-NEXT: %[[C1:.*]] = arith.constant 1 : index
|
|
// CHECK-NEXT: %[[TRUE:.*]] = arith.constant true
|
|
// CHECK-NEXT: %[[LOGICAL:.*]] = fir.convert %[[TRUE]] : (i1) -> !fir.logical<4>
|
|
// CHECK-NEXT: fir.store %[[LOGICAL:.*]] to %[[MASK:.*]] : !fir.ref<!fir.logical<4>>
|
|
// CHECK-NEXT: %[[BOX:.*]] = fir.embox %[[MASK:.*]] : (!fir.ref<!fir.logical<4>>) -> !fir.box<!fir.logical<4>>
|
|
// CHECK-NEXT: hlfir.sum %[[ARRAY:.*]] dim %[[C1]] mask %[[BOX]] : (!fir.box<!fir.array<?xi32>>, index, !fir.box<!fir.logical<4>>) -> i32
|
|
// CHECK-NEXT: return
|
|
// CHECK-NEXT: }
|
|
|
|
// known shape expr mask
|
|
func.func @sum4(%arg0: !fir.box<!fir.array<?xi32>>, %arg1: !hlfir.expr<42x!fir.logical<4>>) {
|
|
%c_1 = arith.constant 1 : index
|
|
%0 = hlfir.sum %arg0 dim %c_1 mask %arg1 : (!fir.box<!fir.array<?xi32>>, index, !hlfir.expr<42x!fir.logical<4>>) -> i32
|
|
return
|
|
}
|
|
// CHECK: func.func @sum4(%[[ARRAY:.*]]: !fir.box<!fir.array<?xi32>>, %[[MASK:.*]]: !hlfir.expr<42x!fir.logical<4>>) {
|
|
// CHECK-NEXT: %[[C1:.*]] = arith.constant 1 : index
|
|
// CHECK-NEXT: hlfir.sum %[[ARRAY]] dim %[[C1]] mask %[[MASK]] : (!fir.box<!fir.array<?xi32>>, index, !hlfir.expr<42x!fir.logical<4>>) -> i32
|
|
// CHECK-NEXT: return
|
|
// CHECK-NEXT: }
|
|
|
|
// assumed shape expr mask
|
|
func.func @sum5(%arg0: !fir.box<!fir.array<?xi32>>, %arg1: !hlfir.expr<?x!fir.logical<4>>) {
|
|
%c_1 = arith.constant 1 : index
|
|
%0 = hlfir.sum %arg0 dim %c_1 mask %arg1 : (!fir.box<!fir.array<?xi32>>, index, !hlfir.expr<?x!fir.logical<4>>) -> i32
|
|
return
|
|
}
|
|
// CHECK: func.func @sum5(%[[ARRAY:.*]]: !fir.box<!fir.array<?xi32>>, %[[MASK:.*]]: !hlfir.expr<?x!fir.logical<4>>) {
|
|
// CHECK-NEXT: %[[C1:.*]] = arith.constant 1 : index
|
|
// CHECK-NEXT: hlfir.sum %[[ARRAY]] dim %[[C1]] mask %[[MASK]] : (!fir.box<!fir.array<?xi32>>, index, !hlfir.expr<?x!fir.logical<4>>) -> i32
|
|
// CHECK-NEXT: return
|
|
// CHECK-NEXT: }
|
|
|
|
// known shape array mask
|
|
func.func @sum6(%arg0: !fir.box<!fir.array<?xi32>>, %arg1: !fir.box<!fir.array<42x!fir.logical<4>>>) {
|
|
%c_1 = arith.constant 1 : index
|
|
%0 = hlfir.sum %arg0 dim %c_1 mask %arg1 : (!fir.box<!fir.array<?xi32>>, index, !fir.box<!fir.array<42x!fir.logical<4>>>) -> i32
|
|
return
|
|
}
|
|
// CHECK: func.func @sum6(%[[ARRAY:.*]]: !fir.box<!fir.array<?xi32>>, %[[MASK:.*]]: !fir.box<!fir.array<42x!fir.logical<4>>>) {
|
|
// CHECK-NEXT: %[[C1:.*]] = arith.constant 1 : index
|
|
// CHECK-NEXT: hlfir.sum %[[ARRAY]] dim %[[C1]] mask %[[MASK]] : (!fir.box<!fir.array<?xi32>>, index, !fir.box<!fir.array<42x!fir.logical<4>>>) -> i32
|
|
// CHECK-NEXT: return
|
|
// CHECK-NEXT: }
|
|
|
|
// assumed shape array mask
|
|
func.func @sum7(%arg0: !fir.box<!fir.array<?xi32>>, %arg1: !fir.box<!fir.array<?x!fir.logical<4>>>) {
|
|
%c_1 = arith.constant 1 : index
|
|
%0 = hlfir.sum %arg0 dim %c_1 mask %arg1 : (!fir.box<!fir.array<?xi32>>, index, !fir.box<!fir.array<?x!fir.logical<4>>>) -> i32
|
|
return
|
|
}
|
|
// CHECK: func.func @sum7(%[[ARRAY:.*]]: !fir.box<!fir.array<?xi32>>, %[[MASK:.*]]: !fir.box<!fir.array<?x!fir.logical<4>>>) {
|
|
// CHECK-NEXT: %[[C1:.*]] = arith.constant 1 : index
|
|
// CHECK-NEXT: hlfir.sum %[[ARRAY]] dim %[[C1]] mask %[[MASK]] : (!fir.box<!fir.array<?xi32>>, index, !fir.box<!fir.array<?x!fir.logical<4>>>) -> i32
|
|
// CHECK-NEXT: return
|
|
// CHECK-NEXT: }
|
|
|
|
// known shape expr return
|
|
func.func @sum8(%arg0: !fir.box<!fir.array<2x2xi32>>, %arg1: i32) {
|
|
%mask = fir.alloca !fir.logical<4>
|
|
%true = arith.constant true
|
|
%true_logical = fir.convert %true : (i1) -> !fir.logical<4>
|
|
fir.store %true_logical to %mask : !fir.ref<!fir.logical<4>>
|
|
%mask_box = fir.embox %mask : (!fir.ref<!fir.logical<4>>) -> !fir.box<!fir.logical<4>>
|
|
%0 = hlfir.sum %arg0 dim %arg1 mask %mask_box : (!fir.box<!fir.array<2x2xi32>>, i32, !fir.box<!fir.logical<4>>) -> !hlfir.expr<2xi32>
|
|
return
|
|
}
|
|
// CHECK: func.func @sum8(%[[ARRAY:.*]]: !fir.box<!fir.array<2x2xi32>>, %[[DIM:.*]]: i32) {
|
|
// CHECK-NEXT: %[[MASK:.*]] = fir.alloca !fir.logical<4>
|
|
// CHECK-NEXT: %[[TRUE:.*]] = arith.constant true
|
|
// CHECK-NEXT: %[[LOGICAL:.*]] = fir.convert %[[TRUE]] : (i1) -> !fir.logical<4>
|
|
// CHECK-NEXT: fir.store %[[LOGICAL]] to %[[MASK]] : !fir.ref<!fir.logical<4>>
|
|
// CHECK-NEXT: %[[BOX:.*]] = fir.embox %0 : (!fir.ref<!fir.logical<4>>) -> !fir.box<!fir.logical<4>>
|
|
// CHECK-NEXT: hlfir.sum %[[ARRAY]] dim %[[DIM]] mask %[[BOX]] : (!fir.box<!fir.array<2x2xi32>>, i32, !fir.box<!fir.logical<4>>) -> !hlfir.expr<2xi32>
|
|
// CHECK-NEXT: return
|
|
// CHECK-NEXT: }
|
|
|
|
// assumed shape expr return
|
|
func.func @sum9(%arg0: !fir.box<!fir.array<?x?xi32>>, %arg1: i32) {
|
|
%mask = fir.alloca !fir.logical<4>
|
|
%true = arith.constant true
|
|
%true_logical = fir.convert %true : (i1) -> !fir.logical<4>
|
|
fir.store %true_logical to %mask : !fir.ref<!fir.logical<4>>
|
|
%mask_box = fir.embox %mask : (!fir.ref<!fir.logical<4>>) -> !fir.box<!fir.logical<4>>
|
|
%0 = hlfir.sum %arg0 dim %arg1 mask %mask_box : (!fir.box<!fir.array<?x?xi32>>, i32, !fir.box<!fir.logical<4>>) -> !hlfir.expr<?xi32>
|
|
return
|
|
}
|
|
// CHECK: func.func @sum9(%[[ARRAY:.*]]: !fir.box<!fir.array<?x?xi32>>, %[[DIM:.*]]: i32) {
|
|
// CHECK-NEXT: %[[MASK:.*]] = fir.alloca !fir.logical<4>
|
|
// CHECK-NEXT: %[[TRUE:.*]] = arith.constant true
|
|
// CHECK-NEXT: %[[LOGICAL:.*]] = fir.convert %[[TRUE]] : (i1) -> !fir.logical<4>
|
|
// CHECK-NEXT: fir.store %[[LOGICAL]] to %[[MASK]] : !fir.ref<!fir.logical<4>>
|
|
// CHECK-NEXT: %[[BOX:.*]] = fir.embox %0 : (!fir.ref<!fir.logical<4>>) -> !fir.box<!fir.logical<4>>
|
|
// CHECK-NEXT: hlfir.sum %[[ARRAY]] dim %[[DIM]] mask %[[BOX]] : (!fir.box<!fir.array<?x?xi32>>, i32, !fir.box<!fir.logical<4>>) -> !hlfir.expr<?xi32>
|
|
// CHECK-NEXT: return
|
|
// CHECK-NEXT: }
|
|
|
|
// hlfir.sum with only an array argument
|
|
func.func @sum10(%arg0: !fir.box<!fir.array<?x?xi32>>) {
|
|
%sum = hlfir.sum %arg0 : (!fir.box<!fir.array<?x?xi32>>) -> i32
|
|
return
|
|
}
|
|
// CHECK: func.func @sum10(%[[ARRAY:.*]]: !fir.box<!fir.array<?x?xi32>>
|
|
// CHECK-NEXT: %[[SUM:.*]] = hlfir.sum %[[ARRAY]] : (!fir.box<!fir.array<?x?xi32>>) -> i32
|
|
// CHECK-NEXT: return
|
|
// CHECK-NEXT: }
|
|
|
|
// hlfir.sum with array and dim argument
|
|
func.func @sum11(%arg0: !fir.box<!fir.array<?x?xi32>>, %arg1: i32) {
|
|
%sum = hlfir.sum %arg0 dim %arg1 : (!fir.box<!fir.array<?x?xi32>>, i32) -> !hlfir.expr<?xi32>
|
|
return
|
|
}
|
|
// CHECK: func.func @sum11(%[[ARRAY:.*]]: !fir.box<!fir.array<?x?xi32>>, %[[DIM:.*]]: i32
|
|
// CHECK-NEXT: %[[SUM:.*]] = hlfir.sum %[[ARRAY]] dim %[[DIM]] : (!fir.box<!fir.array<?x?xi32>>, i32) -> !hlfir.expr<?xi32>
|
|
// CHECK-NEXT: return
|
|
// CHECK-NEXT: }
|
|
|
|
// hlfir.sum with array and mask argument
|
|
func.func @sum12(%arg0: !fir.box<!fir.array<?xi32>>, %arg1: !fir.logical<4>) {
|
|
%sum = hlfir.sum %arg0 mask %arg1 : (!fir.box<!fir.array<?xi32>>, !fir.logical<4>) -> i32
|
|
return
|
|
}
|
|
// CHECK: func.func @sum12(%[[ARRAY:.*]]: !fir.box<!fir.array<?xi32>>, %[[MASK:.*]]: !fir.logical<4>
|
|
// CHECK-NEXT: %[[SUM:.*]] = hlfir.sum %[[ARRAY]] mask %[[MASK]] : (!fir.box<!fir.array<?xi32>>, !fir.logical<4>) -> i32
|
|
// CHECK-NEXT: return
|
|
// CHECK-NEXT: }
|
|
|
|
// hlfir.sum with dim argument with an unusual type
|
|
func.func @sum13(%arg0: !fir.box<!fir.array<?x?xi32>>, %arg1: index) {
|
|
%sum = hlfir.sum %arg0 dim %arg1 : (!fir.box<!fir.array<?x?xi32>>, index) -> !hlfir.expr<?xi32>
|
|
return
|
|
}
|
|
// CHECK: func.func @sum13(%[[ARRAY:.*]]: !fir.box<!fir.array<?x?xi32>>, %[[DIM:.*]]: index
|
|
// CHECK-NEXT: %[[SUM:.*]] = hlfir.sum %[[ARRAY]] dim %[[DIM]] : (!fir.box<!fir.array<?x?xi32>>, index) -> !hlfir.expr<?xi32>
|
|
// CHECK-NEXT: return
|
|
// CHECK-NEXT: }
|
|
|
|
// hlfir.sum with mask argument of unusual type
|
|
func.func @sum14(%arg0: !fir.box<!fir.array<?xi32>>, %arg1: i1) {
|
|
%sum = hlfir.sum %arg0 mask %arg1 : (!fir.box<!fir.array<?xi32>>, i1) -> i32
|
|
return
|
|
}
|
|
// CHECK: func.func @sum14(%[[ARRAY:.*]]: !fir.box<!fir.array<?xi32>>, %[[MASK:.*]]: i1
|
|
// CHECK-NEXT: %[[SUM:.*]] = hlfir.sum %[[ARRAY]] mask %[[MASK]] : (!fir.box<!fir.array<?xi32>>, i1) -> i32
|
|
// CHECK-NEXT: return
|
|
// CHECK-NEXT: }
|
|
|
|
// hlfir.sum with mask argument of ref<array<>> type
|
|
func.func @sum15(%arg0: !fir.box<!fir.array<?xi32>>, %arg1: !fir.ref<!fir.array<?x!fir.logical<4>>>) {
|
|
%sum = hlfir.sum %arg0 mask %arg1 : (!fir.box<!fir.array<?xi32>>, !fir.ref<!fir.array<?x!fir.logical<4>>>) -> i32
|
|
return
|
|
}
|
|
// CHECK: func.func @sum15(%[[ARRAY:.*]]: !fir.box<!fir.array<?xi32>>, %[[MASK:.*]]: !fir.ref<!fir.array<?x!fir.logical<4>>>
|
|
// CHECK-NEXT: %[[SUM:.*]] = hlfir.sum %[[ARRAY]] mask %[[MASK]] : (!fir.box<!fir.array<?xi32>>, !fir.ref<!fir.array<?x!fir.logical<4>>>) -> i32
|
|
// CHECK-NEXT: return
|
|
// CHECK-NEXT: }
|