Skip to content

[CK][Windows]: fmha_fwd_head_grouping guard posix headers and use of posix only functions on Windows #5979

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

Closed
astrelsky wants to merge 1 commit into from
Closed

[CK][Windows]: fmha_fwd_head_grouping guard posix headers and use of posix only functions on Windows #5979

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
23 changes: 23 additions & 0 deletions projects/composablekernel/example/ck_tile/01_fmha/fmha_fwd_head_grouping.hpp
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,9 @@
#include <algorithm>
#include <cctype>
#include <cstdio>
#ifndef _WIN32
#include <dirent.h>
#endif
#include <fstream>
#include <iostream>
#include <limits>
Expand Down Expand Up @@ -102,6 +104,8 @@ inline std::optional<kfd_device_location> get_current_kfd_location()
((props.pciBusID & 0xff) << 8) | ((props.pciDeviceID & 0x1f) << 3)};
}

#ifndef _WIN32

inline std::optional<std::string> find_matching_kfd_node(const kfd_device_location& loc)
{
constexpr const char* kKfdNodesDir = "/sys/class/kfd/kfd/topology/nodes";
Expand Down Expand Up @@ -176,6 +180,25 @@ inline size_t get_kfd_sysfs_llc_cache_bytes()
return read_kfd_node_l3_bytes(*node);
}

#else

inline std::optional<std::string> find_matching_kfd_node(const kfd_device_location& loc)
{
return std::nullopt;
}

inline size_t read_kfd_node_l3_bytes(const std::string& node_name)
{
return 0;
}

inline size_t get_kfd_sysfs_llc_cache_bytes()
{
return 0;
}

#endif // _WIN32

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)
Expand Down