derek/TheRock
1
0 Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity
Files
1a182e2dac58879f215c0fad0f5f78c76dcafe56
TheRock/compiler/amd-llvm/flang/test/Fir/omp_target_allocmem_freemem.fir
T
derek.holloway 7502bf64a4
CodeQL / Analyze (actions) (push) Canceled after 0s
Details
CodeQL / Analyze (python) (push) Canceled after 0s
Details
pre-commit / pre-commit (push) Canceled after 0s
Details
Multi-Arch CI / setup (push) Canceled after 0s
Details
Publish build_manylinux_x86_64 images / publish_build_manylinux_x86_64 (push) Canceled after 0s
Details
Setup (Multi-Arch) / setup (push) Canceled after 0s
Details
Multi-Arch CI / Linux::${{ fromJSON(needs.setup.outputs.linux_build_config || '{}').build_variant_label || 'skip' }} (push) Canceled after 0s
Details
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Details
Publish no_rocm_image_ubi10 image / publish_no_rocm_image_ubi10 (push) Canceled after 0s
Details
Multi-Arch CI / Windows::${{ fromJSON(needs.setup.outputs.windows_build_config || '{}').build_variant_label || 'skip' }} (push) Canceled after 0s
Details
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Details
Multi-Arch CI / Manifest Diff (push) Canceled after 0s
Details
TheRock Manifest Diff Report / Generate Manifest Diff Report (push) Canceled after 0s
Details
Multi-Arch CI / CI Summary (push) Canceled after 0s
Details
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04 (push) Canceled after 0s
Details
Publish no_rocm_image_ubuntu24_04_ocl_rt images / publish_no_rocm_image_ubuntu24_04_ocl_rt (push) Canceled after 0s
Details
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Details
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Details
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_rocgdb (push) Canceled after 0s
Details
Publish no_rocm_image_ubuntu24_04_rocgdb images / publish_no_rocm_image_ubuntu24_04_rocgdb (push) Canceled after 0s
Details
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_ocl_rt (push) Canceled after 0s
Details
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Details
Unit Tests / Unit Tests :: ubuntu-24.04 (push) Canceled after 0s
Details
Unit Tests / Unit Tests :: windows-2022 (push) Canceled after 0s
Details
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_media (push) Canceled after 0s
Details
Unit Tests / Unit Tests Summary (push) Canceled after 0s
Details
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_openmpi (push) Canceled after 0s
Details
Flatten subprojects into a single base monorepo
2026-09-02 16:34:47 -07:00

295 lines
13 KiB
Plaintext
Raw Blame History

// RUN: %flang_fc1 -emit-llvm %s -o - | FileCheck %s
// UNSUPPORTED: system-windows
// Disabled on 32-bit targets due to the additional `trunc` opcodes required
// UNSUPPORTED: target-x86
// UNSUPPORTED: target=sparc-{{.*}}
// UNSUPPORTED: target=sparcel-{{.*}}
// CHECK-LABEL: define void @omp_target_allocmem_scalar_nonchar() {
// CHECK-NEXT: [[TMP1:%.*]] = call ptr @omp_target_alloc(i64 4, i32 0)
// CHECK-NEXT: [[TMP2:%.*]] = ptrtoint ptr [[TMP1]] to i64
// CHECK-NEXT: [[TMP3:%.*]] = inttoptr i64 [[TMP2]] to ptr
// CHECK-NEXT: call void @omp_target_free(ptr [[TMP3]], i32 0)
// CHECK-NEXT: ret void
func.func @omp_target_allocmem_scalar_nonchar() -> () {
%device = arith.constant 0 : i32
%1 = omp.target_allocmem %device : i32, i32
omp.target_freemem %device, %1 : i32, i64
return
}
// CHECK-LABEL: define void @omp_target_allocmem_scalars_nonchar() {
// CHECK-NEXT: [[TMP1:%.*]] = call ptr @omp_target_alloc(i64 400, i32 0)
// CHECK-NEXT: [[TMP2:%.*]] = ptrtoint ptr [[TMP1]] to i64
// CHECK-NEXT: [[TMP3:%.*]] = inttoptr i64 [[TMP2]] to ptr
// CHECK-NEXT: call void @omp_target_free(ptr [[TMP3]], i32 0)
// CHECK-NEXT: ret void
func.func @omp_target_allocmem_scalars_nonchar() -> () {
%device = arith.constant 0 : i32
%0 = arith.constant 100 : index
%1 = omp.target_allocmem %device : i32, i32, %0
omp.target_freemem %device, %1 : i32, i64
return
}
// CHECK-LABEL: define void @omp_target_allocmem_scalar_char() {
// CHECK-NEXT: [[TMP1:%.*]] = call ptr @omp_target_alloc(i64 10, i32 0)
// CHECK-NEXT: [[TMP2:%.*]] = ptrtoint ptr [[TMP1]] to i64
// CHECK-NEXT: [[TMP3:%.*]] = inttoptr i64 [[TMP2]] to ptr
// CHECK-NEXT: call void @omp_target_free(ptr [[TMP3]], i32 0)
// CHECK-NEXT: ret void
func.func @omp_target_allocmem_scalar_char() -> () {
%device = arith.constant 0 : i32
%1 = omp.target_allocmem %device : i32, !fir.char<1,10>
omp.target_freemem %device, %1 : i32, i64
return
}
// CHECK-LABEL: define void @omp_target_allocmem_scalar_char_kind() {
// CHECK-NEXT: [[TMP1:%.*]] = call ptr @omp_target_alloc(i64 20, i32 0)
// CHECK-NEXT: [[TMP2:%.*]] = ptrtoint ptr [[TMP1]] to i64
// CHECK-NEXT: [[TMP3:%.*]] = inttoptr i64 [[TMP2]] to ptr
// CHECK-NEXT: call void @omp_target_free(ptr [[TMP3]], i32 0)
// CHECK-NEXT: ret void
func.func @omp_target_allocmem_scalar_char_kind() -> () {
%device = arith.constant 0 : i32
%1 = omp.target_allocmem %device : i32, !fir.char<2,10>
omp.target_freemem %device, %1 : i32, i64
return
}
// CHECK-LABEL: define void @omp_target_allocmem_scalar_dynchar(
// CHECK-SAME: i32 [[TMP0:%.*]]) {
// CHECK-NEXT: [[TMP2:%.*]] = sext i32 [[TMP0]] to i64
// CHECK-NEXT: [[TMP3:%.*]] = mul i64 1, [[TMP2]]
// CHECK-NEXT: [[TMP4:%.*]] = mul i64 1, [[TMP3]]
// CHECK-NEXT: [[TMP5:%.*]] = call ptr @omp_target_alloc(i64 [[TMP4]], i32 0)
// CHECK-NEXT: [[TMP6:%.*]] = ptrtoint ptr [[TMP5]] to i64
// CHECK-NEXT: [[TMP7:%.*]] = inttoptr i64 [[TMP6]] to ptr
// CHECK-NEXT: call void @omp_target_free(ptr [[TMP7]], i32 0)
// CHECK-NEXT: ret void
func.func @omp_target_allocmem_scalar_dynchar(%l : i32) -> () {
%device = arith.constant 0 : i32
%1 = omp.target_allocmem %device : i32, !fir.char<1,?>(%l : i32)
omp.target_freemem %device, %1 : i32, i64
return
}
// CHECK-LABEL: define void @omp_target_allocmem_scalar_dynchar_kind(
// CHECK-SAME: i32 [[TMP0:%.*]]) {
// CHECK-NEXT: [[TMP2:%.*]] = sext i32 [[TMP0]] to i64
// CHECK-NEXT: [[TMP3:%.*]] = mul i64 2, [[TMP2]]
// CHECK-NEXT: [[TMP4:%.*]] = mul i64 1, [[TMP3]]
// CHECK-NEXT: [[TMP5:%.*]] = call ptr @omp_target_alloc(i64 [[TMP4]], i32 0)
// CHECK-NEXT: [[TMP6:%.*]] = ptrtoint ptr [[TMP5]] to i64
// CHECK-NEXT: [[TMP7:%.*]] = inttoptr i64 [[TMP6]] to ptr
// CHECK-NEXT: call void @omp_target_free(ptr [[TMP7]], i32 0)
// CHECK-NEXT: ret void
func.func @omp_target_allocmem_scalar_dynchar_kind(%l : i32) -> () {
%device = arith.constant 0 : i32
%1 = omp.target_allocmem %device : i32, !fir.char<2,?>(%l : i32)
omp.target_freemem %device, %1 : i32, i64
return
}
// CHECK-LABEL: define void @omp_target_allocmem_array_of_nonchar() {
// CHECK-NEXT: [[TMP1:%.*]] = call ptr @omp_target_alloc(i64 36, i32 0)
// CHECK-NEXT: [[TMP2:%.*]] = ptrtoint ptr [[TMP1]] to i64
// CHECK-NEXT: [[TMP3:%.*]] = inttoptr i64 [[TMP2]] to ptr
// CHECK-NEXT: call void @omp_target_free(ptr [[TMP3]], i32 0)
// CHECK-NEXT: ret void
func.func @omp_target_allocmem_array_of_nonchar() -> () {
%device = arith.constant 0 : i32
%1 = omp.target_allocmem %device : i32, !fir.array<3x3xi32>
omp.target_freemem %device, %1 : i32, i64
return
}
// CHECK-LABEL: define void @omp_target_allocmem_array_of_char() {
// CHECK-NEXT: [[TMP1:%.*]] = call ptr @omp_target_alloc(i64 90, i32 0)
// CHECK-NEXT: [[TMP2:%.*]] = ptrtoint ptr [[TMP1]] to i64
// CHECK-NEXT: [[TMP3:%.*]] = inttoptr i64 [[TMP2]] to ptr
// CHECK-NEXT: call void @omp_target_free(ptr [[TMP3]], i32 0)
// CHECK-NEXT: ret void
func.func @omp_target_allocmem_array_of_char() -> () {
%device = arith.constant 0 : i32
%1 = omp.target_allocmem %device : i32, !fir.array<3x3x!fir.char<1,10>>
omp.target_freemem %device, %1 : i32, i64
return
}
// CHECK-LABEL: define void @omp_target_allocmem_array_of_dynchar(
// CHECK-SAME: i32 [[TMP0:%.*]]) {
// CHECK-NEXT: [[TMP2:%.*]] = sext i32 [[TMP0]] to i64
// CHECK-NEXT: [[TMP3:%.*]] = mul i64 9, [[TMP2]]
// CHECK-NEXT: [[TMP4:%.*]] = mul i64 1, [[TMP3]]
// CHECK-NEXT: [[TMP5:%.*]] = call ptr @omp_target_alloc(i64 [[TMP4]], i32 0)
// CHECK-NEXT: [[TMP6:%.*]] = ptrtoint ptr [[TMP5]] to i64
// CHECK-NEXT: [[TMP7:%.*]] = inttoptr i64 [[TMP6]] to ptr
// CHECK-NEXT: call void @omp_target_free(ptr [[TMP7]], i32 0)
// CHECK-NEXT: ret void
func.func @omp_target_allocmem_array_of_dynchar(%l: i32) -> () {
%device = arith.constant 0 : i32
%1 = omp.target_allocmem %device : i32, !fir.array<3x3x!fir.char<1,?>>(%l : i32)
omp.target_freemem %device, %1 : i32, i64
return
}
// CHECK-LABEL: define void @omp_target_allocmem_dynarray_of_nonchar(
// CHECK-SAME: i64 [[TMP0:%.*]]) {
// CHECK-NEXT: [[TMP2:%.*]] = mul i64 12, [[TMP0]]
// CHECK-NEXT: [[TMP3:%.*]] = mul i64 1, [[TMP2]]
// CHECK-NEXT: [[TMP4:%.*]] = call ptr @omp_target_alloc(i64 [[TMP3]], i32 0)
// CHECK-NEXT: [[TMP5:%.*]] = ptrtoint ptr [[TMP4]] to i64
// CHECK-NEXT: [[TMP6:%.*]] = inttoptr i64 [[TMP5]] to ptr
// CHECK-NEXT: call void @omp_target_free(ptr [[TMP6]], i32 0)
// CHECK-NEXT: ret void
func.func @omp_target_allocmem_dynarray_of_nonchar(%e: index) -> () {
%device = arith.constant 0 : i32
%1 = omp.target_allocmem %device : i32, !fir.array<3x?xi32>, %e
omp.target_freemem %device, %1 : i32, i64
return
}
// CHECK-LABEL: define void @omp_target_allocmem_dynarray_of_nonchar2(
// CHECK-SAME: i64 [[TMP0:%.*]]) {
// CHECK-NEXT: [[TMP2:%.*]] = mul i64 4, [[TMP0]]
// CHECK-NEXT: [[TMP3:%.*]] = mul i64 [[TMP2]], [[TMP0]]
// CHECK-NEXT: [[TMP4:%.*]] = mul i64 1, [[TMP3]]
// CHECK-NEXT: [[TMP5:%.*]] = call ptr @omp_target_alloc(i64 [[TMP4]], i32 0)
// CHECK-NEXT: [[TMP6:%.*]] = ptrtoint ptr [[TMP5]] to i64
// CHECK-NEXT: [[TMP7:%.*]] = inttoptr i64 [[TMP6]] to ptr
// CHECK-NEXT: call void @omp_target_free(ptr [[TMP7]], i32 0)
// CHECK-NEXT: ret void
func.func @omp_target_allocmem_dynarray_of_nonchar2(%e: index) -> () {
%device = arith.constant 0 : i32
%1 = omp.target_allocmem %device : i32, !fir.array<?x?xi32>, %e, %e
omp.target_freemem %device, %1 : i32, i64
return
}
// CHECK-LABEL: define void @omp_target_allocmem_dynarray_of_char(
// CHECK-SAME: i64 [[TMP0:%.*]]) {
// CHECK-NEXT: [[TMP2:%.*]] = mul i64 60, [[TMP0]]
// CHECK-NEXT: [[TMP3:%.*]] = mul i64 1, [[TMP2]]
// CHECK-NEXT: [[TMP4:%.*]] = call ptr @omp_target_alloc(i64 [[TMP3]], i32 0)
// CHECK-NEXT: [[TMP5:%.*]] = ptrtoint ptr [[TMP4]] to i64
// CHECK-NEXT: [[TMP6:%.*]] = inttoptr i64 [[TMP5]] to ptr
// CHECK-NEXT: call void @omp_target_free(ptr [[TMP6]], i32 0)
// CHECK-NEXT: ret void
func.func @omp_target_allocmem_dynarray_of_char(%e : index) -> () {
%device = arith.constant 0 : i32
%1 = omp.target_allocmem %device : i32, !fir.array<3x?x!fir.char<2,10>>, %e
omp.target_freemem %device, %1 : i32, i64
return
}
// CHECK-LABEL: define void @omp_target_allocmem_dynarray_of_char2(
// CHECK-SAME: i64 [[TMP0:%.*]]) {
// CHECK-NEXT: [[TMP2:%.*]] = mul i64 20, [[TMP0]]
// CHECK-NEXT: [[TMP3:%.*]] = mul i64 [[TMP2]], [[TMP0]]
// CHECK-NEXT: [[TMP4:%.*]] = mul i64 1, [[TMP3]]
// CHECK-NEXT: [[TMP5:%.*]] = call ptr @omp_target_alloc(i64 [[TMP4]], i32 0)
// CHECK-NEXT: [[TMP6:%.*]] = ptrtoint ptr [[TMP5]] to i64
// CHECK-NEXT: [[TMP7:%.*]] = inttoptr i64 [[TMP6]] to ptr
// CHECK-NEXT: call void @omp_target_free(ptr [[TMP7]], i32 0)
// CHECK-NEXT: ret void
func.func @omp_target_allocmem_dynarray_of_char2(%e : index) -> () {
%device = arith.constant 0 : i32
%1 = omp.target_allocmem %device : i32, !fir.array<?x?x!fir.char<2,10>>, %e, %e
omp.target_freemem %device, %1 : i32, i64
return
}
// CHECK-LABEL: define void @omp_target_allocmem_dynarray_of_dynchar(
// CHECK-SAME: i32 [[TMP0:%.*]], i64 [[TMP1:%.*]]) {
// CHECK-NEXT: [[TMP3:%.*]] = sext i32 [[TMP0]] to i64
// CHECK-NEXT: [[TMP4:%.*]] = mul i64 6, [[TMP3]]
// CHECK-NEXT: [[TMP5:%.*]] = mul i64 [[TMP4]], [[TMP1]]
// CHECK-NEXT: [[TMP6:%.*]] = mul i64 1, [[TMP5]]
// CHECK-NEXT: [[TMP7:%.*]] = call ptr @omp_target_alloc(i64 [[TMP6]], i32 0)
// CHECK-NEXT: [[TMP8:%.*]] = ptrtoint ptr [[TMP7]] to i64
// CHECK-NEXT: [[TMP9:%.*]] = inttoptr i64 [[TMP8]] to ptr
// CHECK-NEXT: call void @omp_target_free(ptr [[TMP9]], i32 0)
// CHECK-NEXT: ret void
func.func @omp_target_allocmem_dynarray_of_dynchar(%l: i32, %e : index) -> () {
%device = arith.constant 0 : i32
%1 = omp.target_allocmem %device : i32, !fir.array<3x?x!fir.char<2,?>>(%l : i32), %e
omp.target_freemem %device, %1 : i32, i64
return
}
// CHECK-LABEL: define void @omp_target_allocmem_dynarray_of_dynchar2(
// CHECK-SAME: i32 [[TMP0:%.*]], i64 [[TMP1:%.*]]) {
// CHECK-NEXT: [[TMP3:%.*]] = sext i32 [[TMP0]] to i64
// CHECK-NEXT: [[TMP4:%.*]] = mul i64 2, [[TMP3]]
// CHECK-NEXT: [[TMP5:%.*]] = mul i64 [[TMP4]], [[TMP1]]
// CHECK-NEXT: [[TMP6:%.*]] = mul i64 [[TMP5]], [[TMP1]]
// CHECK-NEXT: [[TMP7:%.*]] = mul i64 1, [[TMP6]]
// CHECK-NEXT: [[TMP8:%.*]] = call ptr @omp_target_alloc(i64 [[TMP7]], i32 0)
// CHECK-NEXT: [[TMP9:%.*]] = ptrtoint ptr [[TMP8]] to i64
// CHECK-NEXT: [[TMP10:%.*]] = inttoptr i64 [[TMP9]] to ptr
// CHECK-NEXT: call void @omp_target_free(ptr [[TMP10]], i32 0)
// CHECK-NEXT: ret void
func.func @omp_target_allocmem_dynarray_of_dynchar2(%l: i32, %e : index) -> () {
%device = arith.constant 0 : i32
%1 = omp.target_allocmem %device : i32, !fir.array<?x?x!fir.char<2,?>>(%l : i32), %e, %e
omp.target_freemem %device, %1 : i32, i64
return
}
// CHECK-LABEL: define void @omp_target_allocmem_array_with_holes_nonchar(
// CHECK-SAME: i64 [[TMP0:%.*]], i64 [[TMP1:%.*]]) {
// CHECK-NEXT: [[TMP3:%.*]] = mul i64 240, [[TMP0]]
// CHECK-NEXT: [[TMP4:%.*]] = mul i64 [[TMP3]], [[TMP1]]
// CHECK-NEXT: [[TMP5:%.*]] = mul i64 1, [[TMP4]]
// CHECK-NEXT: [[TMP6:%.*]] = call ptr @omp_target_alloc(i64 [[TMP5]], i32 0)
// CHECK-NEXT: [[TMP7:%.*]] = ptrtoint ptr [[TMP6]] to i64
// CHECK-NEXT: [[TMP8:%.*]] = inttoptr i64 [[TMP7]] to ptr
// CHECK-NEXT: call void @omp_target_free(ptr [[TMP8]], i32 0)
// CHECK-NEXT: ret void
func.func @omp_target_allocmem_array_with_holes_nonchar(%0 : index, %1 : index) -> () {
%device = arith.constant 0 : i32
%2 = omp.target_allocmem %device : i32, !fir.array<4x?x3x?x5xi32>, %0, %1
omp.target_freemem %device, %2 : i32, i64
return
}
// CHECK-LABEL: define void @omp_target_allocmem_array_with_holes_char(
// CHECK-SAME: i64 [[TMP0:%.*]]) {
// CHECK-NEXT: [[TMP2:%.*]] = mul i64 240, [[TMP0]]
// CHECK-NEXT: [[TMP3:%.*]] = mul i64 1, [[TMP2]]
// CHECK-NEXT: [[TMP4:%.*]] = call ptr @omp_target_alloc(i64 [[TMP3]], i32 0)
// CHECK-NEXT: [[TMP5:%.*]] = ptrtoint ptr [[TMP4]] to i64
// CHECK-NEXT: [[TMP6:%.*]] = inttoptr i64 [[TMP5]] to ptr
// CHECK-NEXT: call void @omp_target_free(ptr [[TMP6]], i32 0)
// CHECK-NEXT: ret void
func.func @omp_target_allocmem_array_with_holes_char(%e: index) -> () {
%device = arith.constant 0 : i32
%1 = omp.target_allocmem %device : i32, !fir.array<3x?x4x!fir.char<2,10>>, %e
omp.target_freemem %device, %1 : i32, i64
return
}
// CHECK-LABEL: define void @omp_target_allocmem_array_with_holes_dynchar(
// CHECK-SAME: i64 [[TMP0:%.*]], i64 [[TMP1:%.*]]) {
// CHECK-NEXT: [[TMP3:%.*]] = mul i64 24, [[TMP0]]
// CHECK-NEXT: [[TMP4:%.*]] = mul i64 [[TMP3]], [[TMP1]]
// CHECK-NEXT: [[TMP5:%.*]] = mul i64 1, [[TMP4]]
// CHECK-NEXT: [[TMP6:%.*]] = call ptr @omp_target_alloc(i64 [[TMP5]], i32 0)
// CHECK-NEXT: [[TMP7:%.*]] = ptrtoint ptr [[TMP6]] to i64
// CHECK-NEXT: [[TMP8:%.*]] = inttoptr i64 [[TMP7]] to ptr
// CHECK-NEXT: call void @omp_target_free(ptr [[TMP8]], i32 0)
// CHECK-NEXT: ret void
func.func @omp_target_allocmem_array_with_holes_dynchar(%arg0: index, %arg1: index) -> () {
%device = arith.constant 0 : i32
%1 = omp.target_allocmem %device : i32, !fir.array<3x?x4x!fir.char<2,?>>(%arg0 : index), %arg1
omp.target_freemem %device, %1 : i32, i64
return
}
Reference in new issue View Git Blame Copy Permalink