[OpenMP] Introduce the keepAlive function into the old device RT
Reviewed By: ye-luo Differential Revision: https://reviews.llvm.org/D113110
This commit is contained in:
parent
93bebdc78f
commit
ab9f3f5d25
|
@ -226,4 +226,14 @@ void __kmp_invoke_microtask(kmp_int32 global_tid, kmp_int32 bound_tid, void *fn,
|
|||
}
|
||||
}
|
||||
|
||||
namespace _OMP {
|
||||
/// Helper to keep code alive without introducing a performance penalty.
|
||||
__attribute__((used, weak, optnone)) void keepAlive() {
|
||||
__kmpc_get_hardware_thread_id_in_block();
|
||||
__kmpc_get_hardware_num_threads_in_block();
|
||||
__kmpc_barrier_simple_spmd(nullptr, 0);
|
||||
__kmpc_barrier_simple_generic(nullptr, 0);
|
||||
}
|
||||
} // namespace _OMP
|
||||
|
||||
#pragma omp end declare target
|
||||
|
|
Loading…
Reference in New Issue