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
63 lines
5.3 KiB
Fortran
63 lines
5.3 KiB
Fortran
! Test forall lowering
|
|
|
|
! RUN: %flang_fc1 -emit-hlfir %s -o - | FileCheck %s
|
|
|
|
!*** Test a FORALL construct with an array assignment
|
|
! This is similar to the following embedded WHERE construct test, but the
|
|
! elements are assigned unconditionally.
|
|
subroutine test_forall_with_array_assignment(aa,bb)
|
|
type t
|
|
integer(kind=8) :: block1(64)
|
|
integer(kind=8) :: block2(64)
|
|
end type t
|
|
type(t) :: aa(10), bb(10)
|
|
|
|
forall (i=1:10:2)
|
|
aa(i)%block1 = bb(i+1)%block2
|
|
end forall
|
|
end subroutine test_forall_with_array_assignment
|
|
|
|
! CHECK-LABEL: func.func @_QPtest_forall_with_array_assignment(
|
|
! CHECK-SAME: %[[VAL_0:.*]]: !fir.ref<!fir.array<10x!fir.type<_QFtest_forall_with_array_assignmentTt{block1:!fir.array<64xi64>,block2:!fir.array<64xi64>}>>> {fir.bindc_name = "aa"},
|
|
! CHECK-SAME: %[[VAL_1:.*]]: !fir.ref<!fir.array<10x!fir.type<_QFtest_forall_with_array_assignmentTt{block1:!fir.array<64xi64>,block2:!fir.array<64xi64>}>>> {fir.bindc_name = "bb"}) {
|
|
! CHECK: %[[VAL_2:.*]] = fir.dummy_scope : !fir.dscope
|
|
! CHECK: %[[VAL_13:.*]] = arith.constant 10 : index
|
|
! CHECK: %[[VAL_14:.*]] = fir.shape %[[VAL_13]] : (index) -> !fir.shape<1>
|
|
! CHECK: %[[VAL_15:.*]]:2 = hlfir.declare %[[VAL_0]](%[[VAL_14]]) dummy_scope %[[VAL_2]] arg 1 {uniq_name = "_QFtest_forall_with_array_assignmentEaa"} : (!fir.ref<!fir.array<10x!fir.type<_QFtest_forall_with_array_assignmentTt{block1:!fir.array<64xi64>,block2:!fir.array<64xi64>}>>>, !fir.shape<1>, !fir.dscope) -> (!fir.ref<!fir.array<10x!fir.type<_QFtest_forall_with_array_assignmentTt{block1:!fir.array<64xi64>,block2:!fir.array<64xi64>}>>>, !fir.ref<!fir.array<10x!fir.type<_QFtest_forall_with_array_assignmentTt{block1:!fir.array<64xi64>,block2:!fir.array<64xi64>}>>>)
|
|
! CHECK: %[[VAL_16:.*]] = arith.constant 10 : index
|
|
! CHECK: %[[VAL_17:.*]] = fir.shape %[[VAL_16]] : (index) -> !fir.shape<1>
|
|
! CHECK: %[[VAL_18:.*]]:2 = hlfir.declare %[[VAL_1]](%[[VAL_17]]) dummy_scope %[[VAL_2]] arg 2 {uniq_name = "_QFtest_forall_with_array_assignmentEbb"} : (!fir.ref<!fir.array<10x!fir.type<_QFtest_forall_with_array_assignmentTt{block1:!fir.array<64xi64>,block2:!fir.array<64xi64>}>>>, !fir.shape<1>, !fir.dscope) -> (!fir.ref<!fir.array<10x!fir.type<_QFtest_forall_with_array_assignmentTt{block1:!fir.array<64xi64>,block2:!fir.array<64xi64>}>>>, !fir.ref<!fir.array<10x!fir.type<_QFtest_forall_with_array_assignmentTt{block1:!fir.array<64xi64>,block2:!fir.array<64xi64>}>>>)
|
|
! CHECK: %[[VAL_19:.*]] = arith.constant 1 : i32
|
|
! CHECK: %[[VAL_20:.*]] = arith.constant 10 : i32
|
|
! CHECK: %[[VAL_21:.*]] = arith.constant 2 : i32
|
|
! CHECK: hlfir.forall lb {
|
|
! CHECK: hlfir.yield %[[VAL_19]] : i32
|
|
! CHECK: } ub {
|
|
! CHECK: hlfir.yield %[[VAL_20]] : i32
|
|
! CHECK: } step {
|
|
! CHECK: hlfir.yield %[[VAL_21]] : i32
|
|
! CHECK: } (%[[VAL_22:.*]]: i32) {
|
|
! CHECK: %[[VAL_23:.*]] = hlfir.forall_index "i" %[[VAL_22]] : (i32) -> !fir.ref<i32>
|
|
! CHECK: hlfir.region_assign {
|
|
! CHECK: %[[VAL_24:.*]] = fir.load %[[VAL_23]] : !fir.ref<i32>
|
|
! CHECK: %[[VAL_25:.*]] = arith.constant 1 : i32
|
|
! CHECK: %[[VAL_26:.*]] = arith.addi %[[VAL_24]], %[[VAL_25]] overflow<nsw> : i32
|
|
! CHECK: %[[VAL_27:.*]] = fir.convert %[[VAL_26]] : (i32) -> i64
|
|
! CHECK: %[[VAL_28:.*]] = hlfir.designate %[[VAL_18]]#0 (%[[VAL_27]]) : (!fir.ref<!fir.array<10x!fir.type<_QFtest_forall_with_array_assignmentTt{block1:!fir.array<64xi64>,block2:!fir.array<64xi64>}>>>, i64) -> !fir.ref<!fir.type<_QFtest_forall_with_array_assignmentTt{block1:!fir.array<64xi64>,block2:!fir.array<64xi64>}>>
|
|
! CHECK: %[[VAL_29:.*]] = arith.constant 64 : index
|
|
! CHECK: %[[VAL_30:.*]] = fir.shape %[[VAL_29]] : (index) -> !fir.shape<1>
|
|
! CHECK: %[[VAL_31:.*]] = hlfir.designate %[[VAL_28]]{"block2"}{{ *}}shape %[[VAL_30]] : (!fir.ref<!fir.type<_QFtest_forall_with_array_assignmentTt{block1:!fir.array<64xi64>,block2:!fir.array<64xi64>}>>, !fir.shape<1>) -> !fir.ref<!fir.array<64xi64>>
|
|
! CHECK: hlfir.yield %[[VAL_31]] : !fir.ref<!fir.array<64xi64>>
|
|
! CHECK: } to {
|
|
! CHECK: %[[VAL_32:.*]] = fir.load %[[VAL_23]] : !fir.ref<i32>
|
|
! CHECK: %[[VAL_33:.*]] = fir.convert %[[VAL_32]] : (i32) -> i64
|
|
! CHECK: %[[VAL_34:.*]] = hlfir.designate %[[VAL_15]]#0 (%[[VAL_33]]) : (!fir.ref<!fir.array<10x!fir.type<_QFtest_forall_with_array_assignmentTt{block1:!fir.array<64xi64>,block2:!fir.array<64xi64>}>>>, i64) -> !fir.ref<!fir.type<_QFtest_forall_with_array_assignmentTt{block1:!fir.array<64xi64>,block2:!fir.array<64xi64>}>>
|
|
! CHECK: %[[VAL_35:.*]] = arith.constant 64 : index
|
|
! CHECK: %[[VAL_36:.*]] = fir.shape %[[VAL_35]] : (index) -> !fir.shape<1>
|
|
! CHECK: %[[VAL_37:.*]] = hlfir.designate %[[VAL_34]]{"block1"}{{ *}}shape %[[VAL_36]] : (!fir.ref<!fir.type<_QFtest_forall_with_array_assignmentTt{block1:!fir.array<64xi64>,block2:!fir.array<64xi64>}>>, !fir.shape<1>) -> !fir.ref<!fir.array<64xi64>>
|
|
! CHECK: hlfir.yield %[[VAL_37]] : !fir.ref<!fir.array<64xi64>>
|
|
! CHECK: }
|
|
! CHECK: }
|
|
! CHECK: return
|
|
! CHECK: }
|