Skip to content

[OpenMP][Offload] Restore __kmpc_* function signatures #156104

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 1 commit into from
Sep 4, 2025
Merged

[OpenMP][Offload] Restore __kmpc_* function signatures #156104

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
47 changes: 36 additions & 11 deletions offload/DeviceRTL/src/Parallelism.cpp
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -103,11 +103,10 @@ uint32_t determineNumberOfThreads(int32_t NumThreadsClause,

extern "C" {

[[clang::always_inline]] void
__kmpc_parallel_spmd(IdentTy *ident, int32_t num_threads, void *fn, void **args,
const int64_t nargs, int32_t nt_strict = false,
int32_t nt_severity = severity_fatal,
const char *nt_message = nullptr) {
[[clang::always_inline]] void __kmpc_parallel_spmd_impl(
IdentTy *ident, int32_t num_threads, void *fn, void **args,
const int64_t nargs, int32_t nt_strict = false,
int32_t nt_severity = severity_fatal, const char *nt_message = nullptr) {
uint32_t TId = mapping::getThreadIdInBlock();
uint32_t NumThreads =
determineNumberOfThreads(num_threads, nt_strict, nt_severity, nt_message);
Expand Down Expand Up @@ -163,7 +162,22 @@ __kmpc_parallel_spmd(IdentTy *ident, int32_t num_threads, void *fn, void **args,
return;
}

[[clang::always_inline]] void __kmpc_parallel_51(
[[clang::always_inline]] void __kmpc_parallel_spmd(IdentTy *ident,
int32_t num_threads,
void *fn, void **args,
const int64_t nargs) {
return __kmpc_parallel_spmd_impl(ident, num_threads, fn, args, nargs);
}

[[clang::always_inline]] void __kmpc_parallel_spmd_60(
IdentTy *ident, int32_t num_threads, void *fn, void **args,
const int64_t nargs, int32_t nt_strict = false,
int32_t nt_severity = severity_fatal, const char *nt_message = nullptr) {
return __kmpc_parallel_spmd_impl(ident, num_threads, fn, args, nargs,
nt_strict, nt_severity, nt_message);
}

[[clang::always_inline]] void __kmpc_parallel_impl(
IdentTy *ident, int32_t, int32_t if_expr, int32_t num_threads,
int proc_bind, void *fn, void *wrapper_fn, void **args, int64_t nargs,
int32_t nt_strict = false, int32_t nt_severity = severity_fatal,
Expand Down Expand Up @@ -198,8 +212,11 @@ __kmpc_parallel_spmd(IdentTy *ident, int32_t num_threads, void *fn, void **args,
// This was moved to its own routine so it could be called directly
// in certain situations to avoid resource consumption of unused
// logic in parallel_51.
__kmpc_parallel_spmd(ident, num_threads, fn, args, nargs, nt_strict,
nt_severity, nt_message);
if (nt_strict)
__kmpc_parallel_spmd(ident, num_threads, fn, args, nargs);
else
__kmpc_parallel_spmd_60(ident, num_threads, fn, args, nargs, nt_strict,
nt_severity, nt_message);

return;
}
Expand Down Expand Up @@ -308,14 +325,22 @@ __kmpc_parallel_spmd(IdentTy *ident, int32_t num_threads, void *fn, void **args,
__kmpc_end_sharing_variables();
}

[[clang::always_inline]] void
__kmpc_parallel_51(IdentTy *ident, int32_t id, int32_t if_expr,
int32_t num_threads, int proc_bind, void *fn,
void *wrapper_fn, void **args, int64_t nargs) {
return __kmpc_parallel_impl(ident, id, if_expr, num_threads, proc_bind, fn,
wrapper_fn, args, nargs);
}

[[clang::always_inline]] void __kmpc_parallel_60(
IdentTy *ident, int32_t id, int32_t if_expr, int32_t num_threads,
int proc_bind, void *fn, void *wrapper_fn, void **args, int64_t nargs,
int32_t nt_strict = false, int32_t nt_severity = severity_fatal,
const char *nt_message = nullptr) {
return __kmpc_parallel_51(ident, id, if_expr, num_threads, proc_bind, fn,
wrapper_fn, args, nargs, nt_strict, nt_severity,
nt_message);
return __kmpc_parallel_impl(ident, id, if_expr, num_threads, proc_bind, fn,
wrapper_fn, args, nargs, nt_strict, nt_severity,
nt_message);
}

[[clang::noinline]] bool __kmpc_kernel_parallel(ParallelRegionFnTy *WorkFn) {
Expand Down
Loading