Files
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
Flatten subprojects into a single base monorepo
2026-09-02 16:34:47 -07:00

276 lines
11 KiB
Plaintext

// RUN: fir-opt %s --pass-pipeline="builtin.module(acc-initialize-fir-analyses,acc-implicit-data)" -split-input-file | FileCheck %s
// Test implicit firstprivate behavior for various scalar types in parallel and serial constructs.
// Scalars in parallel/serial constructs should be implicitly firstprivate according to OpenACC spec.
// -----
// CHECK-LABEL: acc.firstprivate.recipe @firstprivatization_ref_i32 : !fir.ref<i32> init {
// CHECK: ^bb0(%{{.*}}: !fir.ref<i32>):
// CHECK: %[[ALLOC:.*]] = fir.alloca i32
// CHECK: acc.yield %[[ALLOC]] : !fir.ref<i32>
// CHECK: } copy {
// CHECK: ^bb0(%[[SRC:.*]]: !fir.ref<i32>, %[[DST:.*]]: !fir.ref<i32>):
// CHECK: %[[LOADED:.*]] = fir.load %[[SRC]] : !fir.ref<i32>
// CHECK: hlfir.assign %[[LOADED]] to %[[DST]] temporary_lhs : i32, !fir.ref<i32>
// CHECK: acc.terminator
// CHECK: }
// CHECK-LABEL: func.func @test_i32_scalar_in_parallel
func.func @test_i32_scalar_in_parallel() {
%scalar = fir.alloca i32 {bindc_name = "i32_var"}
acc.parallel {
%load = fir.load %scalar : !fir.ref<i32>
acc.yield
}
return
}
// CHECK: %[[FIRSTPRIV:.*]] = acc.firstprivate varPtr(%{{.*}} : !fir.ref<i32>) recipe(@firstprivatization_ref_i32) -> !fir.ref<i32> {implicit = true, name = "i32_var"}
// CHECK: acc.parallel firstprivate(%[[FIRSTPRIV]] : !fir.ref<i32>)
// -----
// CHECK-LABEL: acc.firstprivate.recipe @firstprivatization_ref_i64 : !fir.ref<i64> init {
// CHECK: ^bb0(%{{.*}}: !fir.ref<i64>):
// CHECK: %[[ALLOC:.*]] = fir.alloca i64
// CHECK: acc.yield %[[ALLOC]] : !fir.ref<i64>
// CHECK: } copy {
// CHECK: ^bb0(%[[SRC:.*]]: !fir.ref<i64>, %[[DST:.*]]: !fir.ref<i64>):
// CHECK: %[[LOADED:.*]] = fir.load %[[SRC]] : !fir.ref<i64>
// CHECK: hlfir.assign %[[LOADED]] to %[[DST]] temporary_lhs : i64, !fir.ref<i64>
// CHECK: acc.terminator
// CHECK: }
// CHECK-LABEL: func.func @test_i64_scalar_in_parallel
func.func @test_i64_scalar_in_parallel() {
%scalar = fir.alloca i64 {bindc_name = "i64_var"}
acc.parallel {
%load = fir.load %scalar : !fir.ref<i64>
acc.yield
}
return
}
// CHECK: %[[FIRSTPRIV:.*]] = acc.firstprivate varPtr(%{{.*}} : !fir.ref<i64>) recipe(@firstprivatization_ref_i64) -> !fir.ref<i64> {implicit = true, name = "i64_var"}
// CHECK: acc.parallel firstprivate(%[[FIRSTPRIV]] : !fir.ref<i64>)
// -----
// CHECK-LABEL: acc.firstprivate.recipe @firstprivatization_ref_f32 : !fir.ref<f32> init {
// CHECK: ^bb0(%{{.*}}: !fir.ref<f32>):
// CHECK: %[[ALLOC:.*]] = fir.alloca f32
// CHECK: acc.yield %[[ALLOC]] : !fir.ref<f32>
// CHECK: } copy {
// CHECK: ^bb0(%[[SRC:.*]]: !fir.ref<f32>, %[[DST:.*]]: !fir.ref<f32>):
// CHECK: %[[LOADED:.*]] = fir.load %[[SRC]] : !fir.ref<f32>
// CHECK: hlfir.assign %[[LOADED]] to %[[DST]] temporary_lhs : f32, !fir.ref<f32>
// CHECK: acc.terminator
// CHECK: }
// CHECK-LABEL: func.func @test_f32_scalar_in_parallel
func.func @test_f32_scalar_in_parallel() {
%scalar = fir.alloca f32 {bindc_name = "f32_var"}
acc.parallel {
%load = fir.load %scalar : !fir.ref<f32>
acc.yield
}
return
}
// CHECK: %[[FIRSTPRIV:.*]] = acc.firstprivate varPtr(%{{.*}} : !fir.ref<f32>) recipe(@firstprivatization_ref_f32) -> !fir.ref<f32> {implicit = true, name = "f32_var"}
// CHECK: acc.parallel firstprivate(%[[FIRSTPRIV]] : !fir.ref<f32>)
// -----
// CHECK-LABEL: acc.firstprivate.recipe @firstprivatization_ref_f64 : !fir.ref<f64> init {
// CHECK: ^bb0(%{{.*}}: !fir.ref<f64>):
// CHECK: %[[ALLOC:.*]] = fir.alloca f64
// CHECK: acc.yield %[[ALLOC]] : !fir.ref<f64>
// CHECK: } copy {
// CHECK: ^bb0(%[[SRC:.*]]: !fir.ref<f64>, %[[DST:.*]]: !fir.ref<f64>):
// CHECK: %[[LOADED:.*]] = fir.load %[[SRC]] : !fir.ref<f64>
// CHECK: hlfir.assign %[[LOADED]] to %[[DST]] temporary_lhs : f64, !fir.ref<f64>
// CHECK: acc.terminator
// CHECK: }
// CHECK-LABEL: func.func @test_f64_scalar_in_parallel
func.func @test_f64_scalar_in_parallel() {
%scalar = fir.alloca f64 {bindc_name = "f64_var"}
acc.parallel {
%load = fir.load %scalar : !fir.ref<f64>
acc.yield
}
return
}
// CHECK: %[[FIRSTPRIV:.*]] = acc.firstprivate varPtr(%{{.*}} : !fir.ref<f64>) recipe(@firstprivatization_ref_f64) -> !fir.ref<f64> {implicit = true, name = "f64_var"}
// CHECK: acc.parallel firstprivate(%[[FIRSTPRIV]] : !fir.ref<f64>)
// -----
// CHECK-LABEL: acc.firstprivate.recipe @firstprivatization_ref_l32 : !fir.ref<!fir.logical<4>> init {
// CHECK: ^bb0(%{{.*}}: !fir.ref<!fir.logical<4>>):
// CHECK: %[[ALLOC:.*]] = fir.alloca !fir.logical<4>
// CHECK: acc.yield %[[ALLOC]] : !fir.ref<!fir.logical<4>>
// CHECK: } copy {
// CHECK: ^bb0(%[[SRC:.*]]: !fir.ref<!fir.logical<4>>, %[[DST:.*]]: !fir.ref<!fir.logical<4>>):
// CHECK: %[[LOADED:.*]] = fir.load %[[SRC]] : !fir.ref<!fir.logical<4>>
// CHECK: hlfir.assign %[[LOADED]] to %[[DST]] temporary_lhs : !fir.logical<4>, !fir.ref<!fir.logical<4>>
// CHECK: acc.terminator
// CHECK: }
// CHECK-LABEL: func.func @test_logical_scalar_in_parallel
func.func @test_logical_scalar_in_parallel() {
%scalar = fir.alloca !fir.logical<4> {bindc_name = "logical_var"}
acc.parallel {
%load = fir.load %scalar : !fir.ref<!fir.logical<4>>
acc.yield
}
return
}
// CHECK: %[[FIRSTPRIV:.*]] = acc.firstprivate varPtr(%{{.*}} : !fir.ref<!fir.logical<4>>) recipe(@firstprivatization_ref_l32) -> !fir.ref<!fir.logical<4>> {implicit = true, name = "logical_var"}
// CHECK: acc.parallel firstprivate(%[[FIRSTPRIV]] : !fir.ref<!fir.logical<4>>)
// -----
// CHECK-LABEL: acc.firstprivate.recipe @firstprivatization_ref_z32 : !fir.ref<complex<f32>> init {
// CHECK: ^bb0(%{{.*}}: !fir.ref<complex<f32>>):
// CHECK: %[[ALLOC:.*]] = fir.alloca complex<f32>
// CHECK: acc.yield %[[ALLOC]] : !fir.ref<complex<f32>>
// CHECK: } copy {
// CHECK: ^bb0(%[[SRC:.*]]: !fir.ref<complex<f32>>, %[[DST:.*]]: !fir.ref<complex<f32>>):
// CHECK: %[[LOADED:.*]] = fir.load %[[SRC]] : !fir.ref<complex<f32>>
// CHECK: hlfir.assign %[[LOADED]] to %[[DST]] temporary_lhs : complex<f32>, !fir.ref<complex<f32>>
// CHECK: acc.terminator
// CHECK: }
// CHECK-LABEL: func.func @test_complex_scalar_in_parallel
func.func @test_complex_scalar_in_parallel() {
%scalar = fir.alloca complex<f32> {bindc_name = "complex_var"}
acc.parallel {
%load = fir.load %scalar : !fir.ref<complex<f32>>
acc.yield
}
return
}
// CHECK: %[[FIRSTPRIV:.*]] = acc.firstprivate varPtr(%{{.*}} : !fir.ref<complex<f32>>) recipe(@firstprivatization_ref_z32) -> !fir.ref<complex<f32>> {implicit = true, name = "complex_var"}
// CHECK: acc.parallel firstprivate(%[[FIRSTPRIV]] : !fir.ref<complex<f32>>)
// -----
// CHECK-LABEL: acc.firstprivate.recipe @firstprivatization_ref_z64 : !fir.ref<complex<f64>> init {
// CHECK: ^bb0(%{{.*}}: !fir.ref<complex<f64>>):
// CHECK: %[[ALLOC:.*]] = fir.alloca complex<f64>
// CHECK: acc.yield %[[ALLOC]] : !fir.ref<complex<f64>>
// CHECK: } copy {
// CHECK: ^bb0(%[[SRC:.*]]: !fir.ref<complex<f64>>, %[[DST:.*]]: !fir.ref<complex<f64>>):
// CHECK: %[[LOADED:.*]] = fir.load %[[SRC]] : !fir.ref<complex<f64>>
// CHECK: hlfir.assign %[[LOADED]] to %[[DST]] temporary_lhs : complex<f64>, !fir.ref<complex<f64>>
// CHECK: acc.terminator
// CHECK: }
// CHECK-LABEL: func.func @test_complex8_scalar_in_parallel
func.func @test_complex8_scalar_in_parallel() {
%scalar = fir.alloca complex<f64> {bindc_name = "complex8_var"}
acc.parallel {
%load = fir.load %scalar : !fir.ref<complex<f64>>
acc.yield
}
return
}
// CHECK: %[[FIRSTPRIV:.*]] = acc.firstprivate varPtr(%{{.*}} : !fir.ref<complex<f64>>) recipe(@firstprivatization_ref_z64) -> !fir.ref<complex<f64>> {implicit = true, name = "complex8_var"}
// CHECK: acc.parallel firstprivate(%[[FIRSTPRIV]] : !fir.ref<complex<f64>>)
// -----
// Test with serial construct
// CHECK-LABEL: func.func @test_i32_scalar_in_serial
func.func @test_i32_scalar_in_serial() {
%scalar = fir.alloca i32 {bindc_name = "serial_i32_var"}
acc.serial {
%load = fir.load %scalar : !fir.ref<i32>
acc.yield
}
return
}
// CHECK: %[[FIRSTPRIV:.*]] = acc.firstprivate varPtr(%{{.*}} : !fir.ref<i32>) recipe(@firstprivatization_ref_i32) -> !fir.ref<i32> {implicit = true, name = "serial_i32_var"}
// CHECK: acc.serial firstprivate(%[[FIRSTPRIV]] : !fir.ref<i32>)
// -----
// Test with serial construct and f64
// CHECK-LABEL: func.func @test_f64_scalar_in_serial
func.func @test_f64_scalar_in_serial() {
%scalar = fir.alloca f64 {bindc_name = "serial_f64_var"}
acc.serial {
%load = fir.load %scalar : !fir.ref<f64>
acc.yield
}
return
}
// CHECK: %[[FIRSTPRIV:.*]] = acc.firstprivate varPtr(%{{.*}} : !fir.ref<f64>) recipe(@firstprivatization_ref_f64) -> !fir.ref<f64> {implicit = true, name = "serial_f64_var"}
// CHECK: acc.serial firstprivate(%[[FIRSTPRIV]] : !fir.ref<f64>)
// -----
// Test i8 and i16 scalar types
// CHECK-LABEL: acc.firstprivate.recipe @firstprivatization_ref_i8 : !fir.ref<i8> init {
// CHECK: ^bb0(%{{.*}}: !fir.ref<i8>):
// CHECK: %[[ALLOC:.*]] = fir.alloca i8
// CHECK: acc.yield %[[ALLOC]] : !fir.ref<i8>
// CHECK: } copy {
// CHECK: ^bb0(%[[SRC:.*]]: !fir.ref<i8>, %[[DST:.*]]: !fir.ref<i8>):
// CHECK: %[[LOADED:.*]] = fir.load %[[SRC]] : !fir.ref<i8>
// CHECK: hlfir.assign %[[LOADED]] to %[[DST]] temporary_lhs : i8, !fir.ref<i8>
// CHECK: acc.terminator
// CHECK: }
// CHECK-LABEL: func.func @test_i8_scalar_in_parallel
func.func @test_i8_scalar_in_parallel() {
%scalar = fir.alloca i8 {bindc_name = "i8_var"}
acc.parallel {
%load = fir.load %scalar : !fir.ref<i8>
acc.yield
}
return
}
// CHECK: %[[FIRSTPRIV:.*]] = acc.firstprivate varPtr(%{{.*}} : !fir.ref<i8>) recipe(@firstprivatization_ref_i8) -> !fir.ref<i8> {implicit = true, name = "i8_var"}
// CHECK: acc.parallel firstprivate(%[[FIRSTPRIV]] : !fir.ref<i8>)
// -----
// CHECK-LABEL: acc.firstprivate.recipe @firstprivatization_ref_i16 : !fir.ref<i16> init {
// CHECK: ^bb0(%{{.*}}: !fir.ref<i16>):
// CHECK: %[[ALLOC:.*]] = fir.alloca i16
// CHECK: acc.yield %[[ALLOC]] : !fir.ref<i16>
// CHECK: } copy {
// CHECK: ^bb0(%[[SRC:.*]]: !fir.ref<i16>, %[[DST:.*]]: !fir.ref<i16>):
// CHECK: %[[LOADED:.*]] = fir.load %[[SRC]] : !fir.ref<i16>
// CHECK: hlfir.assign %[[LOADED]] to %[[DST]] temporary_lhs : i16, !fir.ref<i16>
// CHECK: acc.terminator
// CHECK: }
// CHECK-LABEL: func.func @test_i16_scalar_in_parallel
func.func @test_i16_scalar_in_parallel() {
%scalar = fir.alloca i16 {bindc_name = "i16_var"}
acc.parallel {
%load = fir.load %scalar : !fir.ref<i16>
acc.yield
}
return
}
// CHECK: %[[FIRSTPRIV:.*]] = acc.firstprivate varPtr(%{{.*}} : !fir.ref<i16>) recipe(@firstprivatization_ref_i16) -> !fir.ref<i16> {implicit = true, name = "i16_var"}
// CHECK: acc.parallel firstprivate(%[[FIRSTPRIV]] : !fir.ref<i16>)