738bab743b
Local vars, marked with pragma allocate, mustbe allocate by the call of
the runtime function and cannot be allocated as other local variables.
Instead, we allocate a space for the pointer in private record and store
the address, returned by kmpc_alloc call in this pointer.
So, for untied tasks
```
#pragma omp task untied
{
S s;
#pragma omp allocate(s) allocator(allocator)
s = x;
}
```
compiler generates something like this:
```
struct task_with_privates {
S *ptr;
};
void entry(task_with_privates *p) {
S *s = p->s;
switch(partid) {
case 1:
p->s = (S*)kmpc_alloc();
kmpc_omp_task();
br exit;
case 2:
*s = x;
kmpc_omp_task();
br exit;
case 2:
~S(s);
kmpc_free((void*)s);
br exit;
}
exit:
}
```
Reviewed By: jdoerfert
Differential Revision: https://reviews.llvm.org/D86558
112 lines
4.6 KiB
C++
112 lines
4.6 KiB
C++
// RUN: %clang_cc1 -verify -fopenmp -triple x86_64-apple-darwin10.6.0 -emit-llvm -o - %s | FileCheck %s
|
|
// RUN: %clang_cc1 -fopenmp -triple x86_64-apple-darwin10.6.0 -x c++ -std=c++11 -emit-pch -o %t %s
|
|
// RUN: %clang_cc1 -fopenmp -triple x86_64-apple-darwin10.6.0 -std=c++11 -include-pch %t -verify %s -emit-llvm -o - | FileCheck %s
|
|
// RUN: %clang_cc1 -verify -fopenmp -triple x86_64-unknown-linux-gnu -emit-llvm -o - %s | FileCheck %s
|
|
// RUN: %clang_cc1 -fopenmp -fnoopenmp-use-tls -triple x86_64-unknown-linux-gnu -x c++ -std=c++11 -emit-pch -o %t %s
|
|
// RUN: %clang_cc1 -fopenmp -fnoopenmp-use-tls -triple x86_64-unknown-linux-gnu -std=c++11 -include-pch %t -verify %s -emit-llvm -o - | FileCheck %s
|
|
|
|
// RUN: %clang_cc1 -verify -fopenmp-simd -triple x86_64-apple-darwin10.6.0 -emit-llvm -o - %s | FileCheck --check-prefix SIMD-ONLY0 %s
|
|
// RUN: %clang_cc1 -fopenmp-simd -triple x86_64-apple-darwin10.6.0 -x c++ -std=c++11 -emit-pch -o %t %s
|
|
// RUN: %clang_cc1 -fopenmp-simd -triple x86_64-apple-darwin10.6.0 -std=c++11 -include-pch %t -verify %s -emit-llvm -o - | FileCheck --check-prefix SIMD-ONLY0 %s
|
|
// RUN: %clang_cc1 -verify -fopenmp-simd -triple x86_64-unknown-linux-gnu -emit-llvm -o - %s | FileCheck --check-prefix SIMD-ONLY0 %s
|
|
// RUN: %clang_cc1 -fopenmp-simd -fnoopenmp-use-tls -triple x86_64-unknown-linux-gnu -x c++ -std=c++11 -emit-pch -o %t %s
|
|
// RUN: %clang_cc1 -fopenmp-simd -fnoopenmp-use-tls -triple x86_64-unknown-linux-gnu -std=c++11 -include-pch %t -verify %s -emit-llvm -o - | FileCheck --check-prefix SIMD-ONLY0 %s
|
|
// SIMD-ONLY0-NOT: {{__kmpc|__tgt}}
|
|
// expected-no-diagnostics
|
|
|
|
#ifndef HEADER
|
|
#define HEADER
|
|
|
|
enum omp_allocator_handle_t {
|
|
omp_null_allocator = 0,
|
|
omp_default_mem_alloc = 1,
|
|
omp_large_cap_mem_alloc = 2,
|
|
omp_const_mem_alloc = 3,
|
|
omp_high_bw_mem_alloc = 4,
|
|
omp_low_lat_mem_alloc = 5,
|
|
omp_cgroup_mem_alloc = 6,
|
|
omp_pteam_mem_alloc = 7,
|
|
omp_thread_mem_alloc = 8,
|
|
KMP_ALLOCATOR_MAX_HANDLE = __UINTPTR_MAX__
|
|
};
|
|
|
|
struct St{
|
|
int a;
|
|
};
|
|
|
|
struct St1{
|
|
int a;
|
|
static int b;
|
|
#pragma omp allocate(b) allocator(omp_default_mem_alloc)
|
|
} d;
|
|
|
|
int a, b, c;
|
|
#pragma omp allocate(a) allocator(omp_large_cap_mem_alloc)
|
|
#pragma omp allocate(b) allocator(omp_const_mem_alloc)
|
|
#pragma omp allocate(d, c) allocator(omp_high_bw_mem_alloc)
|
|
|
|
template <class T>
|
|
struct ST {
|
|
static T m;
|
|
#pragma omp allocate(m) allocator(omp_low_lat_mem_alloc)
|
|
};
|
|
|
|
template <class T> T foo() {
|
|
T v;
|
|
#pragma omp allocate(v) allocator(omp_cgroup_mem_alloc)
|
|
v = ST<T>::m;
|
|
return v;
|
|
}
|
|
|
|
namespace ns{
|
|
int a;
|
|
}
|
|
#pragma omp allocate(ns::a) allocator(omp_pteam_mem_alloc)
|
|
|
|
// CHECK-NOT: call {{.+}} {{__kmpc_alloc|__kmpc_free}}
|
|
|
|
// CHECK-LABEL: @main
|
|
int main () {
|
|
static int a;
|
|
#pragma omp allocate(a) allocator(omp_thread_mem_alloc)
|
|
a=2;
|
|
// CHECK-NOT: {{__kmpc_alloc|__kmpc_free}}
|
|
// CHECK: alloca double,
|
|
// CHECK-NOT: {{__kmpc_alloc|__kmpc_free}}
|
|
double b = 3;
|
|
#pragma omp allocate(b)
|
|
return (foo<int>());
|
|
}
|
|
|
|
// CHECK: define {{.*}}i32 @{{.+}}foo{{.+}}()
|
|
// CHECK: [[GTID:%.+]] = call i32 @__kmpc_global_thread_num(%struct.ident_t* @{{.+}})
|
|
// CHECK-NEXT: [[V_VOID_ADDR:%.+]] = call i8* @__kmpc_alloc(i32 [[GTID]], i64 4, i8* inttoptr (i64 6 to i8*))
|
|
// CHECK-NEXT: [[V_ADDR:%.+]] = bitcast i8* [[V_VOID_ADDR]] to i32*
|
|
// CHECK-NOT: {{__kmpc_alloc|__kmpc_free}}
|
|
// CHECK: store i32 %{{.+}}, i32* [[V_ADDR]],
|
|
// CHECK-NEXT: [[V_VAL:%.+]] = load i32, i32* [[V_ADDR]],
|
|
// CHECK-NEXT: [[V_VOID_ADDR:%.+]] = bitcast i32* [[V_ADDR]] to i8*
|
|
// CHECK-NEXT: call void @__kmpc_free(i32 [[GTID]], i8* [[V_VOID_ADDR]], i8* inttoptr (i64 6 to i8*))
|
|
// CHECK-NOT: {{__kmpc_alloc|__kmpc_free}}
|
|
// CHECK: ret i32 [[V_VAL]]
|
|
|
|
// CHECK-NOT: call {{.+}} {{__kmpc_alloc|__kmpc_free}}
|
|
extern template int ST<int>::m;
|
|
|
|
// CHECK: define void @{{.+}}bar{{.+}}(i32 %{{.+}}, float* {{.+}})
|
|
void bar(int a, float &z) {
|
|
// CHECK: [[A_VOID_PTR:%.+]] = call i8* @__kmpc_alloc(i32 [[GTID:%.+]], i64 4, i8* inttoptr (i64 1 to i8*))
|
|
// CHECK: [[A_ADDR:%.+]] = bitcast i8* [[A_VOID_PTR]] to i32*
|
|
// CHECK: store i32 %{{.+}}, i32* [[A_ADDR]],
|
|
// CHECK: [[Z_VOID_PTR:%.+]] = call i8* @__kmpc_alloc(i32 [[GTID]], i64 8, i8* inttoptr (i64 1 to i8*))
|
|
// CHECK: [[Z_ADDR:%.+]] = bitcast i8* [[Z_VOID_PTR]] to float**
|
|
// CHECK: store float* %{{.+}}, float** [[Z_ADDR]],
|
|
#pragma omp allocate(a,z) allocator(omp_default_mem_alloc)
|
|
// CHECK-NEXT: [[Z_VOID_PTR:%.+]] = bitcast float** [[Z_ADDR]] to i8*
|
|
// CHECK: call void @__kmpc_free(i32 [[GTID]], i8* [[Z_VOID_PTR]], i8* inttoptr (i64 1 to i8*))
|
|
// CHECK-NEXT: [[A_VOID_PTR:%.+]] = bitcast i32* [[A_ADDR]] to i8*
|
|
// CHECK: call void @__kmpc_free(i32 [[GTID]], i8* [[A_VOID_PTR]], i8* inttoptr (i64 1 to i8*))
|
|
// CHECK: ret void
|
|
}
|
|
#endif
|