mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-12 01:10:17 +00:00
5d37d7bff4
* delete obselete files * move files * build * update cmake * update cmake * fix build * reorg examples * update cmake for example and test
21 lines
460 B
C++
21 lines
460 B
C++
#ifndef CK_UTILITY_HPP
|
|
#define CK_UTILITY_HPP
|
|
|
|
#include "config.hpp"
|
|
|
|
namespace ck {
|
|
|
|
__device__ constexpr index_t get_wave_size() { return CK_GPU_WAVE_SIZE; }
|
|
|
|
__device__ index_t get_thread_local_1d_id() { return threadIdx.x; }
|
|
|
|
__device__ index_t get_wave_local_1d_id() { return threadIdx.x / get_wave_size(); }
|
|
|
|
__device__ index_t get_block_1d_id() { return blockIdx.x; }
|
|
|
|
__device__ index_t get_grid_size() { return gridDim.x; }
|
|
|
|
} // namespace ck
|
|
|
|
#endif
|