From 7b8026faa9fcb536a8bd656ca29dae093d97e7e7 Mon Sep 17 00:00:00 2001 From: Illia Silin <98187287+illsilin@users.noreply.github.com> Date: Thu, 11 Apr 2024 16:40:45 -0700 Subject: [PATCH] [HotFix] pass XDL and WMMA macros to libs that use CK (#1234) [ROCm/composable_kernel commit: d7f05fb996a55a37bb005f27d0c09c7595e5aee2] --- include/ck/config.h.in | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/include/ck/config.h.in b/include/ck/config.h.in index 1748344756..eb9049b599 100644 --- a/include/ck/config.h.in +++ b/include/ck/config.h.in @@ -104,6 +104,20 @@ #cmakedefine CK_ENABLE_INSTANCES_ONLY @CK_ENABLE_INSTANCES_ONLY@ #endif +// +// CK kernels which support XDL (MI series) +// +#ifndef CK_USE_XDL +#cmakedefine CK_USE_XDL @CK_USE_XDL@ +#endif + +// +// CK Kernels which support WMMA (recent Navi series) +// +#ifndef CK_USE_WMMA +#cmakedefine CK_USE_WMMA @CK_USE_WMMA@ +#endif + // clang-format on #endif // CK_CONFIG_H_IN