ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-21 21:39:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge commit '6635d1bb888e3f51ec1125ff7d6f54a2ec054a10' into develop

Browse Source
This commit is contained in:
assistant-librarian[bot]
2025-06-10 15:11:42 +00:00
parent ffd987e013
commit 84c5350d9c
6 changed files with 14 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
include/ck/utility/get_id.hpp
View File

@@ -9,8 +9,11 @@ namespace ck {
__host__ __device__ constexpr index_t get_warp_size()
{
// warpSize is defined by HIP
return warpSize;
#if defined(__GFX9__) || !defined(__HIP_DEVICE_COMPILE__)
return 64;
#else
return 32;
#endif
}
__device__ index_t get_thread_local_1d_id() { return threadIdx.x; }

7
include/ck_tile/core/arch/arch.hpp
View File

@@ -50,8 +50,11 @@ enum struct memory_operation_enum : std::uint16_t
CK_TILE_HOST_DEVICE constexpr index_t get_warp_size()
{
// warpSize is defined by HIP
return warpSize;
#if defined(__GFX9__) || !defined(__HIP_DEVICE_COMPILE__)
return 64;
#else
return 32;
#endif
}
CK_TILE_DEVICE index_t get_grid_size() { return gridDim.x; }