SWDEV-402274 - Adds CG API group_dim() in class thread_block

Change-Id: I1c2569fffbc6fc730df9ba80786459c71140d327


[ROCm/clr commit: 422d3ea035]
Этот коммит содержится в:
Sourabh Betigeri
2023-06-01 22:55:49 -07:00
коммит произвёл Rahul Garg
родитель c8e65415ea
Коммит c00cd118d3
2 изменённых файлов: 6 добавлений и 1 удалений
+1
Просмотреть файл
@@ -252,6 +252,7 @@ class thread_block : public thread_group {
__CG_QUALIFIER__ uint32_t thread_rank() const { return internal::workgroup::thread_rank(); }
__CG_QUALIFIER__ bool is_valid() const { return internal::workgroup::is_valid(); }
__CG_QUALIFIER__ void sync() const { internal::workgroup::sync(); }
__CG_QUALIFIER__ dim3 group_dim() { return internal::workgroup::block_dim(); }
};
/** \brief User exposed API interface to construct workgroup cooperative
+5 -1
Просмотреть файл
@@ -195,12 +195,16 @@ __CG_STATIC_QUALIFIER__ uint32_t thread_rank() {
}
__CG_STATIC_QUALIFIER__ bool is_valid() {
// TODO(mahesha) any functionality need to be added here? I believe not
return true;
}
__CG_STATIC_QUALIFIER__ void sync() { __syncthreads(); }
__CG_STATIC_QUALIFIER__ dim3 block_dim() {
return (dim3(static_cast<uint32_t>(blockDim.x), static_cast<uint32_t>(blockDim.y),
static_cast<uint32_t>(blockDim.z)));
}
} // namespace workgroup
namespace tiled_group {