diff --git a/example/ck_tile/01_fmha/fmha_fwd_head_grouping.hpp b/example/ck_tile/01_fmha/fmha_fwd_head_grouping.hpp index 9cd1fb9cdc..9dad951d41 100644 --- a/example/ck_tile/01_fmha/fmha_fwd_head_grouping.hpp +++ b/example/ck_tile/01_fmha/fmha_fwd_head_grouping.hpp @@ -8,13 +8,16 @@ #include #include #include -#include #include #include #include #include #include +#ifdef __linux__ +#include +#endif + #ifndef CK_TILE_FMHA_ENABLE_HEAD_GROUPING #define CK_TILE_FMHA_ENABLE_HEAD_GROUPING 1 #endif @@ -70,6 +73,8 @@ inline std::optional read_property_value(const std::string& filepath, return std::nullopt; } +#if defined(__linux__) + struct kfd_device_location { int domain = 0; @@ -176,6 +181,12 @@ inline size_t get_kfd_sysfs_llc_cache_bytes() return read_kfd_node_l3_bytes(*node); } +#else + +inline size_t get_kfd_sysfs_llc_cache_bytes() { return 0; } + +#endif + inline size_t get_default_llc_cache_bytes_for_arch(const std::string& arch); inline size_t resolve_llc_cache_bytes_uncached(const std::string& arch)