Skip to content

[OpenMP][Offload] Add support for lambdas with debug conditions #172573

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
adurang merged 3 commits into from
Dec 17, 2025
+45 −9
Merged

[OpenMP][Offload] Add support for lambdas with debug conditions #172573

Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
3c26949
[OpenMP][Offload] Add support for lambdas with debug conditions
adurang Dec 12, 2025
1f55377
fixes
adurang Dec 16, 2025
48d3105
format
adurang Dec 16, 2025
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
54 changes: 45 additions & 9 deletions offload/include/Shared/Debug.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -432,15 +432,21 @@ static inline raw_ostream &operator<<(raw_ostream &Os,

// helper templates to support lambdas with different number of arguments
template <typename LambdaTy> struct LambdaHelper {
template <typename T, typename = std::void_t<>>
struct has_two_args : std::false_type {};
template <typename T>
struct has_two_args<T,
std::void_t<decltype(std::declval<T>().operator()(1, 2))>>
: std::true_type {};
template <typename FuncTy, typename RetTy, typename... Args>
static constexpr size_t CountArgs(RetTy (FuncTy::*)(Args...)) {
return sizeof...(Args);
}
template <typename FuncTy, typename RetTy, typename... Args>
static constexpr size_t CountArgs(RetTy (FuncTy::*)(Args...) const) {
return sizeof...(Args);
}

static constexpr size_t NArgs = CountArgs(&LambdaTy::operator());
};

template <typename LambdaTy> struct LambdaOs : public LambdaHelper<LambdaTy> {
static void dispatch(LambdaTy func, llvm::raw_ostream &Os, uint32_t Level) {
if constexpr (has_two_args<LambdaTy>::value)
if constexpr (LambdaHelper<LambdaTy>::NArgs == 2)
Copy link
Contributor

@abhinavgaba abhinavgaba Dec 16, 2025

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This and line 488 are the only new changes vs #172107, and they fix the build failure I saw locally.

func(Os, Level);
else
func(Os);
Expand All @@ -457,8 +463,8 @@ template <typename LambdaTy> struct LambdaHelper {
RealLevel, /*ShouldPrefixNextString=*/true, \
/*ShouldEmitNewLineOnDestruction=*/true}; \
auto F = Callback; \
::llvm::offload::debug::LambdaHelper<decltype(F)>::dispatch(F, OS, \
RealLevel); \
::llvm::offload::debug::LambdaOs<decltype(F)>::dispatch(F, OS, \
RealLevel); \
} \
}

Expand All @@ -476,6 +482,33 @@ template <typename LambdaTy> struct LambdaHelper {
#define ODBG_OS(...) \
ODBG_OS_SELECT(__VA_ARGS__ __VA_OPT__(, ) 3, 2, 1)(__VA_ARGS__)

// helper templates to support lambdas with different number of arguments
template <typename LambdaTy> struct LambdaIf : public LambdaHelper<LambdaTy> {
static void dispatch(LambdaTy func, uint32_t Level) {
if constexpr (LambdaHelper<LambdaTy>::NArgs == 1)
func(Level);
else
func();
}
};

#define ODBG_IF_BASE(Type, Level, Callback) \
if (::llvm::offload::debug::isDebugEnabled()) { \
uint32_t RealLevel = (Level); \
if (::llvm::offload::debug::shouldPrintDebug(GETNAME(TARGET_NAME), (Type), \
RealLevel)) { \
auto F = Callback; \
::llvm::offload::debug::LambdaIf<decltype(F)>::dispatch(F, RealLevel); \
} \
}

#define ODBG_IF_3(Type, Level, Callback) ODBG_IF_BASE(Type, Level, Callback)
#define ODBG_IF_2(Type, Callback) ODBG_IF_3(Type, 1, Callback)
#define ODBG_IF_1(Callback) ODBG_IF_2("default", Callback)
#define ODBG_IF_SELECT(Type, Level, Callback, NArgs, ...) ODBG_IF_##NArgs
#define ODBG_IF(...) \
ODBG_IF_SELECT(__VA_ARGS__ __VA_OPT__(, ) 3, 2, 1)(__VA_ARGS__)

#else

inline bool isDebugEnabled() { return false; }
Expand All @@ -496,6 +529,9 @@ inline bool isDebugEnabled() { return false; }
#define ODBG_OS_STREAM(Stream, Type, Level, Callback)
#define ODBG_OS(...)

#define ODBG_IF_BASE(Type, Level, Callback)
#define ODBG_IF(...)

#endif

} // namespace llvm::offload::debug
Expand Down
Loading