mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-13 01:36:06 +00:00
ec7c2e912e
* format * improving pipeline * fix typo * format * adding thread group * adding thread group * adding thread group * adding gemm pipeline * tweak * refactor * refactor * add missing type convert * refactor * refactor * refactor * clean * fix build * refactor * format * clean up * use remove_cvref_t * clean * clean up * clean up * clean up
19 lines
422 B
C++
19 lines
422 B
C++
#pragma once
|
|
#include "get_id.hpp"
|
|
|
|
namespace ck {
|
|
|
|
template <index_t ThreadPerBlock>
|
|
struct ThisThreadBlock
|
|
{
|
|
static constexpr index_t kNumThread_ = ThreadPerBlock;
|
|
|
|
__device__ static constexpr index_t GetNumOfThread() { return kNumThread_; }
|
|
|
|
__device__ static constexpr bool IsBelong() { return true; }
|
|
|
|
__device__ static index_t GetThreadId() { return get_thread_local_1d_id(); }
|
|
};
|
|
|
|
} // namespace ck
|