mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-16 10:59:55 +00:00
3f732cceab
* adding compilation for multiple targets
* fix build
* clean
* update Jekinsfile
* update readme
* update Jenkins
* use ck::half_t instead of ushort for bf16
* rename enum classes
* clean
* rename
* clean
[ROCm/composable_kernel commit: cd167e492a]
17 lines
418 B
C++
17 lines
418 B
C++
#pragma once
|
|
#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
|