ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-20 04:49:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge commit '0f8e33f81120e5734ef47a6a169ad85c6560cbd8' into develop

Browse Source
This commit is contained in:
assistant-librarian[bot]
2025-09-09 03:25:20 +00:00
parent e8724677d6
commit e702cfd808
131 changed files with 8731 additions and 5329 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
include/ck/host_utility/device_prop.hpp
View File

@@ -75,6 +75,34 @@ inline bool is_xdl_supported()
;
}
template <typename ADataType, typename BDataType, index_t MPerXDL, index_t NPerXDL>
inline bool is_xdl_wmma_supported()
{
if(ck::get_device_name() == "gfx908" || ck::get_device_name() == "gfx90a" ||
ck::get_device_name() == "gfx942" || ck::get_device_name() == "gfx950")
{
return true;
}
#if defined(CK_ENABLE_DYNAMIC_WARP_SIZE)
else if(is_gfx12_supported() || is_gfx11_supported())
{
if constexpr((MPerXDL != 16) || (NPerXDL != 16))
{
return false;
}
if constexpr(sizeof(ADataType) > 2 || sizeof(BDataType) > 2)
{
return false;
}
return true;
}
#endif
else
{
return false;
}
}
inline bool is_lds_direct_load_supported()
{
// Check if direct loads from global memory to LDS are supported.